Add bookmarks test

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/test_bookmarks.py b/tests/test_bookmarks.py
new file mode 100644
index 0000000..a90a43e
--- /dev/null
+++ b/tests/test_bookmarks.py
@@ -0,0 +1,21 @@
+import pytest
+
+@pytest.mark.vcr()
+def test_bookmarks(api, status):
+    status_bookmarked = api.status_bookmark(status)
+    assert status_bookmarked
+    assert status_bookmarked.bookmarked == True
+    
+    bookmarked_statuses = api.bookmarks()
+    assert bookmarked_statuses
+    assert len(bookmarked_statuses) > 0
+    assert status_bookmarked == bookmarked_statuses[0]
+    
+    status_unbookmarked = api.status_unbookmark(status_bookmarked)
+    assert status_unbookmarked
+    assert status_unbookmarked.bookmarked == False
+    
+    bookmarked_statuses_2 = api.bookmarks()
+    assert not bookmarked_statuses_2 is None
+    assert len(bookmarked_statuses_2) == len(bookmarked_statuses) - 1
+