logo

searx

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

test_reddit.py (2520B)

    

  1. from collections import defaultdict
  2. import mock
  3. from searx.engines import reddit
  4. from searx.testing import SearxTestCase
  5. from datetime import datetime
  8. class TestRedditEngine(SearxTestCase):
  10.     def test_request(self):
  11.         query = 'test_query'
  12.         dic = defaultdict(dict)
  13.         params = reddit.request(query, dic)
  14.         self.assertTrue('url' in params)
  15.         self.assertTrue(query in params['url'])
  16.         self.assertTrue('reddit.com' in params['url'])
  18.     def test_response(self):
  19.         resp = mock.Mock(text='{}')
  20.         self.assertEqual(reddit.response(resp), [])
  22.         json = """
  23.         {
  24.             "kind": "Listing",
  25.             "data": {
  26.                 "children": [{
  27.                     "data": {
  28.                         "url": "http://google2.com/",
  29.                         "permalink": "http://google.com/",
  30.                         "title": "Title number one",
  31.                         "selftext": "Sample",
  32.                         "created_utc": 1401219957.0,
  33.                         "thumbnail": "http://image.com/picture.jpg"
  34.                     }
  35.                 }, {
  36.                     "data": {
  37.                         "url": "https://reddit2.com/",
  38.                         "permalink": "https://reddit.com/",
  39.                         "title": "Title number two",
  40.                         "selftext": "Dominus vobiscum",
  41.                         "created_utc": 1438792533.0,
  42.                         "thumbnail": "self"
  43.                     }
  44.                 }]
  45.             }
  46.         }
  47.         """
  49.         resp = mock.Mock(text=json)
  50.         results = reddit.response(resp)
  52.         self.assertEqual(len(results), 2)
  53.         self.assertEqual(type(results), list)
  55.         # testing first result (picture)
  56.         r = results[0]
  57.         self.assertEqual(r['url'], 'http://google.com/')
  58.         self.assertEqual(r['title'], 'Title number one')
  59.         self.assertEqual(r['template'], 'images.html')
  60.         self.assertEqual(r['img_src'], 'http://google2.com/')
  61.         self.assertEqual(r['thumbnail_src'], 'http://image.com/picture.jpg')
  63.         # testing second result (self-post)
  64.         r = results[1]
  65.         self.assertEqual(r['url'], 'https://reddit.com/')
  66.         self.assertEqual(r['title'], 'Title number two')
  67.         self.assertEqual(r['content'], 'Dominus vobiscum')
  68.         created = datetime.fromtimestamp(1438792533.0)
  69.         self.assertEqual(r['publishedDate'], created)
  70.         self.assertTrue('thumbnail_src' not in r)
  71.         self.assertTrue('img_src' not in r)