Merge pull request #109 from codl/test-suite

Test suite

1 files changed, 37 insertions, 0 deletions

diff --git a/tests/test_timeline.py b/tests/test_timeline.py
new file mode 100644
index 0000000..d713fca
--- /dev/null
+++ b/tests/test_timeline.py
@@ -0,0 +1,37 @@
+import pytest
+
+@pytest.mark.vcr()
+def test_public_tl_anonymous(api_anonymous, status):
+    tl = api_anonymous.timeline_public()
+    assert status['id'] in map(lambda st: st['id'], tl)
+    # although tempting, we can't do
+    #     assert status in tl
+    # because the statuses returned in the tl have additional
+    # pagination-related attributes
+
+@pytest.mark.vcr()
+def test_public_tl(api, status):
+    public = api.timeline_public()
+    local = api.timeline_local()
+    assert status['id'] in map(lambda st: st['id'], public)
+    assert status['id'] in map(lambda st: st['id'], local)
+
+@pytest.mark.vcr()
+def test_home_tl(api, status):
+    tl = api.timeline_home()
+    assert status['id'] in map(lambda st: st['id'], tl)
+
+@pytest.mark.vcr()
+def test_hashtag_tl(api):
+    status = api.status_post('#hoot (hashtag toot)')
+    tl = api.timeline_hashtag('hoot')
+    try:
+        assert status['id'] in map(lambda st: st['id'], tl)
+    finally:
+        api.status_delete(status['id'])
+
+@pytest.mark.vcr()
+def test_home_tl_anonymous_throws(api_anonymous):
+    from mastodon.Mastodon import MastodonAPIError
+    with pytest.raises(MastodonAPIError):
+        tl = api_anonymous.timeline_home()