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

  6. 

  7. 

  8. class TestSeedPeerEngine(SearxTestCase):

  9. 

  10.     html = ''

  11.     with open('./tests/unit/engines/seedpeer_fixture.html') as fixture:

  12.         html += fixture.read()

  13. 

  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'])

  22. 

  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, '[]')

  28. 

  29.     def test_response_returns_empty_list(self):

  30.         response = mock.Mock(text='<html></html>')

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

  32. 

  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)

  38. 

  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')