| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- 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_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)
- 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))
|
