# Copyright (c) 2011 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.

"""Presubmit script for perf_expectations.

See http://dev.chromium.org/developers/how-tos/depottools/presubmit-scripts for
details on the presubmit API built into depot_tools.
"""

PERF_EXPECTATIONS = 'tools/perf_expectations/perf_expectations.json'
CONFIG_FILE = 'tools/perf_expectations/chromium_perf_expectations.cfg'

def CheckChangeOnUpload(input_api, output_api):
  run_tests = False
  for path in input_api.LocalPaths():
    if (PERF_EXPECTATIONS == path or CONFIG_FILE == path):
      run_tests = True

  output = []
  if run_tests:
    whitelist = [r'.+_unittest\.py$']
    output.extend(input_api.canned_checks.RunUnitTestsInDirectory(
        input_api, output_api, 'tests', whitelist))
  return output


def CheckChangeOnCommit(input_api, output_api):
  run_tests = False
  for path in input_api.LocalPaths():
    if (PERF_EXPECTATIONS == path or CONFIG_FILE == path):
      run_tests = True

  output = []
  if run_tests:
    whitelist = [r'.+_unittest\.py$']
    output.extend(input_api.canned_checks.RunUnitTestsInDirectory(
        input_api, output_api, 'tests', whitelist))

  output.extend(input_api.canned_checks.CheckDoNotSubmit(input_api,
                                                         output_api))
  return output