1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
#!/usr/bin/env python3
# Copyright 2019 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Integration tests for the config module (via PREUPLOAD.cfg)."""
import argparse
import os
import re
import sys
REPOTOOLS = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
REPO_ROOT = os.path.dirname(os.path.dirname(REPOTOOLS))
def assertEqual(msg, exp, actual):
"""Assert |exp| equals |actual|."""
assert exp == actual, f'{msg}: expected "{exp}" but got "{actual}"'
def assertEnv(var, value):
"""Assert |var| is set in the environment as |value|."""
assert var in os.environ, f'${var} missing in environment'
assertEqual(f'env[{var}]', value, os.environ[var])
def check_commit_id(commit):
"""Check |commit| looks like a git commit id."""
assert len(commit) == 40, f'commit "{commit}" must be 40 chars'
assert re.match(r'^[a-f0-9]+$', commit), \
f'commit "{commit}" must be all hex'
def check_commit_msg(msg):
"""Check the ${PREUPLOAD_COMMIT_MESSAGE} setting."""
assert len(msg) > 1, f'commit message must be at least 2 bytes: {msg}'
def check_repo_root(root):
"""Check the ${REPO_ROOT} setting."""
assertEqual('REPO_ROOT', REPO_ROOT, root)
def check_files(files):
"""Check the ${PREUPLOAD_FILES} setting."""
assert files
def check_env():
"""Verify all exported env vars look sane."""
assertEnv('REPO_PROJECT', 'platform/tools/repohooks')
assertEnv('REPO_PATH', 'tools/repohooks')
assertEnv('REPO_REMOTE', 'aosp')
check_commit_id(os.environ['REPO_LREV'])
print(os.environ['REPO_RREV'])
check_commit_id(os.environ['PREUPLOAD_COMMIT'])
def get_parser():
"""Return a command line parser."""
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('--check-env', action='store_true',
help='Check all exported env vars.')
parser.add_argument('--commit-id',
help='${PREUPLOAD_COMMIT} setting.')
parser.add_argument('--commit-msg',
help='${PREUPLOAD_COMMIT_MESSAGE} setting.')
parser.add_argument('--repo-root',
help='${REPO_ROOT} setting.')
parser.add_argument('files', nargs='+',
help='${PREUPLOAD_FILES} paths.')
return parser
def main(argv):
"""The main entry."""
parser = get_parser()
opts = parser.parse_args(argv)
try:
if opts.check_env:
check_env()
if opts.commit_id is not None:
check_commit_id(opts.commit_id)
if opts.commit_msg is not None:
check_commit_msg(opts.commit_msg)
if opts.repo_root is not None:
check_repo_root(opts.repo_root)
check_files(opts.files)
except AssertionError as e:
print(f'error: {e}', file=sys.stderr)
return 1
return 0
if __name__ == '__main__':
sys.exit(main(sys.argv[1:]))
|
