logo

searx

My custom branche(s) on searx, a meta-search engine git clone https://hacktivis.me/git/searx.git

test_flickr_noapi.py (12259B)

    

  1. from collections import defaultdict

  2. import mock

  3. from searx.engines import flickr_noapi

  4. from searx.testing import SearxTestCase

  5. 

  6. 

  7. class TestFlickrNoapiEngine(SearxTestCase):

  8. 

  9.     def test_build_flickr_url(self):

  10.         url = flickr_noapi.build_flickr_url("uid", "pid")

  11.         self.assertIn("uid", url)

  12.         self.assertIn("pid", url)

  13. 

  14.     def test_request(self):

  15.         query = 'test_query'

  16.         dicto = defaultdict(dict)

  17.         dicto['pageno'] = 1

  18.         dicto['time_range'] = ''

  19.         params = flickr_noapi.request(query, dicto)

  20.         self.assertIn('url', params)

  21.         self.assertIn(query, params['url'])

  22.         self.assertIn('flickr.com', params['url'])

  23. 

  24.     def test_response(self):

  25.         self.assertRaises(AttributeError, flickr_noapi.response, None)

  26.         self.assertRaises(AttributeError, flickr_noapi.response, [])

  27.         self.assertRaises(AttributeError, flickr_noapi.response, '')

  28.         self.assertRaises(AttributeError, flickr_noapi.response, '[]')

  29. 

  30.         response = mock.Mock(text='"search-photos-lite-models","photos":{},"totalItems":')

  31.         self.assertEqual(flickr_noapi.response(response), [])

  32. 

  33.         response = mock.Mock(text='search-photos-lite-models","photos":{"data": []},"totalItems":')

  34.         self.assertEqual(flickr_noapi.response(response), [])

  35. 

  36.         # everthing is ok test

  37.         json = """

  38.         "search-photos-lite-models","photos":

  39.         {

  40.           "_data": [

  41.             {

  42.               "_flickrModelRegistry": "photo-lite-models",

  43.               "title": "This is the title",

  44.               "username": "Owner",

  45.               "pathAlias": "klink692",

  46.               "realname": "Owner",

  47.               "license": 0,

  48.               "ownerNsid": "59729010@N00",

  49.               "canComment": false,

  50.               "commentCount": 14,

  51.               "faveCount": 21,

  52.               "id": "14001294434",

  53.               "sizes": {

  54.                 "c": {

  55.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_c.jpg",

  56.                   "width": 541,

  57.                   "height": 800,

  58.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_c.jpg",

  59.                   "key": "c"

  60.                 },

  61.                 "h": {

  62.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_761d32237a_h.jpg",

  63.                   "width": 1081,

  64.                   "height": 1600,

  65.                   "url": "//c4.staticflickr.com/8/7246/14001294434_761d32237a_h.jpg",

  66.                   "key": "h"

  67.                 },

  68.                 "k": {

  69.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_f145a2c11a_k.jpg",

  70.                   "width": 1383,

  71.                   "height": 2048,

  72.                   "url": "//c4.staticflickr.com/8/7246/14001294434_f145a2c11a_k.jpg",

  73.                   "key": "k"

  74.                 },

  75.                 "l": {

  76.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_b.jpg",

  77.                   "width": 692,

  78.                   "height": 1024,

  79.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_b.jpg",

  80.                   "key": "l"

  81.                 },

  82.                 "m": {

  83.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777.jpg",

  84.                   "width": 338,

  85.                   "height": 500,

  86.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777.jpg",

  87.                   "key": "m"

  88.                 },

  89.                 "n": {

  90.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_n.jpg",

  91.                   "width": 216,

  92.                   "height": 320,

  93.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_n.jpg",

  94.                   "key": "n"

  95.                 },

  96.                 "q": {

  97.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_q.jpg",

  98.                   "width": 150,

  99.                   "height": 150,

  100.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_q.jpg",

  101.                   "key": "q"

  102.                 },

  103.                 "s": {

  104.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_m.jpg",

  105.                   "width": 162,

  106.                   "height": 240,

  107.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_m.jpg",

  108.                   "key": "s"

  109.                 },

  110.                 "sq": {

  111.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_s.jpg",

  112.                   "width": 75,

  113.                   "height": 75,

  114.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_s.jpg",

  115.                   "key": "sq"

  116.                 },

  117.                 "t": {

  118.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_t.jpg",

  119.                   "width": 68,

  120.                   "height": 100,

  121.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_t.jpg",

  122.                   "key": "t"

  123.                 },

  124.                 "z": {

  125.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_z.jpg",

  126.                   "width": 433,

  127.                   "height": 640,

  128.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_z.jpg",

  129.                   "key": "z"

  130.                 }

  131.               }

  132.             }

  133.           ],

  134.           "fetchedStart": true,

  135.           "fetchedEnd": false,

  136.           "totalItems": "4386039"

  137.         },"totalItems":

  138.         """

  139.         json = json.replace('\r\n', '').replace('\n', '').replace('\r', '')

  140.         response = mock.Mock(text=json)

  141.         results = flickr_noapi.response(response)

  142.         self.assertEqual(type(results), list)

  143.         self.assertEqual(len(results), 1)

  144.         self.assertEqual(results[0]['title'], 'This is the title')

  145.         self.assertEqual(results[0]['url'], 'https://www.flickr.com/photos/59729010@N00/14001294434')

  146.         self.assertIn('k.jpg', results[0]['img_src'])

  147.         self.assertIn('n.jpg', results[0]['thumbnail_src'])

  148.         self.assertIn('Owner', results[0]['author'])

  149. 

  150.         # no n size, only the z size

  151.         json = """

  152.         "search-photos-lite-models","photos":

  153.         {

  154.           "_data": [

  155.             {

  156.               "_flickrModelRegistry": "photo-lite-models",

  157.               "title": "This is the title",

  158.               "username": "Owner",

  159.               "pathAlias": "klink692",

  160.               "realname": "Owner",

  161.               "license": 0,

  162.               "ownerNsid": "59729010@N00",

  163.               "canComment": false,

  164.               "commentCount": 14,

  165.               "faveCount": 21,

  166.               "id": "14001294434",

  167.               "sizes": {

  168.                 "z": {

  169.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_z.jpg",

  170.                   "width": 433,

  171.                   "height": 640,

  172.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_z.jpg",

  173.                   "key": "z"

  174.                 }

  175.               }

  176.             }

  177.           ],

  178.           "fetchedStart": true,

  179.           "fetchedEnd": false,

  180.           "totalItems": "4386039"

  181.         },"totalItems":

  182.         """

  183.         response = mock.Mock(text=json)

  184.         results = flickr_noapi.response(response)

  185.         self.assertEqual(type(results), list)

  186.         self.assertEqual(len(results), 1)

  187.         self.assertEqual(results[0]['title'], 'This is the title')

  188.         self.assertEqual(results[0]['url'], 'https://www.flickr.com/photos/59729010@N00/14001294434')

  189.         self.assertIn('z.jpg', results[0]['img_src'])

  190.         self.assertIn('z.jpg', results[0]['thumbnail_src'])

  191.         self.assertIn('Owner', results[0]['author'])

  192. 

  193.         # no z or n size

  194.         json = """

  195.         "search-photos-lite-models","photos":

  196.         {

  197.           "_data": [

  198.             {

  199.               "_flickrModelRegistry": "photo-lite-models",

  200.               "title": "This is the title",

  201.               "username": "Owner",

  202.               "pathAlias": "klink692",

  203.               "realname": "Owner",

  204.               "license": 0,

  205.               "ownerNsid": "59729010@N00",

  206.               "canComment": false,

  207.               "commentCount": 14,

  208.               "faveCount": 21,

  209.               "id": "14001294434",

  210.               "sizes": {

  211.                 "o": {

  212.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_o.jpg",

  213.                   "width": 433,

  214.                   "height": 640,

  215.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_o.jpg",

  216.                   "key": "o"

  217.                 }

  218.               }

  219.             }

  220.           ],

  221.           "fetchedStart": true,

  222.           "fetchedEnd": false,

  223.           "totalItems": "4386039"

  224.         },"totalItems":

  225.         """

  226.         response = mock.Mock(text=json)

  227.         results = flickr_noapi.response(response)

  228.         self.assertEqual(type(results), list)

  229.         self.assertEqual(len(results), 1)

  230.         self.assertEqual(results[0]['title'], 'This is the title')

  231.         self.assertEqual(results[0]['url'], 'https://www.flickr.com/photos/59729010@N00/14001294434')

  232.         self.assertIn('o.jpg', results[0]['img_src'])

  233.         self.assertIn('o.jpg', results[0]['thumbnail_src'])

  234.         self.assertIn('Owner', results[0]['author'])

  235. 

  236.         # no image test

  237.         json = """

  238.         "search-photos-lite-models","photos":

  239.         {

  240.           "_data": [

  241.             {

  242.               "_flickrModelRegistry": "photo-lite-models",

  243.               "title": "This is the title",

  244.               "username": "Owner",

  245.               "pathAlias": "klink692",

  246.               "realname": "Owner",

  247.               "license": 0,

  248.               "ownerNsid": "59729010@N00",

  249.               "canComment": false,

  250.               "commentCount": 14,

  251.               "faveCount": 21,

  252.               "id": "14001294434",

  253.               "sizes": {

  254.               }

  255.             }

  256.           ],

  257.           "fetchedStart": true,

  258.           "fetchedEnd": false,

  259.           "totalItems": "4386039"

  260.         },"totalItems":

  261.         """

  262.         response = mock.Mock(text=json)

  263.         results = flickr_noapi.response(response)

  264.         self.assertEqual(type(results), list)

  265.         self.assertEqual(len(results), 0)

  266. 

  267.         # null test

  268.         json = """

  269.         "search-photos-models","photos":

  270.         {

  271.           "_data": [null],

  272.           "fetchedStart": true,

  273.           "fetchedEnd": false,

  274.           "totalItems": "4386039"

  275.         },"totalItems":

  276.         """

  277.         response = mock.Mock(text=json)

  278.         results = flickr_noapi.response(response)

  279.         self.assertEqual(type(results), list)

  280.         self.assertEqual(len(results), 0)

  281. 

  282.         # no ownerNsid test

  283.         json = """

  284.         "search-photos-lite-models","photos":

  285.         {

  286.           "_data": [

  287.             {

  288.               "_flickrModelRegistry": "photo-lite-models",

  289.               "title": "This is the title",

  290.               "username": "Owner",

  291.               "pathAlias": "klink692",

  292.               "realname": "Owner",

  293.               "license": 0,

  294.               "canComment": false,

  295.               "commentCount": 14,

  296.               "faveCount": 21,

  297.               "id": "14001294434",

  298.               "sizes": {

  299.                 "o": {

  300.                   "displayUrl": "//farm8.staticflickr.com/7246/14001294434_410f653777_o.jpg",

  301.                   "width": 433,

  302.                   "height": 640,

  303.                   "url": "//c4.staticflickr.com/8/7246/14001294434_410f653777_o.jpg",

  304.                   "key": "o"

  305.                 }

  306.               }

  307.             }

  308.           ],

  309.           "fetchedStart": true,

  310.           "fetchedEnd": false,

  311.           "totalItems": "4386039"

  312.         },"totalItems":

  313.         """

  314.         response = mock.Mock(text=json)

  315.         results = flickr_noapi.response(response)

  316.         self.assertEqual(type(results), list)

  317.         self.assertEqual(len(results), 0)

  318. 

  319.         # garbage test

  320.         json = r"""

  321.         {"toto":[

  322.             {"id":200,"name":"Artist Name",

  323.             "link":"http:\/\/www.flickr.com\/artist\/1217","type":"artist"}

  324.         ]}

  325.         """

  326.         response = mock.Mock(text=json)

  327.         results = flickr_noapi.response(response)

  328.         self.assertEqual(type(results), list)

  329.         self.assertEqual(len(results), 0)