123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122 |
- # -*- 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):
- query = 'test_query'
- dicto = defaultdict(dict)
- dicto['pageno'] = 1
- dicto['language'] = 'fr_FR'
- dicto['safesearch'] = 1
- 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('fr' in params['cookies']['SRCHHPGUSR'])
-
- dicto['language'] = 'all'
- params = bing_images.request(query, dicto)
- self.assertIn('SRCHHPGUSR', params['cookies'])
- self.assertIn('en', params['cookies']['SRCHHPGUSR'])
-
- 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='<html></html>')
- self.assertEqual(bing_images.response(response), [])
-
- response = mock.Mock(text='<html></html>')
- self.assertEqual(bing_images.response(response), [])
-
- html = """
- <div class="dg_u" style="width:178px;height:144px;left:17px;top:0px"><div>
- <a href="/images/search?q=south&view=detailv2&&id=7E92863981CCFB89FBDD55205C742DFDA3290CF6&selectedIndex=9&ccid=vzvIfv5u&simid=608055786735667000&thid=OIP.Mbf3bc87efe6e0e476be8cc34bf6cd80eH0" ihk="OIP.Mbf3bc87efe6e0e476be8cc34bf6cd80eH0" t1="South Carolina" t2="747 x 589 · 29 kB · gif" t3="www.digital-topo-maps.com/county-map/south-carolina.shtml" hh="236" hw="300" m='{ns:"images",k:"5117",mid:"7E92863981CCFB89FBDD55205C742DFDA3290CF6",md5:"bf3bc87efe6e0e476be8cc34bf6cd80e",surl:"http://www.digital-topo-maps.com/county-map/south-carolina.shtml",imgurl:"http://www.digital-topo-maps.com/county-map/south-carolina-county-map.gif",tid:"OIP.Mbf3bc87efe6e0e476be8cc34bf6cd80eH0",ow:"480",docid:"608055786735667000",oh:"378",tft:"45"}' mid="7E92863981CCFB89FBDD55205C742DFDA3290CF6" h="ID=images,5117.1">
- <img class="img_hid" src2="https://tse4.mm.bing.net/th?id=OIP.Mbf3bc87efe6e0e476be8cc34bf6cd80eH0&w=210&h=154&c=7&rs=1&qlt=90&o=4&pid=1.1" style="width:210px;height:154px;" width="210" height="154">
- </a>
-
- </div></div>
- """ # noqa
- 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), 1)
- self.assertEqual(results[0]['title'], 'South Carolina')
- self.assertEqual(results[0]['url'],
- 'http://www.digital-topo-maps.com/county-map/south-carolina.shtml')
- self.assertEqual(results[0]['content'], '')
- self.assertEqual(results[0]['thumbnail_src'],
- 'https://www.bing.com/th?id=OIP.Mbf3bc87efe6e0e476be8cc34bf6cd80eH0')
- self.assertEqual(results[0]['img_src'],
- 'http://www.digital-topo-maps.com/county-map/south-carolina-county-map.gif')
-
- html = """
- <a href="#" ihk="HN.608003696942779811"
- m="{ns:"images",k:"5045",
- mid:"59EB92C317974F34517A1CCAEBEF76A578E08DEE",
- surl:"http://www.page.url/",
- imgurl:"http://test.url/Test%20Query.jpg",oh:"238",
- tft:"0",oi:"http://www.image.url/Images/Test%20Query.jpg"}"
- mid="59EB92C317974F34517A1CCAEBEF76A578E08DEE" onclick="return false;"
- t1="Test Query" t2="650 x 517 · 31 kB · jpeg" t3="www.short.url" h="ID=images,5045.1">
- <img src="https://tse4.mm.bing.net/th?id=HN.608003696942779811&o=4&pid=1.7"
- style="height:144px;" width="178" height="144"/>
- </a>
- """
- response = mock.Mock(text=html)
- results = bing_images.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 0)
-
- html = """
- <div class="dg_u" style="width:178px;height:144px;left:17px;top:0px"><div>
- <a href="#" ihk="HN.608003696942779811"
- m="{ns:"images",k:"5045",
- mid:"659EB92C317974F34517A1CCAEBEF76A578E08DEE",
- surl:"http://www.page.url/",imgurl:"http://test.url/Test%20Query.jpg",
- oh:"238",tft:"0",oi:"http://www.image.url/Images/Test%20Query.jpg"}"
- mid="59EB92C317974F34517A1CCAEBEF76A578E08DEE" onclick="return false;"
- t1="Test Query" t2="650 x 517 · 31 kB · jpeg" t3="www.short.url" h="ID=images,5045.1">
- <img src="https://tse4.mm.bing.net/th?id=HN.608003696942779811&o=4&pid=1.7"
- style="height:144px;" width="178" height="144"/>
- </a>
- </div></div>
- <div class="dg_u" style="width:178px;height:144px;left:17px;top:0px"><div>
- <a href="#" ihk="HN.608003696942779811"
- m="{ns:"images",k:"5045",
- mid:"659EB92C317974F34517A1CCAEBEF76A578E08DEE",
- surl:"http://www.page.url/",imgurl:"http://test.url/Test%20Query.jpg",
- oh:"238",tft:"0",oi:"http://www.image.url/Images/Test%20Query.jpg"}"
- mid="59EB92C317974F34517A1CCAEBEF76A578E08DEE" onclick="return false;"
- t1="Test Query" t2="650 x 517 · 31 kB · jpeg" t3="www.short.url" h="ID=images,5045.1">
- <img src="https://tse4.mm.bing.net/th?id=HN.608003696942779811&o=4&pid=1.7"
- style="height:144px;" width="178" height="144"/>
- </a>
- </div></div>
- <div class="dg_u" style="width:178px;height:144px;left:17px;top:0px"><div>
- <a href="#" ihk="HN.608003696942779811"
- m="{ns:"images",k:"5045",
- mid:"659EB92C317974F34517A1CCAEBEF76A578E08DEE",
- surl:"http://www.page.url/",imgurl:"http://test.url/Test%20Query.jpg",
- oh:"238",tft:"0",oi:"http://www.image.url/Images/Test%20Query.jpg"}"
- mid="59EB92C317974F34517A1CCAEBEF76A578E08DEE" onclick="return false;"
- t1="Test Query" t2="650 x 517 · 31 kB · jpeg" t3="www.short.url" h="ID=images,5045.1">
- <img src="https://tse4.mm.bing.net/th?id=HN.608003696942779811&o=4&pid=1.7"
- style="height:144px;" width="178" height="144"/>
- </a>
- </div></div>
- """
- 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)
|