@@ -0,0 +1,108 @@
1+from collections import defaultdict
2+import mock
3+from searx.engines import google_images
4+from searx.testing import SearxTestCase
5+
6+
7+class TestGoogleImagesEngine(SearxTestCase):
8+
9+ def test_request(self):
10+ query = 'test_query'
11+ dicto = defaultdict(dict)
12+ dicto['pageno'] = 1
13+ params = google_images.request(query, dicto)
14+ self.assertTrue('url' in params)
15+ self.assertTrue(query in params['url'])
16+ self.assertTrue('googleapis.com' in params['url'])
17+
18+ def test_response(self):
19+ self.assertRaises(AttributeError, google_images.response, None)
20+ self.assertRaises(AttributeError, google_images.response, [])
21+ self.assertRaises(AttributeError, google_images.response, '')
22+ self.assertRaises(AttributeError, google_images.response, '[]')
23+
24+ response = mock.Mock(text='{}')
25+ self.assertEqual(google_images.response(response), [])
26+
27+ response = mock.Mock(text='{"data": []}')
28+ self.assertEqual(google_images.response(response), [])
29+
30+ json = """
31+ {
32+ "responseData": {
33+ "results": [
34+ {
35+ "GsearchResultClass": "GimageSearch",
36+ "width": "400",
37+ "height": "400",
38+ "imageId": "ANd9GcQbYb9FJuAbG_hT4i8FeC0O0x-P--EHdzgRIF9ao97nHLl7C2mREn6qTQ",
39+ "tbWidth": "124",
40+ "tbHeight": "124",
41+ "unescapedUrl": "http://unescaped.url.jpg",
42+ "url": "http://image.url.jpg",
43+ "visibleUrl": "insolitebuzz.fr",
44+ "title": "This is the title",
45+ "titleNoFormatting": "Petit test sympa qui rend fou tout le monde ! A faire",
46+ "originalContextUrl": "http://this.is.the.url",
47+ "content": "<b>test</b>",
48+ "contentNoFormatting": "test",
49+ "tbUrl": "http://thumbnail.url"
50+ }
51+ ]
52+ },
53+ "responseDetails": null,
54+ "responseStatus": 200
55+ }
56+ """
57+ response = mock.Mock(text=json)
58+ results = google_images.response(response)
59+ self.assertEqual(type(results), list)
60+ self.assertEqual(len(results), 1)
61+ self.assertEqual(results[0]['title'], 'This is the title')
62+ self.assertEqual(results[0]['url'], 'http://this.is.the.url')
63+ self.assertEqual(results[0]['thumbnail_src'], 'http://thumbnail.url')
64+ self.assertEqual(results[0]['img_src'], 'http://image.url.jpg')
65+ self.assertEqual(results[0]['content'], '<b>test</b>')
66+
67+ json = """
68+ {
69+ "responseData": {
70+ "results": [
71+ {
72+ "GsearchResultClass": "GimageSearch",
73+ "width": "400",
74+ "height": "400",
75+ "imageId": "ANd9GcQbYb9FJuAbG_hT4i8FeC0O0x-P--EHdzgRIF9ao97nHLl7C2mREn6qTQ",
76+ "tbWidth": "124",
77+ "tbHeight": "124",
78+ "unescapedUrl": "http://unescaped.url.jpg",
79+ "visibleUrl": "insolitebuzz.fr",
80+ "title": "This is the title",
81+ "titleNoFormatting": "Petit test sympa qui rend fou tout le monde ! A faire",
82+ "originalContextUrl": "http://this.is.the.url",
83+ "content": "<b>test</b>",
84+ "contentNoFormatting": "test",
85+ "tbUrl": "http://thumbnail.url"
86+ }
87+ ]
88+ },
89+ "responseDetails": null,
90+ "responseStatus": 200
91+ }
92+ """
93+ response = mock.Mock(text=json)
94+ results = google_images.response(response)
95+ self.assertEqual(type(results), list)
96+ self.assertEqual(len(results), 0)
97+
98+ json = """
99+ {
100+ "responseData": {},
101+ "responseDetails": null,
102+ "responseStatus": 200
103+ }
104+ """
105+ response = mock.Mock(text=json)
106+ results = google_images.response(response)
107+ self.assertEqual(type(results), list)
108+ self.assertEqual(len(results), 0)