forked from PostHog/drf-exceptions-hog
-
Notifications
You must be signed in to change notification settings - Fork 0
/
runtests.py
141 lines (105 loc) · 3.42 KB
/
runtests.py
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
"""
Forked from https://github.com/encode/django-rest-framework
"""
import subprocess
import sys
import pytest
PYTEST_ARGS = {
"default": [],
}
PROJECT = ["exceptions_hog", "tests", "test_project"]
FLAKE8_ARGS = PROJECT
ISORT_ARGS = ["--check-only", "--diff"] + PROJECT
BLACK_ARGS = ["--check"] + PROJECT
MYPY_ARGS = PROJECT
def exit_on_failure(ret, message=None):
if ret:
sys.exit(ret)
def flake8_main(args):
print("Running flake8 code linting")
ret = subprocess.call(["flake8"] + args)
print("❗️ flake8 failed" if ret else "✅ flake8 passed")
return ret
def isort_main(args):
print("Running isort code checking")
ret = subprocess.call(["isort"] + args)
if ret:
print("❗️ isort failed: Fix by running `isort --recursive .`")
else:
print("✅ isort passed")
return ret
def black_main(args):
print("Running black format checking")
ret = subprocess.call(["black"] + args)
print("❗️ black failed") if ret else print("✅ black passed")
return ret
def mypy_main(args):
print("Running mypy typechecking")
ret = subprocess.call(["mypy"] + args)
print("❗️ mypy failed") if ret else print("✅ mypy passed")
return ret
def split_class_and_function(string):
class_string, function_string = string.split(".", 1)
return "%s and %s" % (class_string, function_string)
def is_function(string):
# `True` if it looks like a test function is included in the string.
return string.startswith("test_") or ".test_" in string
def is_class(string):
# `True` if first character is uppercase - assume it's a class name.
return string[0] == string[0].upper()
if __name__ == "__main__":
try:
sys.argv.remove("--no-lint")
except ValueError:
run_flake8 = True
run_isort = True
run_black = True
run_mypy = True
else:
run_flake8 = False
run_isort = False
run_black = False
run_mypy = False
try:
sys.argv.remove("--lint-only")
except ValueError:
run_tests = True
else:
run_tests = False
if len(sys.argv) > 1:
pytest_args = sys.argv[1:]
first_arg = pytest_args[0]
try:
pytest_args.remove("--coverage")
except ValueError:
pass
else:
pytest_args = [
"--cov",
".",
"--cov-report",
"xml",
] + pytest_args
if first_arg.startswith("-"):
# `runtests.py [flags]`
pytest_args = ["tests"] + pytest_args
elif is_class(first_arg) and is_function(first_arg):
# `runtests.py TestCase.test_function [flags]`
expression = split_class_and_function(first_arg)
pytest_args = ["tests", "-k", expression] + pytest_args[1:]
elif is_class(first_arg) or is_function(first_arg):
# `runtests.py TestCase [flags]`
# `runtests.py test_function [flags]`
pytest_args = ["tests", "-k", pytest_args[0]] + pytest_args[1:]
else:
pytest_args = [""]
if run_flake8:
exit_on_failure(flake8_main(FLAKE8_ARGS))
if run_isort:
exit_on_failure(isort_main(ISORT_ARGS))
if run_black:
exit_on_failure(black_main(BLACK_ARGS))
if run_mypy:
exit_on_failure(mypy_main(MYPY_ARGS))
if run_tests:
exit_on_failure(pytest.main(pytest_args))