add test for link headers. this adds requests-mock as a dependency
This commit is contained in:
parent
eca31ea732
commit
a815c10f9b
3 changed files with 29 additions and 0 deletions
1
Pipfile
1
Pipfile
|
@ -13,3 +13,4 @@ pytest-cov = "*"
|
|||
vcrpy = "*"
|
||||
pytest-vcr = "<1"
|
||||
pytest-mock = "*"
|
||||
requests-mock = "*"
|
||||
|
|
8
Pipfile.lock
generated
8
Pipfile.lock
generated
|
@ -313,6 +313,14 @@
|
|||
],
|
||||
"version": "==2.20.1"
|
||||
},
|
||||
"requests-mock": {
|
||||
"hashes": [
|
||||
"sha256:7a5fa99db5e3a2a961b6f20ed40ee6baeff73503cf0a553cc4d679409e6170fb",
|
||||
"sha256:8ca0628dc66d3f212878932fd741b02aa197ad53fd2228164800a169a4a826af"
|
||||
],
|
||||
"index": "pypi",
|
||||
"version": "==1.5.2"
|
||||
},
|
||||
"six": {
|
||||
"hashes": [
|
||||
"sha256:70e8a77beed4562e7f14fe23a786b54f6296e34344c23bc42f07b15018ff98e9",
|
||||
|
|
|
@ -1,5 +1,10 @@
|
|||
import pytest
|
||||
from contextlib import contextmanager
|
||||
try:
|
||||
from mock import MagicMock
|
||||
except ImportError:
|
||||
from unittest.mock import MagicMock
|
||||
import requests_mock
|
||||
|
||||
UNLIKELY_HASHTAG = "fgiztsshwiaqqiztpmmjbtvmescsculuvmgjgopwoeidbcrixp"
|
||||
|
||||
|
@ -44,3 +49,18 @@ def test_fetch_remaining(api):
|
|||
hashtag_remaining = api.fetch_remaining(hashtag)
|
||||
assert hashtag_remaining
|
||||
assert len(hashtag_remaining) >= 30
|
||||
|
||||
def test_link_headers(api):
|
||||
rmock = requests_mock.Adapter()
|
||||
api.session.mount(api.api_base_url, rmock)
|
||||
|
||||
_id='abc1234'
|
||||
|
||||
rmock.register_uri('GET', requests_mock.ANY, json=[{"foo": "bar"}], headers={"link":"""
|
||||
<{base}/api/v1/timelines/tag/{tag}?max_id={_id}>; rel="next", <{base}/api/v1/timelines/tag/{tag}?since_id={_id}>; rel="prev"
|
||||
""".format(base=api.api_base_url, tag=UNLIKELY_HASHTAG, _id=_id).strip()
|
||||
})
|
||||
|
||||
resp = api.timeline_hashtag(UNLIKELY_HASHTAG)
|
||||
assert resp[0]._pagination_next['max_id'] == _id
|
||||
assert resp[0]._pagination_prev['since_id'] == _id
|
||||
|
|
Loading…
Reference in a new issue