1 files changed, 34 insertions, 0 deletions

diff --git a/tests/test_pagination.py b/tests/test_pagination.py
new file mode 100644
index 0000000..5b59143
--- /dev/null
+++ b/tests/test_pagination.py
@@ -0,0 +1,34 @@
+import pytest
+from contextlib import contextmanager
+
+
+@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_remaining(api):
+    UNLIKELY_HASHTAG = "fgiztsshwiaqqiztpmmjbtvmescsculuvmgjgopwoeidbcrixp"
+    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