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
  7. class TestSubtitleseekerEngine(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 = 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'])
  19.     def test_response(self):
  20.         dicto = defaultdict(dict)
  21.         dicto['language'] = 'fr-FR'
  22.         response = mock.Mock(search_params=dicto)
  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, '[]')
  29.         response = mock.Mock(text='<html></html>', search_params=dicto)
  30.         self.assertEqual(subtitleseeker.response(response), [])
  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'])
  72.         dicto['language'] = 'pt-BR'
  73.         results = subtitleseeker.response(response)
  74.         self.assertEqual(results[0]['url'], 'http://this.is.the.url/Brazilian/')
  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'])
  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)