diff options
Diffstat (limited to 'tests/test_pagination.py')
-rw-r--r-- | tests/test_pagination.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/tests/test_pagination.py b/tests/test_pagination.py index 9f26140..0c83fe4 100644 --- a/tests/test_pagination.py +++ b/tests/test_pagination.py | |||
@@ -16,7 +16,7 @@ UNLIKELY_HASHTAG = "fgiztsshwiaqqiztpmmjbtvmescsculuvmgjgopwoeidbcrixp" | |||
16 | def many_statuses(api, n=10, suffix=''): | 16 | def many_statuses(api, n=10, suffix=''): |
17 | statuses = list() | 17 | statuses = list() |
18 | for i in range(n): | 18 | for i in range(n): |
19 | status = api.status_post("Toot number {}!{}".format(i, suffix)) | 19 | status = api.status_post(f"Toot number {i}!{suffix}") |
20 | statuses.append(status) | 20 | statuses.append(status) |
21 | yield statuses | 21 | yield statuses |
22 | for status in statuses: | 22 | for status in statuses: |
@@ -102,10 +102,15 @@ def test_link_headers(api): | |||
102 | 102 | ||
103 | _id='abc1234' | 103 | _id='abc1234' |
104 | 104 | ||
105 | rmock.register_uri('GET', requests_mock.ANY, json=[{"foo": "bar"}], headers={"link":""" | 105 | rmock.register_uri( |
106 | <{base}/api/v1/timelines/tag/{tag}?max_id={_id}>; rel="next", <{base}/api/v1/timelines/tag/{tag}?since_id={_id}>; rel="prev" | 106 | 'GET', |
107 | """.format(base=api.api_base_url, tag=UNLIKELY_HASHTAG, _id=_id).strip() | 107 | requests_mock.ANY, |
108 | }) | 108 | json=[{"foo": "bar"}], |
109 | headers={ | ||
110 | "link": f"<{api.api_base_url}/api/v1/timelines/tag/{UNLIKELY_HASHTAG}?max_id={_id}>; rel=\"next\", " | ||
111 | f"<{api.api_base_url}/api/v1/timelines/tag/{UNLIKELY_HASHTAG}?since_id={_id}>; rel=\"prev\"" | ||
112 | } | ||
113 | ) | ||
109 | 114 | ||
110 | resp = api.timeline_hashtag(UNLIKELY_HASHTAG) | 115 | resp = api.timeline_hashtag(UNLIKELY_HASHTAG) |
111 | assert resp._pagination_next['max_id'] == _id | 116 | assert resp._pagination_next['max_id'] == _id |