test(bookmarks): ✅ update tests for bookmarks

Author: MerleLiuKun

pytwitter/rate_limit.py

@@ -126,6 +126,17 @@ def get_limit(self, auth_type, method="GET"):
     regex=re.compile(r"/users/\d+/likes/\d+"),
     LIMIT_USER_POST=50,
 )
+USER_BOOKMARK_TWEET = Endpoint(
+    resource="/users/:id/bookmarks",
+    regex=re.compile(r"/users/\d+/bookmarks"),
+    LIMIT_USER_GET=180,
+    LIMIT_USER_POST=50,
+)
+USER_BOOKMARK_TWEET_REMOVE = Endpoint(
+    resource="/users/:id/bookmarks/:tweet_id",
+    regex=re.compile(r"/users/\d+/bookmarks/\d+"),
+    LIMIT_USER_DELETE=50,
+)
 TWEET_HIDDEN = Endpoint(
     resource="/tweets/:id/hidden",
     regex=re.compile(r"/tweets/\d+/hidden"),

testdata/apis/tweet/tweets_by_user_bookmark.json

@@ -0,0 +1 @@
+{"data":[{"created_at":"2021-02-18T17:12:47.000Z","source":"Twitter Web App","id":"1362449997430542337","text":"Honored to be the first developer to be featured in @TwitterDev's love fest 🥰♥️😍 https://t.co/g8TsPoZsij"},{"created_at":"2021-02-26T21:38:43.000Z","source":"Twitter Web App","id":"1365416026435854338","text":"We're so happy for our Official Partner @Brandwatch and their big news. https://t.co/3DwWBNSq0o https://t.co/bDUGbgPkKO"},{"created_at":"2020-08-20T16:41:00.000Z","source":"Twitter Web App","id":"1296487407475462144","text":"Check out this feature on @TwitterDev to learn more about how we're mining social media data to make sense of this evolving #publichealth crisis https://t.co/sIFLXRSvEX."},{"created_at":"2020-08-14T18:55:42.000Z","source":"Twitter for Android","id":"1294346980072624128","text":"I awake from five years of slumber https://t.co/OEPVyAFcfB"},{"created_at":"2020-07-14T21:38:10.000Z","source":"Twitter for  iPhone","id":"1283153843367206912","text":"@wongmjane Wish we could tell you more, but I’m only a teapot 👀"}],"meta":{"result_count":5,"next_token":"zldjwdz3w6sba13nbs0mbravfipbtqvbiqplg9h0p4k"}}

tests/apis/test_tweets.py

@@ -70,7 +70,7 @@ def test_get_tweets(api, helpers):
 
 
 @responses.activate
-def test_like_and_unlike_tweet(api_with_user, helpers):
+def test_like_and_unlike_tweet(api_with_user):
     user_id, tweet_id = "123456", "10987654321"
 
     responses.add(
@@ -125,6 +125,51 @@ def test_tweet_liking_users(api, helpers):
     assert resp_json["includes"]["tweets"][0]["id"] == "1383963809702846470"
 
 
+@responses.activate
+def test_get_user_bookmarks(api_with_user, helpers):
+    user_id = "2244994945"
+    tweets_data = helpers.load_json_data(
+        "testdata/apis/tweet/tweets_by_user_bookmark.json"
+    )
+    responses.add(
+        responses.GET,
+        url=f"https://api.twitter.com/2/users/{user_id}/bookmarks",
+        json=tweets_data,
+    )
+
+    resp = api_with_user.get_bookmark_tweets(
+        user_id=user_id,
+        max_results=5,
+        expansions="attachments.media_keys",
+        media_fields="type,duration_ms",
+    )
+    assert len(resp.data) == 5
+    assert resp.data[0].id == "1362449997430542337"
+
+
+@responses.activate
+def test_user_bookmark_tweet_or_remove_tweet(api_with_user):
+    user_id, tweet_id = "2244994945", "1228393702244134912"
+
+    responses.add(
+        responses.POST,
+        url=f"https://api.twitter.com/2/users/{user_id}/bookmarks",
+        json={"data": {"bookmarked": True}},
+    )
+
+    resp = api_with_user.bookmark_tweet(user_id=user_id, tweet_id=tweet_id)
+    assert resp["data"]["bookmarked"]
+
+    responses.add(
+        responses.DELETE,
+        url=f"https://api.twitter.com/2/users/{user_id}/bookmarks/{tweet_id}",
+        json={"data": {"bookmarked": False}},
+    )
+
+    resp = api_with_user.bookmark_tweet_remove(user_id=user_id, tweet_id=tweet_id)
+    assert not resp["data"]["bookmarked"]
+
+
 @responses.activate
 def test_get_tweet_quote_tweets(api, helpers):
     tweets_data = helpers.load_json_data(