diff options
Diffstat (limited to 'PRESUBMIT_test_mocks.py')
-rw-r--r-- | PRESUBMIT_test_mocks.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/PRESUBMIT_test_mocks.py b/PRESUBMIT_test_mocks.py index 7d286b9..5caa156 100644 --- a/PRESUBMIT_test_mocks.py +++ b/PRESUBMIT_test_mocks.py @@ -29,10 +29,15 @@ class MockInputApi(object): def AffectedFiles(self, file_filter=None): return self.files + def AffectedSourceFiles(self, file_filter=None): + return self.files + def PresubmitLocalPath(self): return os.path.dirname(__file__) def ReadFile(self, filename, mode='rU'): + if hasattr(filename, 'AbsoluteLocalPath'): + filename = filename.AbsoluteLocalPath() for file_ in self.files: if file_.LocalPath() == filename: return '\n'.join(file_.NewContents()) @@ -96,6 +101,11 @@ class MockFile(object): return self._local_path +class MockAffectedFile(MockFile): + def AbsoluteLocalPath(self): + return self._local_path + + class MockChange(object): """Mock class for Change class. |