# 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, white_list=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.*']
             }]
    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)