| # Copyright 2019 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| |
| def _GenerateTestCommand(input_api, output_api, file_name, affected_list): |
| if not input_api.AffectedFiles( |
| file_filter=lambda x: input_api.FilterSourceFile( |
| x, files_to_check=affected_list)): |
| return None |
| |
| if input_api.is_committing: |
| message_type = output_api.PresubmitError |
| else: |
| message_type = output_api.PresubmitPromptWarning |
| |
| test_path = input_api.os_path.join(input_api.PresubmitLocalPath(), |
| file_name) |
| if input_api.is_windows: |
| cmd = [input_api.python_executable, test_path] |
| else: |
| cmd = [test_path] |
| |
| # Adds "//third_party" to the path, so that the jinja2 module can be found |
| # during import. |
| env = input_api.environ.copy() |
| import_path = [ |
| input_api.os_path.join(input_api.change.RepositoryRoot(), |
| 'third_party') |
| ] |
| if env.get('PYTHONPATH'): |
| import_path.append(env.get('PYTHONPATH')) |
| env['PYTHONPATH'] = input_api.os_path.pathsep.join(import_path) |
| |
| test_cmd = input_api.Command( |
| name=file_name, cmd=cmd, kwargs={'env': env}, message=message_type) |
| return test_cmd |
| |
| |
| def _RunTests(input_api, output_api): |
| tests = [{ |
| 'file_name': 'json5_generator_unittest.py', |
| 'affected_list': [r'.*json5_generator.*', r'.*\btests[\\\/].*'] |
| }, { |
| 'file_name': 'make_runtime_features_utilities_unittest.py', |
| 'affected_list': [r'.*make_runtime_features_utilities.*'] |
| }, { |
| 'file_name': 'make_document_policy_features_unittest.py', |
| 'affected_list': [r'.*make_document_policy_features.*'] |
| }, { |
| 'file_name': 'make_document_policy_features_tests.py', |
| 'affected_list': [r'.*make_document_policy_features.*'] |
| }] |
| test_commands = [] |
| for test in tests: |
| test_commands.append( |
| _GenerateTestCommand(input_api, output_api, test['file_name'], |
| test['affected_list'])) |
| return input_api.RunTests( |
| [command for command in test_commands if command]) |
| |
| |
| def CheckChangeOnUpload(input_api, output_api): |
| return _RunTests(input_api, output_api) |
| |
| |
| def CheckChangeOnCommit(input_api, output_api): |
| return _RunTests(input_api, output_api) |