Merge pull request #504 from kvch/frinkiac-engine

Simpson screenshot search engine

1 files changed, 49 insertions, 0 deletions

diff --git a/tests/unit/engines/test_frinkiac.py b/tests/unit/engines/test_frinkiac.py
new file mode 100644
index 000000000..f3eb021d2
--- /dev/null
+++ b/tests/unit/engines/test_frinkiac.py
@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+from collections import defaultdict
+import mock
+from json import dumps
+from searx.engines import frinkiac
+from searx.testing import SearxTestCase
+
+
+class TestFrinkiacEngine(SearxTestCase):
+
+    def test_request(self):
+        query = 'test_query'
+        request_dict = defaultdict(dict)
+        params = frinkiac.request(query, request_dict)
+        self.assertTrue('url' in params)
+
+    def test_response(self):
+        self.assertRaises(AttributeError, frinkiac.response, None)
+        self.assertRaises(AttributeError, frinkiac.response, [])
+        self.assertRaises(AttributeError, frinkiac.response, '')
+        self.assertRaises(AttributeError, frinkiac.response, '[]')
+
+        text = """
+[{"Id":770931,
+  "Episode":"S06E18",
+  "Timestamp":534616,
+  "Filename":""},
+ {"Id":1657080,
+  "Episode":"S12E14",
+  "Timestamp":910868,
+  "Filename":""},
+ {"Id":1943753,
+  "Episode":"S14E21",
+  "Timestamp":773439,
+  "Filename":""},
+ {"Id":107835,
+  "Episode":"S02E03",
+  "Timestamp":531709,
+  "Filename":""}]
+        """
+
+        response = mock.Mock(text=text)
+        results = frinkiac.response(response)
+        self.assertEqual(type(results), list)
+        self.assertEqual(len(results), 4)
+        self.assertEqual(results[0]['title'], u'S06E18')
+        self.assertEqual(results[0]['url'], 'https://frinkiac.com/?p=caption&e=S06E18&t=534616')
+        self.assertEqual(results[0]['thumbnail_src'], 'https://frinkiac.com/img/S06E18/534616/medium.jpg')
+        self.assertEqual(results[0]['img_src'], 'https://frinkiac.com/img/S06E18/534616.jpg')