logo

searx

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

test_dailymotion.py (4116B)

    

  1. # -*- coding: utf-8 -*-
  2. from collections import defaultdict
  3. import mock
  4. from searx.engines import dailymotion
  5. from searx.testing import SearxTestCase
  8. class TestDailymotionEngine(SearxTestCase):
  10.     def test_request(self):
  11.         dailymotion.supported_languages = ['en', 'fr']
  12.         query = 'test_query'
  13.         dicto = defaultdict(dict)
  14.         dicto['pageno'] = 0
  15.         dicto['language'] = 'fr-FR'
  16.         params = dailymotion.request(query, dicto)
  17.         self.assertTrue('url' in params)
  18.         self.assertTrue(query in params['url'])
  19.         self.assertTrue('dailymotion.com' in params['url'])
  20.         self.assertTrue('fr' in params['url'])
  22.     def test_response(self):
  23.         self.assertRaises(AttributeError, dailymotion.response, None)
  24.         self.assertRaises(AttributeError, dailymotion.response, [])
  25.         self.assertRaises(AttributeError, dailymotion.response, '')
  26.         self.assertRaises(AttributeError, dailymotion.response, '[]')
  28.         response = mock.Mock(text='{}')
  29.         self.assertEqual(dailymotion.response(response), [])
  31.         response = mock.Mock(text='{"data": []}')
  32.         self.assertEqual(dailymotion.response(response), [])
  34.         json = """
  35.         {
  36.         "page": 1,
  37.         "limit": 5,
  38.         "explicit": false,
  39.         "total": 289487,
  40.         "has_more": true,
  41.         "list": [
  42.             {
  43.             "created_time": 1422173451,
  44.             "title": "Title",
  45.             "description": "Description",
  46.             "duration": 81,
  47.             "url": "http://www.url",
  48.             "thumbnail_360_url": "http://thumbnail",
  49.             "id": "x2fit7q"
  50.             }
  51.         ]
  52.         }
  53.         """
  54.         response = mock.Mock(text=json)
  55.         results = dailymotion.response(response)
  56.         self.assertEqual(type(results), list)
  57.         self.assertEqual(len(results), 1)
  58.         self.assertEqual(results[0]['title'], 'Title')
  59.         self.assertEqual(results[0]['url'], 'http://www.url')
  60.         self.assertEqual(results[0]['content'], 'Description')
  61.         self.assertIn('x2fit7q', results[0]['embedded'])
  63.         json = r"""
  64.         {"toto":[
  65.             {"id":200,"name":"Artist Name",
  66.             "link":"http:\/\/www.dailymotion.com\/artist\/1217","type":"artist"}
  67.         ]}
  68.         """
  69.         response = mock.Mock(text=json)
  70.         results = dailymotion.response(response)
  71.         self.assertEqual(type(results), list)
  72.         self.assertEqual(len(results), 0)
  74.     def test_fetch_supported_languages(self):
  75.         json = r"""
  76.         {"list":[{"code":"af","name":"Afrikaans","native_name":"Afrikaans",
  77.                   "localized_name":"Afrikaans","display_name":"Afrikaans"},
  78.                  {"code":"ar","name":"Arabic","native_name":"\u0627\u0644\u0639\u0631\u0628\u064a\u0629",
  79.                   "localized_name":"Arabic","display_name":"Arabic"},
  80.                  {"code":"la","name":"Latin","native_name":null,
  81.                   "localized_name":"Latin","display_name":"Latin"}
  82.         ]}
  83.         """
  84.         response = mock.Mock(text=json)
  85.         languages = dailymotion._fetch_supported_languages(response)
  86.         self.assertEqual(type(languages), dict)
  87.         self.assertEqual(len(languages), 3)
  88.         self.assertIn('af', languages)
  89.         self.assertIn('ar', languages)
  90.         self.assertIn('la', languages)
  92.         self.assertEqual(type(languages['af']), dict)
  93.         self.assertEqual(type(languages['ar']), dict)
  94.         self.assertEqual(type(languages['la']), dict)
  96.         self.assertIn('name', languages['af'])
  97.         self.assertIn('name', languages['ar'])
  98.         self.assertNotIn('name', languages['la'])
  100.         self.assertIn('english_name', languages['af'])
  101.         self.assertIn('english_name', languages['ar'])
  102.         self.assertIn('english_name', languages['la'])
  104.         self.assertEqual(languages['af']['name'], 'Afrikaans')
  105.         self.assertEqual(languages['af']['english_name'], 'Afrikaans')
  106.         self.assertEqual(languages['ar']['name'], u'العربية')
  107.         self.assertEqual(languages['ar']['english_name'], 'Arabic')
  108.         self.assertEqual(languages['la']['english_name'], 'Latin')