import mock from collections import defaultdict from searx.engines import tokyotoshokan from searx.testing import SearxTestCase from datetime import datetime class TestTokyotoshokanEngine(SearxTestCase): def test_request(self): query = 'test_query' dic = defaultdict(dict) dic['pageno'] = 1 params = tokyotoshokan.request(query, dic) self.assertTrue('url' in params) self.assertTrue(query in params['url']) self.assertTrue('tokyotosho.info' in params['url']) def test_response(self): resp = mock.Mock(text='') self.assertEqual(tokyotoshokan.response(resp), []) html = """
Koyomimonogatari Details
Authorized: Yes Submitter: Ohys | Size: 10.5MB | Date: 2016-03-26 16:41 UTC | Comment: sample comment S: 53 L: 18 C: 0 ID: 975700
Owarimonogatari Details
Submitter: Ohys | Size: 932.84EB | Date: QWERTY-03-26 16:41 UTC S: 0
""" resp = mock.Mock(text=html) results = tokyotoshokan.response(resp) self.assertEqual(type(results), list) self.assertEqual(len(results), 2) # testing the first result, which has correct format # and should have all information fields filled r = results[0] self.assertEqual(r['url'], 'http://www.nyaa.se/f') self.assertEqual(r['title'], 'Koyomimonogatari') self.assertEqual(r['magnetlink'], 'magnet:?xt=urn:btih:4c19eb46b5113685fbd2288ed2531b0b') self.assertEqual(r['filesize'], int(1024 * 1024 * 10.5)) self.assertEqual(r['publishedDate'], datetime(2016, 03, 26, 16, 41)) self.assertEqual(r['content'], 'Comment: sample comment') self.assertEqual(r['seed'], 53) self.assertEqual(r['leech'], 18) # testing the second result, which does not include magnet link, # seed & leech info, and has incorrect size & creation date r = results[1] self.assertEqual(r['url'], 'http://google.com/q') self.assertEqual(r['title'], 'Owarimonogatari') self.assertFalse('magnetlink' in r) self.assertFalse('filesize' in r) self.assertFalse('content' in r) self.assertFalse('publishedDate' in r) self.assertFalse('seed' in r) self.assertFalse('leech' in r)