123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236 |
- # -*- coding: utf-8 -*-
- from collections import defaultdict
- import mock
- from searx.engines import wolframalpha_noapi
- from searx.testing import SearxTestCase
- class TestWolframAlphaNoAPIEngine(SearxTestCase):
- def test_request(self):
- query = 'test_query'
- dicto = defaultdict(dict)
- dicto['pageno'] = 1
- params = wolframalpha_noapi.request(query, dicto)
- self.assertIn('url', params)
- self.assertIn(query, params['url'])
- self.assertIn('wolframalpha.com', params['url'])
- def test_response(self):
- self.assertRaises(AttributeError, wolframalpha_noapi.response, None)
- self.assertRaises(AttributeError, wolframalpha_noapi.response, [])
- self.assertRaises(AttributeError, wolframalpha_noapi.response, '')
- self.assertRaises(AttributeError, wolframalpha_noapi.response, '[]')
- # response = mock.Mock(text='<html></html>')
- # self.assertEqual(wolframalpha_noapi.response(response), [])
- html = """
- <!DOCTYPE html>
- <title> sqrt(-1) - Wolfram|Alpha</title>
- <meta charset="utf-8" />
- <body>
- <script type="text/javascript">
- try {
- document.domain = "wolframalpha.com";
- context = parent ? parent : document;
- } catch(e){}
- try {
- if (typeof(context.$) == "undefined") {
- context = window;
- } else {
- $=context.$;
- }
- }
- catch(e){ context = window;}
- try {
- if (typeof context.jsonArray.popups.pod_0100 == "undefined" ) {
- context.jsonArray.popups.pod_0100 = [];
- }
- context.jsonArray.popups.pod_0100.push( {"stringified": "sqrt(-1)","mInput": "","mOutput": "", "popLinks": {} });
- } catch(e) { }
- try {
- $("#results #pod_0100:not(iframe #pod_0100)")
- .add("#showsteps #pod_0100:not(iframe #pod_0100)")
- .add(".results-pod #pod_0100:not(iframe #pod_0100)")
- .data("tempFileID", 'MSP44501e0dda34g97a0c8900003i71207d6491ab22')
- .data("podIdentifier", '\x22Input\x22')
- .data("podShortIdentifier", '\x22Input\x22')
- .data("buttonStates", '\x22\x22')
- .data("scanner", '\x22\x22');
- $("#results #pod_0100-popup:not(iframe #pod_0100-popup)")
- .add("#showsteps #pod_0100-popup:not(iframe #pod_0100-popup)")
- .add(".results-pod #pod_0100-popup:not(iframe #pod_0100-popup)")
- .data("tempFileID", 'MSP44501e0dda34g97a0c8900003i71207d6491ab22')
- .data("podIdentifier", '\x22Input\x22')
- .data("podShortIdentifier", '\x22Input\x22')
- .data("buttonStates", '\x22\x22')
- .data("scanner", '\x22\x22');
- $("#results #subpod_0100_1")
- .add("#showsteps #subpod_0100_1:not(iframe #subpod_0100_1)")
- .add(".results-pod #subpod_0100_1")
- .data("tempFileID", "MSP44511e0dda34g97a0c89000059490h319161eea3")
- .data("cellDataTempFile", "MSP44521e0dda34g97a0c89000011378c50d38ede6h")
- .data("tempFileServer", "")
- .data("dataSources", "")
- .data("sources", "")
- .data("sharetype", "1")
- .data("shareable", "false");
- } catch(e){}
- //false
- try {
- if (typeof context.jsonArray.popups.pod_0200 == "undefined" ) {
- context.jsonArray.popups.pod_0200 = [];
- }
- context.jsonArray.popups.pod_0200.push( {"stringified": "i","mInput": "","mOutput": "", "popLinks": {} });
- } catch(e) { }
- try {
- $("#results #pod_0200:not(iframe #pod_0200)")
- .add("#showsteps #pod_0200:not(iframe #pod_0200)")
- .add(".results-pod #pod_0200:not(iframe #pod_0200)")
- .data("tempFileID", 'MSP44541e0dda34g97a0c8900004f449i50fa482fd8')
- .data("podIdentifier", '\x22Result\x22')
- .data("podShortIdentifier", '\x22Result\x22')
- .data("buttonStates", '\x22Result\x22\x20\x2D\x3E\x20\x7BAll,\x20None,\x20None,\x20None,\x20None\x7D')
- .data("scanner", '\x22\x22');
- $("#results #pod_0200-popup:not(iframe #pod_0200-popup)")
- .add("#showsteps #pod_0200-popup:not(iframe #pod_0200-popup)")
- .add(".results-pod #pod_0200-popup:not(iframe #pod_0200-popup)")
- .data("tempFileID", 'MSP44541e0dda34g97a0c8900004f449i50fa482fd8')
- .data("podIdentifier", '\x22Result\x22')
- .data("podShortIdentifier", '\x22Result\x22')
- .data("buttonStates", '\x22Result\x22\x20\x2D\x3E\x20\x7BAll,\x20None,\x20None\x7D')
- .data("scanner", '\x22\x22');
- $("#results #subpod_0200_1")
- .add("#showsteps #subpod_0200_1:not(iframe #subpod_0200_1)")
- .add(".results-pod #subpod_0200_1")
- .data("tempFileID", "MSP44551e0dda34g97a0c8900003gdgd37faa7272e0")
- .data("cellDataTempFile", "MSP44561e0dda34g97a0c89000018ea1iae00104g13")
- .data("tempFileServer", "")
- .data("dataSources", "")
- .data("sources", "")
- .data("sharetype", "1")
- .data("shareable", "false");
- } catch(e){}
- </script>
- </body>
- </html>
- """
- response = mock.Mock(text=html)
- results = wolframalpha_noapi.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 2)
- self.assertIn("i", results[0]['answer'])
- self.assertIn("sqrt(-1) - Wolfram|Alpha", results[1]['title'])
- self.assertIn("http://www.wolframalpha.com/input/?i=sqrt%28-1%29", results[1]['url'])
- html = """
- <!DOCTYPE html>
- <title> integral 1/x - Wolfram|Alpha</title>
- <meta charset="utf-8" />
- <body>
- <script type="text/javascript">
- //true
- try {
- document.domain = "wolframalpha.com";
- context = parent ? parent : document;
- } catch(e){}
- try {
- if (typeof(context.$) == "undefined") {
- context = window;
- } else {
- $=context.$;
- }
- }
- catch(e){ context = window;}
- try {
- if (typeof context.jsonArray.popups.pod_0100 == "undefined" ) {
- context.jsonArray.popups.pod_0100 = [];
- }
- context.jsonArray.popups.pod_0100.push( {"stringified": "integral 1\/x dx = log(x)+constant"});
- } catch(e) { }
- try {
- $("#results #pod_0100:not(iframe #pod_0100)")
- .add("#showsteps #pod_0100:not(iframe #pod_0100)")
- .add(".results-pod #pod_0100:not(iframe #pod_0100)")
- .data("tempFileID", 'MSP2051if2202e8bg0757100000d119b05egf583d3')
- .data("podIdentifier", '\x22IndefiniteIntegral\x22')
- .data("podShortIdentifier", '\x22IndefiniteIntegral\x22')
- .data("buttonStates", '\x22Indefinite\x20integral\x22\x20\x2D\x3E\x20\x7B\x7D')
- .data("scanner", '\x22\x22');
- $("#results #pod_0100-popup:not(iframe #pod_0100-popup)")
- .add("#showsteps #pod_0100-popup:not(iframe #pod_0100-popup)")
- .add(".results-pod #pod_0100-popup:not(iframe #pod_0100-popup)")
- .data("tempFileID", 'MSP2051if2202e8bg0757100000d119b05egf583d3')
- .data("podIdentifier", '\x22IndefiniteIntegral\x22')
- .data("podShortIdentifier", '\x22IndefiniteIntegral\x22')
- .data("buttonStates", '\x22Indefinite\x20integral\x22\x20\x2D\x3E\x20\x7B\x7D')
- .data("scanner", '\x22\x22');
- $("#results #subpod_0100_1")
- .add("#showsteps #subpod_0100_1:not(iframe #subpod_0100_1)")
- .add(".results-pod #subpod_0100_1")
- .data("tempFileID", "MSP2071if2202e8bg0757100004dg60f2a4ca8cf73")
- .data("cellDataTempFile", "MSP2081if2202e8bg0757100001h18329f72fe90fg")
- .data("tempFileServer", "")
- .data("dataSources", "")
- .data("sources", "")
- .data("sharetype", "1")
- .data("shareable", "false");
- } catch(e){}
- //false
- try {
- $("#results #pod_0200:not(iframe #pod_0200)")
- .add("#showsteps #pod_0200:not(iframe #pod_0200)")
- .add(".results-pod #pod_0200:not(iframe #pod_0200)")
- .data("tempFileID", '')
- .data("podIdentifier", '\x22Plot\x22')
- .data("podShortIdentifier", '')
- .data("buttonStates", '')
- .data("scanner", '\x22\x22');
- $("#results #pod_0200-popup:not(iframe #pod_0200-popup)")
- .add("#showsteps #pod_0200-popup:not(iframe #pod_0200-popup)")
- .add(".results-pod #pod_0200-popup:not(iframe #pod_0200-popup)")
- .data("tempFileID", '')
- .data("podIdentifier", '\x22Plot\x22')
- .data("podShortIdentifier", '')
- .data("buttonStates", '')
- .data("scanner", '\x22\x22');
- } catch(e){}
- </script>
- </body>
- </html>
- """
- response = mock.Mock(text=html)
- results = wolframalpha_noapi.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 2)
- self.assertIn("log(x)+c", results[0]['answer'])
- self.assertIn("integral 1/x - Wolfram|Alpha", results[1]['title'])
- self.assertIn("http://www.wolframalpha.com/input/?i=integral+1%2Fx", results[1]['url'])
|