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)