class PatchCheckerTest(unittest.TestCase):
"""Test the PatchChecker class."""
class MockStyleChecker(object):
def __init__(self):
self.checked_files = []
"""A list of (file_path, line_numbers) pairs."""
def check_file(self, file_path, line_numbers):
self.checked_files.append((file_path, line_numbers))
def setUp(self):
style_checker = self.MockStyleChecker()
self._style_checker = style_checker
self._patch_checker = PatchChecker(style_checker)
def _call_check_patch(self, patch_string):
self._patch_checker.check(patch_string)
def _assert_checked(self, checked_files):
self.assertEquals(self._style_checker.checked_files, checked_files)
def test_check_patch(self):
# The modified line_numbers array for this patch is: [2].
self._call_check_patch("""diff --git a/__init__.py b/__init__.py
index ef65bee..e3db70e 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,1 +1,2 @@
# Required for Python to search this directory for module files
+# New line
""")
self._assert_checked([("__init__.py", set([2]))])
def setUp(self):
style_checker = self.MockStyleChecker()
self._style_checker = style_checker
self._patch_checker = PatchChecker(style_checker)
