From 655a45e086f8c2e05be14dc55aa1fe366ffbb171 Mon Sep 17 00:00:00 2001 From: codl Date: Tue, 28 Nov 2017 01:41:18 +0100 Subject: add test for pagination --- tests/test_pagination.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 tests/test_pagination.py (limited to 'tests/test_pagination.py') diff --git a/tests/test_pagination.py b/tests/test_pagination.py new file mode 100644 index 0000000..30bdf89 --- /dev/null +++ b/tests/test_pagination.py @@ -0,0 +1,24 @@ +import pytest +from contextlib import contextmanager + + +@contextmanager +def many_statuses(api, n=10): + statuses = list() + for i in range(n): + status = api.status_post("Toot number {}!".format(i)) + 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 -- cgit v1.2.3