test_duden.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. from collections import defaultdict
  2. import mock
  3. from searx.engines import duden
  4. from searx.testing import SearxTestCase
  5. from datetime import datetime
  6. class TestDudenEngine(SearxTestCase):
  7. def test_request(self):
  8. query = 'Haus'
  9. dic = defaultdict(dict)
  10. data = [
  11. [1, 'https://www.duden.de/suchen/dudenonline/Haus'],
  12. [2, 'https://www.duden.de/suchen/dudenonline/Haus?search_api_fulltext=&page=1']
  13. ]
  14. for page_no, exp_res in data:
  15. dic['pageno'] = page_no
  16. params = duden.request(query, dic)
  17. self.assertTrue('url' in params)
  18. self.assertTrue(query in params['url'])
  19. self.assertTrue('duden.de' in params['url'])
  20. self.assertEqual(params['url'], exp_res)
  21. def test_response(self):
  22. resp = mock.Mock(text='<html></html>')
  23. self.assertEqual(duden.response(resp), [])
  24. html = """
  25. <section class="vignette">
  26. <h2"> <a href="/rechtschreibung/Haus">
  27. <strong>This is the title also here</strong>
  28. </a> </h2>
  29. <p>This is the content</p>
  30. </section>
  31. """
  32. resp = mock.Mock(text=html)
  33. results = duden.response(resp)
  34. self.assertEqual(len(results), 1)
  35. self.assertEqual(type(results), list)
  36. # testing result (dictionary entry)
  37. r = results[0]
  38. self.assertEqual(r['url'], 'https://www.duden.de/rechtschreibung/Haus')
  39. self.assertEqual(r['title'], 'This is the title also here')
  40. self.assertEqual(r['content'], 'This is the content')