# -*- coding: utf-8 -*- from collections import defaultdict import mock from searx.engines import twitter from searx.testing import SearxTestCase class TestTwitterEngine(SearxTestCase): def test_request(self): query = 'test_query' dicto = defaultdict(dict) dicto['pageno'] = 0 dicto['language'] = 'fr_FR' params = twitter.request(query, dicto) self.assertIn('url', params) self.assertIn(query, params['url']) self.assertIn('twitter.com', params['url']) self.assertIn('cookies', params) self.assertIn('lang', params['cookies']) self.assertIn('fr', params['cookies']['lang']) dicto['language'] = 'all' params = twitter.request(query, dicto) self.assertIn('cookies', params) self.assertIn('lang', params['cookies']) self.assertIn('en', params['cookies']['lang']) def test_response(self): self.assertRaises(AttributeError, twitter.response, None) self.assertRaises(AttributeError, twitter.response, []) self.assertRaises(AttributeError, twitter.response, '') self.assertRaises(AttributeError, twitter.response, '[]') response = mock.Mock(text='') self.assertEqual(twitter.response(response), []) html = """

""" response = mock.Mock(text=html) results = twitter.response(response) self.assertEqual(type(results), list) self.assertEqual(len(results), 1) self.assertEqual(results[0]['title'], '@TitleName') self.assertEqual(results[0]['url'], 'https://twitter.com/this.is.the.url') self.assertIn(u'This is the content', results[0]['content']) # self.assertIn(u'This is the content étude à€', results[0]['content']) html = """

This should be the title

this.meta.com

This should be the content.

""" response = mock.Mock(text=html) results = twitter.response(response) self.assertEqual(type(results), list) self.assertEqual(len(results), 0)