Browse Source

Deezer's unit test

Cqoicebordel 10 years ago
parent
commit
0f52cc7542
2 changed files with 57 additions and 0 deletions
  1. 56 0
      searx/tests/engines/test_deezer.py
  2. 1 0
      searx/tests/test_engines.py

+ 56 - 0
searx/tests/engines/test_deezer.py

@@ -0,0 +1,56 @@
+from collections import defaultdict
+import mock
+from searx.engines import deezer
+from searx.testing import SearxTestCase
+
+
+class TestDeezerEngine(SearxTestCase):
+
+    def test_request(self):
+        query = 'test_query'
+        dicto = defaultdict(dict)
+        dicto['pageno'] = 0
+        params = deezer.request(query, dicto)
+        self.assertTrue('url' in params)
+        self.assertTrue(query in params['url'])
+        self.assertTrue('deezer.com' in params['url'])
+
+    def test_response(self):
+        self.assertRaises(AttributeError, deezer.response, None)
+        self.assertRaises(AttributeError, deezer.response, [])
+        self.assertRaises(AttributeError, deezer.response, '')
+        self.assertRaises(AttributeError, deezer.response, '[]')
+
+        response = mock.Mock(text='{}')
+        self.assertEqual(deezer.response(response), [])
+
+        response = mock.Mock(text='{"data": []}')
+        self.assertEqual(deezer.response(response), [])
+
+        json = """
+        {"data":[
+            {"id":100, "title":"Title of track",
+            "link":"http:\/\/www.deezer.com\/track\/1094042","duration":232,
+            "artist":{"id":200,"name":"Artist Name",
+                "link":"http:\/\/www.deezer.com\/artist\/1217","type":"artist"},
+            "album":{"id":118106,"title":"Album Title","type":"album"},"type":"track"}
+        ]}
+        """
+        response = mock.Mock(text=json)
+        results = deezer.response(response)
+        self.assertEqual(type(results), list)
+        self.assertEqual(len(results), 1)
+        self.assertEqual(results[0]['title'], 'Title of track')
+        self.assertEqual(results[0]['url'], 'http://www.deezer.com/track/1094042')
+        self.assertEqual(results[0]['content'], 'Artist Name • Album Title • Title of track')
+
+        json = """
+        {"data":[
+            {"id":200,"name":"Artist Name",
+            "link":"http:\/\/www.deezer.com\/artist\/1217","type":"artist"}
+        ]}
+        """
+        response = mock.Mock(text=json)
+        results = deezer.response(response)
+        self.assertEqual(type(results), list)
+        self.assertEqual(len(results), 0)

+ 1 - 0
searx/tests/test_engines.py

@@ -1,3 +1,4 @@
 from searx.tests.engines.test_bing import *  # noqa
+from searx.tests.engines.test_deezer import *  # noqa
 from searx.tests.engines.test_dummy import *  # noqa
 from searx.tests.engines.test_github import *  # noqa