| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091 |
- import mock
- from searx.testing import SearxTestCase
- from searx import utils
- class TestUtils(SearxTestCase):
- def test_gen_useragent(self):
- self.assertIsInstance(utils.gen_useragent(), str)
- self.assertIsNotNone(utils.gen_useragent())
- self.assertTrue(utils.gen_useragent().startswith('Mozilla'))
- def test_searx_useragent(self):
- self.assertIsInstance(utils.searx_useragent(), str)
- self.assertIsNotNone(utils.searx_useragent())
- self.assertTrue(utils.searx_useragent().startswith('searx'))
- def test_highlight_content(self):
- self.assertEqual(utils.highlight_content(0, None), None)
- self.assertEqual(utils.highlight_content(None, None), None)
- self.assertEqual(utils.highlight_content('', None), None)
- self.assertEqual(utils.highlight_content(False, None), None)
- contents = [
- '<html></html>'
- 'not<'
- ]
- for content in contents:
- self.assertEqual(utils.highlight_content(content, None), content)
- content = 'a'
- query = 'test'
- self.assertEqual(utils.highlight_content(content, query), content)
- query = 'a test'
- self.assertEqual(utils.highlight_content(content, query), content)
- def test_html_to_text(self):
- html = """
- <a href="/testlink" class="link_access_account">
- <span class="toto">
- <span>
- <img src="test.jpg" />
- </span>
- </span>
- <span class="titi">
- Test text
- </span>
- </a>
- """
- self.assertIsInstance(utils.html_to_text(html), unicode)
- self.assertIsNotNone(utils.html_to_text(html))
- self.assertEqual(utils.html_to_text(html), "Test text")
- class TestHTMLTextExtractor(SearxTestCase):
- def setUp(self):
- self.html_text_extractor = utils.HTMLTextExtractor()
- def test__init__(self):
- self.assertEqual(self.html_text_extractor.result, [])
- def test_handle_charref(self):
- self.html_text_extractor.handle_charref('xF')
- self.assertIn(u'\x0f', self.html_text_extractor.result)
- self.html_text_extractor.handle_charref('XF')
- self.assertIn(u'\x0f', self.html_text_extractor.result)
- self.html_text_extractor.handle_charref('97')
- self.assertIn(u'a', self.html_text_extractor.result)
- def test_handle_entityref(self):
- entity = 'test'
- self.html_text_extractor.handle_entityref(entity)
- self.assertIn(entity, self.html_text_extractor.result)
- class TestUnicodeWriter(SearxTestCase):
- def setUp(self):
- self.unicode_writer = utils.UnicodeWriter(mock.MagicMock())
- def test_write_row(self):
- row = [1, 2, 3]
- self.assertEqual(self.unicode_writer.writerow(row), None)
- def test_write_rows(self):
- self.unicode_writer.writerow = mock.MagicMock()
- rows = [1, 2, 3]
- self.unicode_writer.writerows(rows)
- self.assertEqual(self.unicode_writer.writerow.call_count, len(rows))
|
