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
|
import os
import string
import random
import subprocess
import time
import tempfile
from devbot import config
from devbot import command
def run(cmd):
args = [cmd, "--home-dir", config.home_dir]
resolution = config.get_pref("RESOLUTION")
if resolution:
args.extend(["--resolution", resolution])
output = config.get_pref("OUTPUT")
if output:
args.extend(["--output", output])
command.run(args)
def run_test(test_cmd, test_path):
temp_dir = tempfile.mkdtemp("devbot-test-display")
display_path = os.path.join(temp_dir, "display")
args = [test_cmd,
"--display-path", display_path,
"--virtual"]
output_fd, output_name = tempfile.mkstemp("devbot-test-output")
output_file = os.fdopen(output_fd)
test_cmd_process = subprocess.Popen(args,
stdout=output_file,
stderr=subprocess.STDOUT)
while True:
if not os.path.exists(display_path):
time.sleep(1)
else:
break
with open(display_path) as f:
os.environ["DISPLAY"] = f.read()
os.unlink(display_path)
os.rmdir(temp_dir)
try:
command.run(["python", "-u", test_path])
result = True
except subprocess.CalledProcessError:
result = False
test_cmd_process.terminate()
output_file.seek(0)
with open(config.log_path, "a") as f:
f.write(output_file.read())
output_file.close()
os.unlink(output_name)
return result
def collect_logs(source_path):
logs = {}
for filename in os.listdir(source_path):
if filename.endswith(".log"):
path = os.path.join(source_path, filename)
with open(path) as f:
logs[filename] = f.read()
with open(config.log_path, "a") as f:
for filename, log in logs.items():
f.write("\n===== %s =====\n\n%s" % (filename, log))
def _get_random_id():
return ''.join(random.choice(string.letters) for i in xrange(8))
|
