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
|
#!/usr/bin/env python3
# Copyright 2024 Huawei Cloud Computing Technology Co., Ltd.
#
# 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.
import json
import os
import subprocess
import time
time_start: float = 0
time_stop: float = 0
result: str = "UNKNOWN"
stderr: str = None
stdout: str = None
def end(x):
return "\n" if x else ""
def dump_results() -> None:
global time_start, time_stop, result, stdout, stderr
with open("result", "w") as f:
print(f"{result}", file=f)
with open("time-start", "w") as f:
print(f"{time_start:.3f}", file=f)
with open("time-stop", "w") as f:
print(f"{time_stop:.3f}", file=f)
with open("stdout", "w") as f:
print(f"{stdout}", file=f, end=end(stdout))
with open("stderr", "w") as f:
print(f"{stderr}", file=f, end=end(stderr))
class dumper:
def __enter__(self):
global time_start
time_start = time.time()
def __exit__(self, type, value, traceback):
dump_results()
with dumper():
TEMP_DIR = os.path.realpath("scratch")
os.makedirs(TEMP_DIR, exist_ok=True)
WORK_DIR = os.path.realpath("work")
os.makedirs(WORK_DIR, exist_ok=True)
ENV = dict(os.environ, TEST_TMPDIR=TEMP_DIR, TMPDIR=TEMP_DIR, TERM="xterm-256color")
with open("test-launcher.json") as f:
test_launcher = json.load(f)
with open("test-args.json") as f:
test_args = json.load(f)
ret = subprocess.run(
test_launcher + ["../test", "--color", "always"] + test_args,
cwd=WORK_DIR,
env=ENV,
capture_output=True,
)
time_stop = time.time()
result = "PASS" if ret.returncode == 0 else "FAIL"
stdout = ret.stdout.decode("utf-8")
stderr = ret.stderr.decode("utf-8")
if result != "PASS":
exit(1)
|
