blob: 2bd280e06486e324e2aa0bb22a29f44ad389aa2a [file] [log] [blame]
# Copyright 2015 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 devil.
See http://dev.chromium.org/developers/how-tos/depottools/presubmit-scripts for
details on the presubmit API built into depot_tools.
"""
USE_PYTHON3 = True
def _RunPylint(input_api, output_api):
return input_api.RunTests(
input_api.canned_checks.RunPylint(
input_api, output_api, pylintrc='pylintrc'))
def _RunUnitTests(input_api, output_api):
def J(*dirs):
"""Returns a path relative to presubmit directory."""
return input_api.os_path.join(input_api.PresubmitLocalPath(), 'devil',
*dirs)
test_env = dict(input_api.environ)
test_env.update({
'PYTHONDONTWRITEBYTECODE': '1',
'PYTHONPATH': ':'.join([J(), J('..')]),
})
message_type = (output_api.PresubmitError if input_api.is_committing else
output_api.PresubmitPromptWarning)
return input_api.RunTests([
input_api.Command(name='devil/bin/run_py_tests',
cmd=[
input_api.os_path.join(
input_api.PresubmitLocalPath(), 'bin',
'run_py_tests')
],
kwargs={'env': test_env},
message=message_type),
input_api.Command(name='devil/bin/run_py3_tests',
cmd=[
input_api.os_path.join(
input_api.PresubmitLocalPath(), 'bin',
'run_py3_tests')
],
kwargs={'env': test_env},
message=message_type,
python3=True),
])
def _EnsureNoPylibUse(input_api, output_api):
def other_python_files(f):
this_presubmit_file = input_api.os_path.join(input_api.PresubmitLocalPath(),
'PRESUBMIT.py')
return (f.LocalPath().endswith('.py')
and not f.AbsoluteLocalPath() == this_presubmit_file)
changed_files = input_api.AffectedSourceFiles(other_python_files)
import_error_re = input_api.re.compile(
r'(from pylib.* import)|(import pylib)')
errors = []
for f in changed_files:
errors.extend('%s:%d' % (f.LocalPath(), line_number)
for line_number, line_text in f.ChangedContents()
if import_error_re.search(line_text))
if errors:
return [
output_api.PresubmitError(
'pylib modules should not be imported from devil modules.',
items=errors)
]
return []
def CommonChecks(input_api, output_api):
output = []
output += _RunPylint(input_api, output_api)
output += _RunUnitTests(input_api, output_api)
output += _EnsureNoPylibUse(input_api, output_api)
return output
def CheckChangeOnUpload(input_api, output_api):
return CommonChecks(input_api, output_api)
def CheckChangeOnCommit(input_api, output_api):
return CommonChecks(input_api, output_api)