# -*- coding: utf-8 -*- from collections import defaultdict import mock from searx.engines import bing_images from searx.testing import SearxTestCase class TestBingImagesEngine(SearxTestCase): def test_request(self): bing_images.supported_languages = ['fr-FR', 'en-US'] bing_images.language_aliases = {} query = 'test_query' dicto = defaultdict(dict) dicto['pageno'] = 1 dicto['language'] = 'fr-FR' dicto['safesearch'] = 1 dicto['time_range'] = '' params = bing_images.request(query, dicto) self.assertTrue('url' in params) self.assertTrue(query in params['url']) self.assertTrue('bing.com' in params['url']) self.assertTrue('SRCHHPGUSR' in params['cookies']) self.assertTrue('DEMOTE' in params['cookies']['SRCHHPGUSR']) self.assertTrue('_EDGE_S' in params['cookies']) self.assertTrue('fr-fr' in params['cookies']['_EDGE_S']) dicto['language'] = 'fr' params = bing_images.request(query, dicto) self.assertTrue('_EDGE_S' in params['cookies']) self.assertTrue('fr-fr' in params['cookies']['_EDGE_S']) dicto['language'] = 'all' params = bing_images.request(query, dicto) self.assertTrue('_EDGE_S' in params['cookies']) self.assertTrue('en-us' in params['cookies']['_EDGE_S']) def test_response(self): self.assertRaises(AttributeError, bing_images.response, None) self.assertRaises(AttributeError, bing_images.response, []) self.assertRaises(AttributeError, bing_images.response, '') self.assertRaises(AttributeError, bing_images.response, '[]') response = mock.Mock(text='') self.assertEqual(bing_images.response(response), []) response = mock.Mock(text='') self.assertEqual(bing_images.response(response), []) html = """

1 x 1 - jpeg 1.example.org

2 x 2 - jpeg 2.example.org

3 x 3 - jpeg 3.example.org

""" html = html.replace('\r\n', '').replace('\n', '').replace('\r', '') response = mock.Mock(text=html) results = bing_images.response(response) self.assertEqual(type(results), list) self.assertEqual(len(results), 3) self.assertEqual(results[0]['title'], 'Page 1 title') self.assertEqual(results[0]['url'], 'page_url') self.assertEqual(results[0]['content'], '') self.assertEqual(results[0]['thumbnail_src'], 'thumb_url') self.assertEqual(results[0]['img_src'], 'img_url') self.assertEqual(results[0]['img_format'], '1 x 1 - jpeg') self.assertEqual(results[0]['source'], '1.example.org') def test_fetch_supported_languages(self): html = """

Germany Norway

Argentina

""" response = mock.Mock(text=html) languages = list(bing_images._fetch_supported_languages(response)) self.assertEqual(len(languages), 3) self.assertIn('de-DE', languages) self.assertIn('no-NO', languages) self.assertIn('es-AR', languages)