Index: PRESUBMIT_test_mocks.py |
diff --git a/PRESUBMIT_test_mocks.py b/PRESUBMIT_test_mocks.py |
index 7d286b9daa13260d70db6b1b43b624067c656381..5caa156593415fca998701940c2f03add77e52cf 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. |