From 2d81e9b2cbd3e3c1abd3a472220c667e89b29f5e Mon Sep 17 00:00:00 2001
From: Lorenz Diener <lorenzd@gmail.com>
Date: Mon, 30 Jul 2018 23:08:30 +0200
Subject: Add filter tests

---
 tests/test_filters.py | 91 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 91 insertions(+)
 create mode 100644 tests/test_filters.py

(limited to 'tests/test_filters.py')

diff --git a/tests/test_filters.py b/tests/test_filters.py
new file mode 100644
index 0000000..dc39ddb
--- /dev/null
+++ b/tests/test_filters.py
@@ -0,0 +1,91 @@
+import pytest
+import time
+
+@pytest.mark.vcr()
+def test_filter_create(api):
+    keyword_filter = api.filter_create("anime", ['notifications'], irreversible = False, whole_word = True, expires_in = None)
+    try:
+        assert(keyword_filter)
+        
+        all_filters = api.filters()
+        assert(keyword_filter in all_filters)
+        assert(keyword_filter.irreversible == False)
+        assert(keyword_filter.whole_word == True)
+        
+        keyword_filter_2 = api.filter(keyword_filter.id)
+        assert(keyword_filter == keyword_filter_2)
+    finally:
+        api.filter_delete(keyword_filter)
+  
+  
+    keyword_filter = api.filter_create("anime", ['notifications'], irreversible = False, whole_word = False, expires_in = None)
+    try:
+        assert(keyword_filter)
+        assert(keyword_filter.irreversible == False)
+        assert(keyword_filter.whole_word == False)
+        
+        all_filters = api.filters()
+        assert(keyword_filter in all_filters)
+        
+        keyword_filter_2 = api.filter(keyword_filter.id)
+        assert(keyword_filter == keyword_filter_2)
+    finally:
+        api.filter_delete(keyword_filter)
+        
+@pytest.mark.vcr()
+def test_filter_serverside(api, api2):
+    api.account_follow(api2.account_verify_credentials())
+    keyword_filter_1 = api.filter_create("anime", ['home'], irreversible = True, whole_word = False, expires_in = None)
+    keyword_filter_2 = api.filter_create("girugamesh", ['home'], irreversible = True, whole_word = True, expires_in = None)
+    keyword_filter_3 = api.filter_create("japanimation", ['notifications'], irreversible = True, whole_word = True, expires_in = None)
+    time.sleep(2)
+    status_1 = api2.toot("I love animes")
+    status_2 = api2.toot("Girugamesh!")
+    status_3 = api2.toot("Girugameshnetworking!")
+    status_4 = api2.toot("I love japanimation!")
+    time.sleep(2)
+    tl = api.timeline_home()
+    try:
+        assert not status_1['id'] in map(lambda st: st['id'], tl)
+        assert not status_2['id'] in map(lambda st: st['id'], tl)
+        assert status_3['id'] in map(lambda st: st['id'], tl)
+        assert status_4['id'] in map(lambda st: st['id'], tl)
+    finally:
+        api.filter_delete(keyword_filter_1)
+        api.filter_delete(keyword_filter_2)
+        api.filter_delete(keyword_filter_3)
+        api2.status_delete(status_1)
+        api2.status_delete(status_2)
+        api2.status_delete(status_3)
+        api2.status_delete(status_4)
+        
+@pytest.mark.vcr()
+def test_filter_clientside(api, api2):
+    api.account_follow(api2.account_verify_credentials())
+    keyword_filter_1 = api.filter_create("anime", ['home'], irreversible = False, whole_word = False, expires_in = None)
+    keyword_filter_2 = api.filter_create("girugamesh", ['home'], irreversible = False, whole_word = True, expires_in = None)
+    keyword_filter_3 = api.filter_create("japanimation", ['notifications'], irreversible = False, whole_word = True, expires_in = None)
+    status_1 = api2.toot("I love animes")
+    status_2 = api2.toot("Girugamesh!")
+    status_3 = api2.toot("Girugameshnetworking!")
+    status_4 = api2.toot("I love japanimation!")
+    tl = api.timeline_home()
+    try:
+        assert status_1['id'] in map(lambda st: st['id'], tl)
+        assert status_2['id'] in map(lambda st: st['id'], tl)
+        assert status_3['id'] in map(lambda st: st['id'], tl)
+        assert status_4['id'] in map(lambda st: st['id'], tl)
+        
+        filtered = api.filters_apply(tl, [keyword_filter_1, keyword_filter_2, keyword_filter_3], 'home')
+        assert not status_1['id'] in map(lambda st: st['id'], filtered)
+        assert not status_2['id'] in map(lambda st: st['id'], filtered)
+        assert status_3['id'] in map(lambda st: st['id'], filtered)
+        assert status_4['id'] in map(lambda st: st['id'], filtered)
+    finally:
+        api.filter_delete(keyword_filter_1)
+        api.filter_delete(keyword_filter_2)
+        api.filter_delete(keyword_filter_3)
+        api2.status_delete(status_1)
+        api2.status_delete(status_2)
+        api2.status_delete(status_3)
+        api2.status_delete(status_4)
-- 
cgit v1.2.3