forked from Ponysearch/Ponysearch
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
from collections import defaultdict
|
|
import mock
|
|
from json import dumps
|
|
from searx.engines import frinkiac
|
|
from searx.testing import SearxTestCase
|
|
|
|
|
|
class TestFrinkiacEngine(SearxTestCase):
|
|
|
|
def test_request(self):
|
|
query = 'test_query'
|
|
request_dict = defaultdict(dict)
|
|
params = frinkiac.request(query, request_dict)
|
|
self.assertTrue('url' in params)
|
|
|
|
def test_response(self):
|
|
self.assertRaises(AttributeError, frinkiac.response, None)
|
|
self.assertRaises(AttributeError, frinkiac.response, [])
|
|
self.assertRaises(AttributeError, frinkiac.response, '')
|
|
self.assertRaises(AttributeError, frinkiac.response, '[]')
|
|
|
|
text = """
|
|
[{"Id":770931,
|
|
"Episode":"S06E18",
|
|
"Timestamp":534616,
|
|
"Filename":""},
|
|
{"Id":1657080,
|
|
"Episode":"S12E14",
|
|
"Timestamp":910868,
|
|
"Filename":""},
|
|
{"Id":1943753,
|
|
"Episode":"S14E21",
|
|
"Timestamp":773439,
|
|
"Filename":""},
|
|
{"Id":107835,
|
|
"Episode":"S02E03",
|
|
"Timestamp":531709,
|
|
"Filename":""}]
|
|
"""
|
|
|
|
response = mock.Mock(text=text)
|
|
results = frinkiac.response(response)
|
|
self.assertEqual(type(results), list)
|
|
self.assertEqual(len(results), 4)
|
|
self.assertEqual(results[0]['title'], u'S06E18')
|
|
self.assertEqual(results[0]['url'], 'https://frinkiac.com/?p=caption&e=S06E18&t=534616')
|
|
self.assertEqual(results[0]['thumbnail_src'], 'https://frinkiac.com/img/S06E18/534616/medium.jpg')
|
|
self.assertEqual(results[0]['img_src'], 'https://frinkiac.com/img/S06E18/534616.jpg')
|