eris/tools_test.py

226 lines
7.5 KiB
Python
Executable file

#!/usr/bin/env python3
# Copyright (C) 2016 Andrew Hamilton. All rights reserved.
# Licensed under the Artistic License 2.0.
import contextlib
import os
import subprocess
import unittest
import unittest.mock
os.environ["TERM"] = "xterm-256color"
import fill3
import golden
import tools
os.environ["TZ"] = "GMT"
VIGIL_ROOT = os.path.dirname(__file__)
def widget_to_string(widget):
appearance = widget.appearance_min()
return str(fill3.join("\n", appearance))
@contextlib.contextmanager
def chdir(path):
old_cwd = os.getcwd()
os.chdir(path)
try:
yield
finally:
os.chdir(old_cwd)
def result_path(tool, input_filename):
filename = tool.__qualname__ + "-" + input_filename.replace(".", "_")
return os.path.join(VIGIL_ROOT, "golden-files", "results", filename)
def run_tool(tool, input_filename):
with chdir("golden-files"):
return tool(os.path.join(".", "input", input_filename))
class ToolsTestCase(unittest.TestCase):
def _test_tool(self, tool, sub_tests):
for input_filename, expected_status in sub_tests:
with self.subTest(input_filename=input_filename):
status, result = run_tool(tool, input_filename)
golden_path = result_path(tool, input_filename)
text = widget_to_string(result)
text = text.replace(os.environ["HOME"], "/home/EVERY_USER")
golden.assertGolden(text, golden_path)
self.assertEqual(status, expected_status)
def test_metadata(self):
mock_stat_result = unittest.mock.Mock(
st_mode=0o755, st_mtime=1454282045, st_ctime=1454282045,
st_atime=1454282047, st_size=12, st_uid=1111, st_gid=1111,
st_nlink=2)
mock_pw_entry = unittest.mock.Mock(pw_name="foo")
with unittest.mock.patch.object(os, "stat",
return_value=mock_stat_result):
with unittest.mock.patch.object(tools.pwd, "getpwuid",
return_value=mock_pw_entry):
self._test_tool(tools.metadata,
[("hi3.py", tools.Status.normal)])
def test_contents(self):
self._test_tool(tools.contents, [("hi3.py", tools.Status.normal)])
HI_OK = [("hi3.py", tools.Status.ok), ("hi.py", tools.Status.ok)]
def test_python_syntax(self):
self._test_tool(tools.python_syntax, self.HI_OK)
def test_python_unittests(self):
self._test_tool(tools.python_unittests,
[("hi3.py", tools.Status.not_applicable),
("hi.py", tools.Status.not_applicable),
("hi3_test.py", tools.Status.ok),
("test_foo.py", tools.Status.ok)])
HI_NORMAL = [("hi3.py", tools.Status.normal),
("hi.py", tools.Status.normal)]
def test_pydoc(self):
self._test_tool(tools.pydoc, self.HI_NORMAL)
def test_mypy(self):
self._test_tool(tools.mypy, [("hi3.py", tools.Status.ok),
("hi.py", tools.Status.normal)])
def test_python_coverage(self):
self._test_tool(tools.python_coverage, self.HI_NORMAL)
# Not testing python_profile, because it is non-deterministic.
def test_pep8(self):
self._test_tool(tools.pep8, self.HI_OK)
def test_pyflakes(self):
self._test_tool(tools.pyflakes, self.HI_OK)
def test_pylint(self):
self._test_tool(tools.pylint, self.HI_OK)
def test_python_gut(self):
self._test_tool(tools.python_gut, self.HI_NORMAL)
def test_python_modulefinder(self):
self._test_tool(tools.python_modulefinder, self.HI_NORMAL)
def test_python_mccable(self):
self._test_tool(tools.python_mccabe, self.HI_OK)
def test_bandit(self):
self._test_tool(tools.bandit, [("hi3.py", tools.Status.ok),
("hi.py", tools.Status.ok)])
def test_disassemble_pyc(self):
self._test_tool(tools.disassemble_pyc,
[("hi3.cpython-34.pyc", tools.Status.normal)])
def test_perl_syntax(self):
self._test_tool(tools.perl_syntax,
[("perl.pl", tools.Status.ok),
# ("perl6.pl", tools.Status.problem)
])
def test_perldoc(self):
self._test_tool(tools.perldoc,
[("perl.pl", tools.Status.not_applicable),
("contents.pod", tools.Status.normal)])
def test_perltidy(self):
self._test_tool(tools.perltidy, [("perl.pl", tools.Status.normal)])
# def test_perl6_syntax(self):
# self._test_tool(tools.perl6_syntax,
# [("perl6.p6", tools.Status.problem)])
def test_uncrustify(self):
self._test_tool(tools.uncrustify,
[("closure-util.java", tools.Status.problem),
("hello.c", tools.Status.normal),
("hello.h", tools.Status.normal),
("hello.cpp", tools.Status.normal)])
def test_splint(self):
self._test_tool(tools.splint, [("hello.c", tools.Status.ok),
("hello.h", tools.Status.ok)])
def test_objdump_headers(self):
self._test_tool(tools.objdump_headers,
[("Mcrt1.o", tools.Status.normal)])
def test_objdump_disassemble(self):
self._test_tool(tools.objdump_disassemble,
[("Mcrt1.o", tools.Status.problem)])
def test_readelf(self):
self._test_tool(tools.readelf, [("Mcrt1.o", tools.Status.normal)])
def test_unzip(self):
self._test_tool(tools.unzip, [("hi.zip", tools.Status.normal)])
def test_tar_gz(self):
self._test_tool(tools.tar_gz, [("hi.tar.gz", tools.Status.normal),
("hi.tgz", tools.Status.normal)])
def test_tar_bz2(self):
self._test_tool(tools.tar_bz2, [("hi.tar.bz2", tools.Status.normal)])
def test_nm(self):
self._test_tool(tools.nm, [("libieee.a", tools.Status.normal),
("libpcprofile.so", tools.Status.normal)])
def test_pdf2txt(self):
self._test_tool(tools.pdf2txt, [("standard.pdf", tools.Status.normal)])
def test_html_syntax(self):
self._test_tool(tools.html_syntax, [("hi.html", tools.Status.problem)])
def test_tidy(self):
self._test_tool(tools.tidy, [("hi.html", tools.Status.normal)])
def test_html2text(self):
self._test_tool(tools.html2text, [("hi.html", tools.Status.normal)])
def test_bcpp(self):
self._test_tool(tools.bcpp, [("hello.cpp", tools.Status.normal)])
def test_php5_syntax(self):
self._test_tool(tools.php5_syntax, [("root.php", tools.Status.ok)])
class LruCacheWithEvictionTestCase(unittest.TestCase):
def _assert_cache(self, func, hits, misses, current_size):
cache_info = func.cache_info()
self.assertEqual(cache_info.hits, hits)
self.assertEqual(cache_info.misses, misses)
self.assertEqual(cache_info.currsize, current_size)
def test_lru_cache_with_eviction(self):
@tools.lru_cache_with_eviction()
def a(foo):
return foo
self._assert_cache(a, 0, 0, 0)
self.assertEqual(a(1), 1)
self._assert_cache(a, 0, 1, 1)
a(1)
self._assert_cache(a, 1, 1, 1)
a.evict(1)
self._assert_cache(a, 1, 1, 1)
a(1)
self._assert_cache(a, 1, 2, 2)
if __name__ == "__main__":
golden.main()