浏览代码

Deezer's unit test

Cqoicebordel 10 年前
父节点
当前提交
0f52cc7542
共有 2 个文件被更改,包括 57 次插入0 次删除
  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 @@
1
+from collections import defaultdict
2
+import mock
3
+from searx.engines import deezer
4
+from searx.testing import SearxTestCase
5
+
6
+
7
+class TestDeezerEngine(SearxTestCase):
8
+
9
+    def test_request(self):
10
+        query = 'test_query'
11
+        dicto = defaultdict(dict)
12
+        dicto['pageno'] = 0
13
+        params = deezer.request(query, dicto)
14
+        self.assertTrue('url' in params)
15
+        self.assertTrue(query in params['url'])
16
+        self.assertTrue('deezer.com' in params['url'])
17
+
18
+    def test_response(self):
19
+        self.assertRaises(AttributeError, deezer.response, None)
20
+        self.assertRaises(AttributeError, deezer.response, [])
21
+        self.assertRaises(AttributeError, deezer.response, '')
22
+        self.assertRaises(AttributeError, deezer.response, '[]')
23
+
24
+        response = mock.Mock(text='{}')
25
+        self.assertEqual(deezer.response(response), [])
26
+
27
+        response = mock.Mock(text='{"data": []}')
28
+        self.assertEqual(deezer.response(response), [])
29
+
30
+        json = """
31
+        {"data":[
32
+            {"id":100, "title":"Title of track",
33
+            "link":"http:\/\/www.deezer.com\/track\/1094042","duration":232,
34
+            "artist":{"id":200,"name":"Artist Name",
35
+                "link":"http:\/\/www.deezer.com\/artist\/1217","type":"artist"},
36
+            "album":{"id":118106,"title":"Album Title","type":"album"},"type":"track"}
37
+        ]}
38
+        """
39
+        response = mock.Mock(text=json)
40
+        results = deezer.response(response)
41
+        self.assertEqual(type(results), list)
42
+        self.assertEqual(len(results), 1)
43
+        self.assertEqual(results[0]['title'], 'Title of track')
44
+        self.assertEqual(results[0]['url'], 'http://www.deezer.com/track/1094042')
45
+        self.assertEqual(results[0]['content'], 'Artist Name • Album Title • Title of track')
46
+
47
+        json = """
48
+        {"data":[
49
+            {"id":200,"name":"Artist Name",
50
+            "link":"http:\/\/www.deezer.com\/artist\/1217","type":"artist"}
51
+        ]}
52
+        """
53
+        response = mock.Mock(text=json)
54
+        results = deezer.response(response)
55
+        self.assertEqual(type(results), list)
56
+        self.assertEqual(len(results), 0)

+ 1
- 0
searx/tests/test_engines.py 查看文件

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