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
  7. class TestFlickrNoapiEngine(SearxTestCase):
  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)
  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'])
  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, '[]')
  30.         response = mock.Mock(text='"search-photos-lite-models","photos":{},"totalItems":')
  31.         self.assertEqual(flickr_noapi.response(response), [])
  33.         response = mock.Mock(text='search-photos-lite-models","photos":{"data": []},"totalItems":')
  34.         self.assertEqual(flickr_noapi.response(response), [])
  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'])
  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'])
  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'])
  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)
  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)
  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)
  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)