logo

searx

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

test_subtitleseeker.py (5993B)

    

  1. from collections import defaultdict

  2. import mock

  3. from searx.engines import subtitleseeker

  4. from searx.testing import SearxTestCase

  5. 

  6. 

  7. class TestSubtitleseekerEngine(SearxTestCase):

  8. 

  9.     def test_request(self):

  10.         query = 'test_query'

  11.         dicto = defaultdict(dict)

  12.         dicto['pageno'] = 1

  13.         dicto['language'] = 'fr-FR'

  14.         params = subtitleseeker.request(query, dicto)

  15.         self.assertTrue('url' in params)

  16.         self.assertTrue(query in params['url'])

  17.         self.assertTrue('subtitleseeker.com' in params['url'])

  18. 

  19.     def test_response(self):

  20.         dicto = defaultdict(dict)

  21.         dicto['language'] = 'fr-FR'

  22.         response = mock.Mock(search_params=dicto)

  23. 

  24.         self.assertRaises(AttributeError, subtitleseeker.response, None)

  25.         self.assertRaises(AttributeError, subtitleseeker.response, [])

  26.         self.assertRaises(AttributeError, subtitleseeker.response, '')

  27.         self.assertRaises(AttributeError, subtitleseeker.response, '[]')

  28. 

  29.         response = mock.Mock(text='<html></html>', search_params=dicto)

  30.         self.assertEqual(subtitleseeker.response(response), [])

  31. 

  32.         html = """

  33.         <div class="boxRows">

  34.             <div class="boxRowsInner" style="width:600px;">

  35.                 <img src="http://static.subtitleseeker.com/images/movie.gif"

  36.                     style="width:16px; height:16px;" class="icon">

  37.                 <a href="http://this.is.the.url/"

  38.                     class="blue" title="Title subtitle" >

  39.                     This is the Title

  40.                 </a>

  41.                 <br><br>

  42.                 <span class="f10b grey-dark arial" style="padding:0px 0px 5px 20px">

  43.                     "Alternative Title"

  44.                 </span>

  45.             </div>

  46.             <div class="boxRowsInner f12b red" style="width:70px;">

  47.                 1998

  48.             </div>

  49.             <div class="boxRowsInner grey-web f12" style="width:120px;">

  50.                 <img src="http://static.subtitleseeker.com/images/basket_put.png"

  51.                     style="width:16px; height:16px;" class="icon">

  52.                 1039 Subs

  53.             </div>

  54.             <div class="boxRowsInner grey-web f10" style="width:130px;">

  55.                 <img src="http://static.subtitleseeker.com/images/arrow_refresh_small.png"

  56.                     style="width:16px; height:16px;" class="icon">

  57.                 1 hours ago

  58.             </div>

  59.             <div class="clear"></div>

  60.         </div>

  61.         """

  62.         response = mock.Mock(text=html, search_params=dicto)

  63.         results = subtitleseeker.response(response)

  64.         self.assertEqual(type(results), list)

  65.         self.assertEqual(len(results), 1)

  66.         self.assertEqual(results[0]['title'], 'This is the Title')

  67.         self.assertEqual(results[0]['url'], 'http://this.is.the.url/French/')

  68.         self.assertIn('1998', results[0]['content'])

  69.         self.assertIn('1039 Subs', results[0]['content'])

  70.         self.assertIn('Alternative Title', results[0]['content'])

  71. 

  72.         dicto['language'] = 'pt-BR'

  73.         results = subtitleseeker.response(response)

  74.         self.assertEqual(results[0]['url'], 'http://this.is.the.url/Brazilian/')

  75. 

  76.         html = """

  77.         <div class="boxRows">

  78.             <div class="boxRowsInner" style="width:600px;">

  79.                 <img src="http://static.subtitleseeker.com/images/movie.gif"

  80.                     style="width:16px; height:16px;" class="icon">

  81.                 <a href="http://this.is.the.url/"

  82.                     class="blue" title="Title subtitle" >

  83.                     This is the Title

  84.                 </a>

  85.             </div>

  86.             <div class="boxRowsInner f12b red" style="width:70px;">

  87.                 1998

  88.             </div>

  89.             <div class="boxRowsInner grey-web f12" style="width:120px;">

  90.                 <img src="http://static.subtitleseeker.com/images/basket_put.png"

  91.                     style="width:16px; height:16px;" class="icon">

  92.                 1039 Subs

  93.             </div>

  94.             <div class="boxRowsInner grey-web f10" style="width:130px;">

  95.                 <img src="http://static.subtitleseeker.com/images/arrow_refresh_small.png"

  96.                     style="width:16px; height:16px;" class="icon">

  97.                 1 hours ago

  98.             </div>

  99.             <div class="clear"></div>

  100.         </div>

  101.         """

  102.         subtitleseeker.language = 'English'

  103.         response = mock.Mock(text=html, search_params=dicto)

  104.         results = subtitleseeker.response(response)

  105.         self.assertEqual(type(results), list)

  106.         self.assertEqual(len(results), 1)

  107.         self.assertEqual(results[0]['title'], 'This is the Title')

  108.         self.assertEqual(results[0]['url'], 'http://this.is.the.url/English/')

  109.         self.assertIn('1998', results[0]['content'])

  110.         self.assertIn('1039 Subs', results[0]['content'])

  111. 

  112.         html = """

  113.         <div class="boxRowsInner" style="width:600px;">

  114.             <img src="http://static.subtitleseeker.com/images/movie.gif"

  115.                 style="width:16px; height:16px;" class="icon">

  116.             <a href="http://this.is.the.url/"

  117.                 class="blue" title="Title subtitle" >

  118.                 This is the Title

  119.             </a>

  120.         </div>

  121.         <div class="boxRowsInner f12b red" style="width:70px;">

  122.             1998

  123.         </div>

  124.         <div class="boxRowsInner grey-web f12" style="width:120px;">

  125.             <img src="http://static.subtitleseeker.com/images/basket_put.png"

  126.                 style="width:16px; height:16px;" class="icon">

  127.             1039 Subs

  128.         </div>

  129.         <div class="boxRowsInner grey-web f10" style="width:130px;">

  130.             <img src="http://static.subtitleseeker.com/images/arrow_refresh_small.png"

  131.                 style="width:16px; height:16px;" class="icon">

  132.             1 hours ago

  133.         </div>

  134.         """

  135.         response = mock.Mock(text=html, search_params=dicto)

  136.         results = subtitleseeker.response(response)

  137.         self.assertEqual(type(results), list)

  138.         self.assertEqual(len(results), 0)