forked from microsoft/vscode-python
/
__main__.py
56 lines (42 loc) · 1.49 KB
/
__main__.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
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
import argparse
import sys
import pytest
from . import DEBUG_ADAPTER_ROOT, SRC_ROOT, TEST_ROOT, TESTING_TOOLS_ROOT
def parse_args():
parser = argparse.ArgumentParser()
# To mark a test as functional: (decorator) @pytest.mark.functional
parser.add_argument("--functional", dest="markers", action="append_const", const="functional")
parser.add_argument(
"--no-functional", dest="markers", action="append_const", const="not functional"
)
args, remainder = parser.parse_known_args()
ns = vars(args)
if remainder:
for arg in remainder:
if arg.startswith("-") and arg not in ("-v", "--verbose", "-h", "--help"):
specific = False
break
else:
specific = True
else:
specific = False
args.specific = specific
return ns, remainder
def main(pytestargs, markers=None, specific=False):
sys.path.insert(1, TESTING_TOOLS_ROOT)
sys.path.insert(1, DEBUG_ADAPTER_ROOT)
if not specific:
pytestargs.insert(0, TEST_ROOT)
pytestargs.insert(0, "--rootdir")
pytestargs.insert(1, SRC_ROOT)
for marker in reversed(markers or ()):
pytestargs.insert(0, marker)
pytestargs.insert(0, "-m")
ec = pytest.main(pytestargs)
return ec
if __name__ == "__main__":
mainkwargs, pytestargs = parse_args()
ec = main(pytestargs, **mainkwargs)
sys.exit(ec)