67 líneas
2,2 KiB
Python
67 líneas
2,2 KiB
Python
import pytest
|
|
from contextlib import contextmanager
|
|
try:
|
|
from mock import MagicMock
|
|
except ImportError:
|
|
from unittest.mock import MagicMock
|
|
import requests_mock
|
|
|
|
UNLIKELY_HASHTAG = "fgiztsshwiaqqiztpmmjbtvmescsculuvmgjgopwoeidbcrixp"
|
|
|
|
|
|
@contextmanager
|
|
def many_statuses(api, n=10, suffix=''):
|
|
statuses = list()
|
|
for i in range(n):
|
|
status = api.status_post("Toot number {}!{}".format(i, suffix))
|
|
statuses.append(status)
|
|
yield statuses
|
|
for status in statuses:
|
|
api.status_delete(status['id'])
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
def test_fetch_next_previous(api):
|
|
account = api.account_verify_credentials()
|
|
with many_statuses(api):
|
|
statuses = api.account_statuses(account['id'], limit=5)
|
|
next_statuses = api.fetch_next(statuses)
|
|
assert next_statuses
|
|
previous_statuses = api.fetch_previous(next_statuses)
|
|
assert previous_statuses
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
def test_fetch_next_previous_from_pagination_info(api):
|
|
account = api.account_verify_credentials()
|
|
with many_statuses(api):
|
|
statuses = api.account_statuses(account['id'], limit=5)
|
|
next_statuses = api.fetch_next(statuses[-1]._pagination_next)
|
|
assert next_statuses
|
|
previous_statuses = api.fetch_previous(next_statuses[0]._pagination_prev)
|
|
assert previous_statuses
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
def test_fetch_remaining(api):
|
|
with many_statuses(api, n=30, suffix=' #'+UNLIKELY_HASHTAG):
|
|
hashtag = api.timeline_hashtag(UNLIKELY_HASHTAG, limit=10)
|
|
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
|