logo

searx

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

test_seedpeer.py (1959B)

    

  1. import mock
  2. from collections import defaultdict
  3. from searx.engines import seedpeer
  4. from searx.testing import SearxTestCase
  5. from datetime import datetime
  8. class TestSeedPeerEngine(SearxTestCase):
  10.     html = ''
  11.     with open('./tests/unit/engines/seedpeer_fixture.html') as fixture:
  12.         html += fixture.read()
  14.     def test_request(self):
  15.         query = 'test_query'
  16.         dicto = defaultdict(dict)
  17.         dicto['pageno'] = 1
  18.         params = seedpeer.request(query, dicto)
  19.         self.assertIn('url', params)
  20.         self.assertIn(query, params['url'])
  21.         self.assertIn('seedpeer.eu', params['url'])
  23.     def test_response_raises_attr_error_on_empty_response(self):
  24.         self.assertRaises(AttributeError, seedpeer.response, None)
  25.         self.assertRaises(AttributeError, seedpeer.response, [])
  26.         self.assertRaises(AttributeError, seedpeer.response, '')
  27.         self.assertRaises(AttributeError, seedpeer.response, '[]')
  29.     def test_response_returns_empty_list(self):
  30.         response = mock.Mock(text='<html></html>')
  31.         self.assertEqual(seedpeer.response(response), [])
  33.     def test_response_returns_all_results(self):
  34.         response = mock.Mock(text=self.html)
  35.         results = seedpeer.response(response)
  36.         self.assertTrue(isinstance(results, list))
  37.         self.assertEqual(len(results), 2)
  39.     def test_response_returns_correct_results(self):
  40.         response = mock.Mock(text=self.html)
  41.         results = seedpeer.response(response)
  42.         self.assertEqual(
  43.             results[0]['title'], 'Narcos - Season 2 - 720p WEBRiP - x265 HEVC - ShAaNiG '
  44.         )
  45.         self.assertEqual(
  46.             results[0]['url'],
  47.             'http://www.seedpeer.eu/details/11685972/Narcos---Season-2---720p-WEBRiP---x265-HEVC---ShAaNiG.html'
  48.         )
  49.         self.assertEqual(results[0]['content'], '2.48 GB, 1 day')
  50.         self.assertEqual(results[0]['seed'], '861')
  51.         self.assertEqual(results[0]['leech'], '332')