summaryrefslogtreecommitdiff
path: root/tests/unit/engines/test_flickr_noapi.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/engines/test_flickr_noapi.py')
-rw-r--r--tests/unit/engines/test_flickr_noapi.py357
1 files changed, 357 insertions, 0 deletions
diff --git a/tests/unit/engines/test_flickr_noapi.py b/tests/unit/engines/test_flickr_noapi.py
new file mode 100644
index 000000000..67699f2f0
--- /dev/null
+++ b/tests/unit/engines/test_flickr_noapi.py
@@ -0,0 +1,357 @@
+from collections import defaultdict
+import mock
+from searx.engines import flickr_noapi
+from searx.testing import SearxTestCase
+
+
+class TestFlickrNoapiEngine(SearxTestCase):
+
+ def test_build_flickr_url(self):
+ url = flickr_noapi.build_flickr_url("uid", "pid")
+ self.assertIn("uid", url)
+ self.assertIn("pid", url)
+
+ def test_request(self):
+ query = 'test_query'
+ dicto = defaultdict(dict)
+ dicto['pageno'] = 1
+ dicto['time_range'] = ''
+ params = flickr_noapi.request(query, dicto)
+ self.assertIn('url', params)
+ self.assertIn(query, params['url'])
+ self.assertIn('flickr.com', params['url'])
+
+ def test_response(self):
+ self.assertRaises(AttributeError, flickr_noapi.response, None)
+ self.assertRaises(AttributeError, flickr_noapi.response, [])
+ self.assertRaises(AttributeError, flickr_noapi.response, '')
+ self.assertRaises(AttributeError, flickr_noapi.response, '[]')
+
+ response = mock.Mock(text='"modelExport:{"legend":[],"main":{"search-photos-lite-models":[{"photos":{}}]}}')
+ self.assertEqual(flickr_noapi.response(response), [])
+
+ response = \
+ mock.Mock(text='"modelExport:{"legend":[],"main":{"search-photos-lite-models":[{"photos":{"_data":[]}}]}}')
+ self.assertEqual(flickr_noapi.response(response), [])
+
+ # everthing is ok test
+ json = """
+ modelExport: {
+ "legend": [
+ [
+ "search-photos-lite-models",
+ "0",
+ "photos",
+ "_data",
+ "0"
+ ]
+ ],
+ "main": {
+ "search-photos-lite-models": [
+ {
+ "photos": {
+ "_data": [
+ {
+ "_flickrModelRegistry": "photo-lite-models",
+ "title": "This%20is%20the%20title",
+ "username": "Owner",
+ "pathAlias": "klink692",
+ "realname": "Owner",
+ "license": 0,
+ "ownerNsid": "59729010@N00",
+ "canComment": false,
+ "commentCount": 14,
+ "faveCount": 21,
+ "id": "14001294434",
+ "sizes": {
+ "c": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_c.jpg",
+ "width": 541,
+ "height": 800,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_c.jpg",
+ "key": "c"
+ },
+ "h": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_761d32237a_h.jpg",
+ "width": 1081,
+ "height": 1600,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_761d32237a_h.jpg",
+ "key": "h"
+ },
+ "k": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_f145a2c11a_k.jpg",
+ "width": 1383,
+ "height": 2048,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_f145a2c11a_k.jpg",
+ "key": "k"
+ },
+ "l": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_b.jpg",
+ "width": 692,
+ "height": 1024,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_b.jpg",
+ "key": "l"
+ },
+ "m": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777.jpg",
+ "width": 338,
+ "height": 500,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777.jpg",
+ "key": "m"
+ },
+ "n": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_n.jpg",
+ "width": 216,
+ "height": 320,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_n.jpg",
+ "key": "n"
+ },
+ "q": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_q.jpg",
+ "width": 150,
+ "height": 150,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_q.jpg",
+ "key": "q"
+ },
+ "s": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_m.jpg",
+ "width": 162,
+ "height": 240,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_m.jpg",
+ "key": "s"
+ },
+ "sq": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_s.jpg",
+ "width": 75,
+ "height": 75,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_s.jpg",
+ "key": "sq"
+ },
+ "t": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_t.jpg",
+ "width": 68,
+ "height": 100,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_t.jpg",
+ "key": "t"
+ },
+ "z": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_z.jpg",
+ "width": 433,
+ "height": 640,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_z.jpg",
+ "key": "z"
+ }
+ }
+ }
+ ]
+ }
+ }
+ ]
+ }
+ }
+ """
+ # Flickr serves search results in a json block named 'modelExport' buried inside a script tag,
+ # this json is served as a single line terminating with a comma.
+ json = ''.join(json.split()) + ',\n'
+ response = mock.Mock(text=json)
+ results = flickr_noapi.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 1)
+ self.assertEqual(results[0]['title'], 'This is the title')
+ self.assertEqual(results[0]['url'], 'https://www.flickr.com/photos/59729010@N00/14001294434')
+ self.assertIn('k.jpg', results[0]['img_src'])
+ self.assertIn('n.jpg', results[0]['thumbnail_src'])
+ self.assertIn('Owner', results[0]['author'])
+
+ # no n size, only the z size
+ json = """
+ modelExport: {
+ "legend": [
+ [
+ "search-photos-lite-models",
+ "0",
+ "photos",
+ "_data",
+ "0"
+ ]
+ ],
+ "main": {
+ "search-photos-lite-models": [
+ {
+ "photos": {
+ "_data": [
+ {
+ "_flickrModelRegistry": "photo-lite-models",
+ "title": "This%20is%20the%20title",
+ "username": "Owner",
+ "pathAlias": "klink692",
+ "realname": "Owner",
+ "license": 0,
+ "ownerNsid": "59729010@N00",
+ "canComment": false,
+ "commentCount": 14,
+ "faveCount": 21,
+ "id": "14001294434",
+ "sizes": {
+ "z": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_z.jpg",
+ "width": 433,
+ "height": 640,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_z.jpg",
+ "key": "z"
+ }
+ }
+ }
+ ]
+ }
+ }
+ ]
+ }
+ }
+ """
+ json = ''.join(json.split()) + ',\n'
+ response = mock.Mock(text=json)
+ results = flickr_noapi.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 1)
+ self.assertEqual(results[0]['title'], 'This is the title')
+ self.assertEqual(results[0]['url'], 'https://www.flickr.com/photos/59729010@N00/14001294434')
+ self.assertIn('z.jpg', results[0]['img_src'])
+ self.assertIn('z.jpg', results[0]['thumbnail_src'])
+ self.assertIn('Owner', results[0]['author'])
+
+ # no z or n size
+ json = """
+ modelExport: {
+ "legend": [
+ [
+ "search-photos-lite-models",
+ "0",
+ "photos",
+ "_data",
+ "0"
+ ]
+ ],
+ "main": {
+ "search-photos-lite-models": [
+ {
+ "photos": {
+ "_data": [
+ {
+ "_flickrModelRegistry": "photo-lite-models",
+ "title": "This%20is%20the%20title",
+ "username": "Owner",
+ "pathAlias": "klink692",
+ "realname": "Owner",
+ "license": 0,
+ "ownerNsid": "59729010@N00",
+ "canComment": false,
+ "commentCount": 14,
+ "faveCount": 21,
+ "id": "14001294434",
+ "sizes": {
+ "o": {
+ "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_o.jpg",
+ "width": 433,
+ "height": 640,
+ "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_o.jpg",
+ "key": "o"
+ }
+ }
+ }
+ ]
+ }
+ }
+ ]
+ }
+ }
+ """
+ json = ''.join(json.split()) + ',\n'
+ response = mock.Mock(text=json)
+ results = flickr_noapi.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 1)
+ self.assertEqual(results[0]['title'], 'This is the title')
+ self.assertEqual(results[0]['url'], 'https://www.flickr.com/photos/59729010@N00/14001294434')
+ self.assertIn('o.jpg', results[0]['img_src'])
+ self.assertIn('o.jpg', results[0]['thumbnail_src'])
+ self.assertIn('Owner', results[0]['author'])
+
+ # no image test
+ json = """
+ modelExport: {
+ "legend": [
+ [
+ "search-photos-lite-models",
+ "0",
+ "photos",
+ "_data",
+ "0"
+ ]
+ ],
+ "main": {
+ "search-photos-lite-models": [
+ {
+ "photos": {
+ "_data": [
+ {
+ "_flickrModelRegistry": "photo-lite-models",
+ "title": "This is the title",
+ "username": "Owner",
+ "pathAlias": "klink692",
+ "realname": "Owner",
+ "license": 0,
+ "ownerNsid": "59729010@N00",
+ "canComment": false,
+ "commentCount": 14,
+ "faveCount": 21,
+ "id": "14001294434",
+ "sizes": {
+ }
+ }
+ ]
+ }
+ }
+ ]
+ }
+ }
+ """
+ json = ''.join(json.split()) + ',\n'
+ response = mock.Mock(text=json)
+ results = flickr_noapi.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 0)
+
+ # null test
+ json = """
+ modelExport: {
+ "legend": [null],
+ "main": {
+ "search-photos-lite-models": [
+ {
+ "photos": {
+ "_data": [null]
+ }
+ }
+ ]
+ }
+ }
+ """
+ json = ''.join(json.split()) + ',\n'
+ response = mock.Mock(text=json)
+ results = flickr_noapi.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 0)
+
+ # garbage test
+ json = r"""
+ {"toto":[
+ {"id":200,"name":"Artist Name",
+ "link":"http:\/\/www.flickr.com\/artist\/1217","type":"artist"}
+ ]}
+ """
+ json = ''.join(json.split()) + ',\n'
+ response = mock.Mock(text=json)
+ results = flickr_noapi.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 0)
generated by cgit v1.2.3 (git 2.25.1) at 2026-03-22 22:55:27 +0000