From 8810273010afd91b0586f57d9c5794e7dd617d10 Mon Sep 17 00:00:00 2001 From: Gabor Nagy Date: Sat, 12 Jul 2014 15:46:55 +0200 Subject: [PATCH] Cover searx.utils --- .coveragerc | 20 ++++++++++++ Makefile | 6 ++-- searx/tests/test_utils.py | 69 +++++++++++++++++++++++++++++++++++++++ 3 files changed, 92 insertions(+), 3 deletions(-) create mode 100644 .coveragerc create mode 100644 searx/tests/test_utils.py diff --git a/.coveragerc b/.coveragerc new file mode 100644 index 000000000..4f50efc40 --- /dev/null +++ b/.coveragerc @@ -0,0 +1,20 @@ +[run] +branch = True +source = + searx/engines + searx/__init__.py + searx/autocomplete.py + searx/https_rewrite.py + searx/languages.py + searx/search.py + searx/testing.py + searx/utils.py + searx/webapp.py + +[report] +show_missing = True +exclude_lines = + if __name__ == .__main__.: + +[html] +directory = coverage diff --git a/Makefile b/Makefile index 2ef352eb1..f56060868 100644 --- a/Makefile +++ b/Makefile @@ -29,9 +29,9 @@ flake8: .installed.cfg @bin/flake8 ./searx/ coverage: .installed.cfg - @bin/coverage run --source=./searx/ --branch bin/test - @bin/coverage report --show-missing - @bin/coverage html --directory ./coverage + @bin/coverage run bin/test + @bin/coverage report + @bin/coverage html production: bin/buildout production.cfg setup.py bin/buildout -c production.cfg $(options) diff --git a/searx/tests/test_utils.py b/searx/tests/test_utils.py new file mode 100644 index 000000000..817fd4372 --- /dev/null +++ b/searx/tests/test_utils.py @@ -0,0 +1,69 @@ +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 = [ + '' + '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))