logo

searx

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

test_yacy.py (3720B)

    

  1. from collections import defaultdict
  2. import mock
  3. from searx.engines import yacy
  4. from searx.testing import SearxTestCase
  7. class TestYacyEngine(SearxTestCase):
  9.     def test_request(self):
  10.         query = 'test_query'
  11.         dicto = defaultdict(dict)
  12.         dicto['pageno'] = 1
  13.         dicto['language'] = 'fr_FR'
  14.         params = yacy.request(query, dicto)
  15.         self.assertIn('url', params)
  16.         self.assertIn(query, params['url'])
  17.         self.assertIn('localhost', params['url'])
  18.         self.assertIn('fr', params['url'])
  20.     def test_response(self):
  21.         self.assertRaises(AttributeError, yacy.response, None)
  22.         self.assertRaises(AttributeError, yacy.response, [])
  23.         self.assertRaises(AttributeError, yacy.response, '')
  24.         self.assertRaises(AttributeError, yacy.response, '[]')
  26.         response = mock.Mock(text='{}')
  27.         self.assertEqual(yacy.response(response), [])
  29.         response = mock.Mock(text='{"data": []}')
  30.         self.assertEqual(yacy.response(response), [])
  32.         json = """
  33.         {
  34.           "channels": [
  35.             {
  36.               "title": "YaCy P2P-Search for test",
  37.               "description": "Search for test",
  38.               "link": "http://search.yacy.de:7001/yacysearch.html?query=test&resource=global&contentdom=0",
  39.               "image": {
  40.                 "url": "http://search.yacy.de:7001/env/grafics/yacy.png",
  41.                 "title": "Search for test",
  42.                 "link": "http://search.yacy.de:7001/yacysearch.html?query=test&resource=global&contentdom=0"
  43.               },
  44.               "totalResults": "249",
  45.               "startIndex": "0",
  46.               "itemsPerPage": "5",
  47.               "searchTerms": "test",
  48.               "items": [
  49.                 {
  50.                   "title": "This is the title",
  51.                   "link": "http://this.is.the.url",
  52.                   "code": "",
  53.                   "description": "This should be the content",
  54.                   "pubDate": "Sat, 08 Jun 2013 02:00:00 +0200",
  55.                   "size": "44213",
  56.                   "sizename": "43 kbyte",
  57.                   "guid": "lzh_1T_5FP-A",
  58.                   "faviconCode": "XTS4uQ_5FP-A",
  59.                   "host": "www.gamestar.de",
  60.                   "path": "/spiele/city-of-heroes-freedom/47019.html",
  61.                   "file": "47019.html",
  62.                   "urlhash": "lzh_1T_5FP-A",
  63.                   "ranking": "0.20106804"
  64.                 },
  65.                 {
  66.                   "title": "This is the title2",
  67.                   "icon": "/ViewImage.png?maxwidth=96&maxheight=96&code=7EbAbW6BpPOA",
  68.                   "image": "http://image.url/image.png",
  69.                   "cache": "/ViewImage.png?quadratic=&url=http://golem.ivwbox.de/cgi-bin/ivw/CP/G_INET?d=14071378",
  70.                   "url": "http://this.is.the.url",
  71.                   "urlhash": "7EbAbW6BpPOA",
  72.                   "host": "www.golem.de",
  73.                   "width": "-1",
  74.                   "height": "-1"
  75.                 }
  76.               ]
  77.             }
  78.           ]
  79.         }
  80.         """
  81.         response = mock.Mock(text=json)
  82.         results = yacy.response(response)
  83.         self.assertEqual(type(results), list)
  84.         self.assertEqual(len(results), 2)
  85.         self.assertEqual(results[0]['title'], 'This is the title')
  86.         self.assertEqual(results[0]['url'], 'http://this.is.the.url')
  87.         self.assertEqual(results[0]['content'], 'This should be the content')
  88.         self.assertEqual(results[1]['img_src'], 'http://image.url/image.png')
  89.         self.assertEqual(results[1]['content'], '')
  90.         self.assertEqual(results[1]['url'], 'http://this.is.the.url')
  91.         self.assertEqual(results[1]['title'], 'This is the title2')