Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 import os | 6 import os |
| 7 import re | 7 import re |
| 8 import unittest | 8 import unittest |
| 9 | 9 |
| 10 import PRESUBMIT | 10 import PRESUBMIT |
| 11 | 11 |
| 12 | 12 |
| 13 class MockInputApi(object): | 13 class MockInputApi(object): |
| 14 def __init__(self): | 14 def __init__(self): |
| 15 self.re = re | 15 self.re = re |
| 16 self.os_path = os.path | 16 self.os_path = os.path |
| 17 self.files = [] | |
| 18 | |
| 19 def AddFile(self, f): | |
|
M-A Ruel
2012/12/05 19:30:17
I don't think it's necessary.
| |
| 20 self.files.append(f) | |
| 21 | |
| 22 def AffectedFiles(self): | |
| 23 return self.files | |
| 24 | |
| 25 | |
| 26 class MockOutputApi(object): | |
| 27 class PresubmitResult(object): | |
| 28 def __init__(self, message, items=None, long_text=''): | |
| 29 self.message = message | |
| 30 self.items = items | |
| 31 self.long_text = long_text | |
| 32 | |
| 33 class PresubmitError(PresubmitResult): | |
| 34 pass | |
| 35 | |
| 36 class PresubmitPromptWarning(PresubmitResult): | |
| 37 pass | |
| 38 | |
| 39 class PresubmitNotifyResult(PresubmitResult): | |
| 40 pass | |
| 17 | 41 |
| 18 | 42 |
| 19 class MockFile(object): | 43 class MockFile(object): |
| 20 def __init__(self, local_path, new_contents): | 44 def __init__(self, local_path, new_contents): |
| 21 self._local_path = local_path | 45 self._local_path = local_path |
| 22 self._new_contents = new_contents | 46 self._new_contents = new_contents |
| 23 self._changed_contents = [(i + 1, l) for i, l in enumerate(new_contents)] | 47 self._changed_contents = [(i + 1, l) for i, l in enumerate(new_contents)] |
| 24 | 48 |
| 25 def ChangedContents(self): | 49 def ChangedContents(self): |
| 26 return self._changed_contents | 50 return self._changed_contents |
| (...skipping 167 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 194 ' ScopedTempDir temp_dir_;', | 218 ' ScopedTempDir temp_dir_;', |
| 195 '>>>>>>> master'] | 219 '>>>>>>> master'] |
| 196 errors = PRESUBMIT._CheckForVersionControlConflictsInFile( | 220 errors = PRESUBMIT._CheckForVersionControlConflictsInFile( |
| 197 MockInputApi(), MockFile('some/path/foo_platform.cc', lines)) | 221 MockInputApi(), MockFile('some/path/foo_platform.cc', lines)) |
| 198 self.assertEqual(3, len(errors)) | 222 self.assertEqual(3, len(errors)) |
| 199 self.assertTrue('1' in errors[0]) | 223 self.assertTrue('1' in errors[0]) |
| 200 self.assertTrue('3' in errors[1]) | 224 self.assertTrue('3' in errors[1]) |
| 201 self.assertTrue('5' in errors[2]) | 225 self.assertTrue('5' in errors[2]) |
| 202 | 226 |
| 203 | 227 |
| 228 class BadExtensionsTest(unittest.TestCase): | |
| 229 def testBadRejFile(self): | |
| 230 mock_input_api = MockInputApi() | |
| 231 mock_input_api.AddFile(MockFile('some/path/foo.cc', '')) | |
|
M-A Ruel
2012/12/05 19:30:17
mock_input_api.files = [
MockFile(...),
..
| |
| 232 mock_input_api.AddFile(MockFile('some/path/foo.cc.rej', '')) | |
| 233 mock_input_api.AddFile(MockFile('some/path2/bar.h.rej', '')) | |
| 234 | |
| 235 results = PRESUBMIT._CheckPatchFiles(mock_input_api, MockOutputApi()) | |
| 236 self.assertEqual(1, len(results)) | |
| 237 self.assertEqual(2, len(results[0].items)) | |
| 238 self.assertTrue('foo.cc.rej' in results[0].items[0]) | |
| 239 self.assertTrue('bar.h.rej' in results[0].items[1]) | |
| 240 | |
| 241 def testBadOrigFile(self): | |
| 242 mock_input_api = MockInputApi() | |
| 243 mock_input_api.AddFile(MockFile('other/path/qux.h.orig', '')) | |
| 244 mock_input_api.AddFile(MockFile('other/path/qux.h', '')) | |
| 245 mock_input_api.AddFile(MockFile('other/path/qux.cc', '')) | |
| 246 | |
| 247 results = PRESUBMIT._CheckPatchFiles(mock_input_api, MockOutputApi()) | |
| 248 self.assertEqual(1, len(results)) | |
| 249 self.assertEqual(1, len(results[0].items)) | |
| 250 self.assertTrue('qux.h.orig' in results[0].items[0]) | |
| 251 | |
| 252 def testGoodFiles(self): | |
| 253 mock_input_api = MockInputApi() | |
| 254 mock_input_api.AddFile(MockFile('other/path/qux.h', '')) | |
| 255 mock_input_api.AddFile(MockFile('other/path/qux.cc', '')) | |
| 256 results = PRESUBMIT._CheckPatchFiles(mock_input_api, MockOutputApi()) | |
| 257 self.assertEqual(0, len(results)) | |
| 258 | |
| 259 | |
| 204 if __name__ == '__main__': | 260 if __name__ == '__main__': |
| 205 unittest.main() | 261 unittest.main() |
| OLD | NEW |