blob: c130b822fbb77ad226c78ac86f74ea022160a295 (
plain)
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
|
#!/usr/bin/env python
# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
from cStringIO import StringIO
import pexpect
import signal
import sys
import time
TIMEOUT=10
RESULT_ID_TIMEOUT = 0
RESULT_ID_PASS = 1
RESULT_ID_FAIL = 2
RESULT_ID_EOF = 3
EXPECT_LIST = [pexpect.TIMEOUT, 'Pass!', 'Fail!', pexpect.EOF]
class Tee(object):
def __init__(self, target):
self._target = target
def write(self, data):
sys.stdout.write(data)
self._target.write(data)
def flush(self):
sys.stdout.flush()
self._target.flush()
def RunOnce(test_name, log):
child = pexpect.spawn('build/host/{0}/{0}.exe'.format(test_name),
timeout=TIMEOUT)
child.logfile = log
try:
return child.expect(EXPECT_LIST)
finally:
if child.isalive():
child.kill(signal.SIGTERM)
child.read()
log = StringIO()
tee_log = Tee(log)
test_name = sys.argv[1]
start_time = time.time()
result_id = RunOnce(test_name, tee_log)
elapsed_time = time.time() - start_time
if result_id == RESULT_ID_TIMEOUT:
sys.stderr.write('Test %s timed out after %d seconds!\n' %
(test_name, TIMEOUT))
failed = True
elif result_id == RESULT_ID_PASS:
sys.stderr.write('Test %s passed! (%.3f seconds)\n' %
(test_name, elapsed_time))
failed = False
elif result_id == RESULT_ID_FAIL:
sys.stderr.write('Test %s failed! (%.3f seconds)\n' %
(test_name, elapsed_time))
failed = True
elif result_id == RESULT_ID_EOF:
sys.stderr.write('Test %s terminated unexpectedly! (%.3f seconds)\n' %
(test_name, elapsed_time))
failed = True
if failed:
sys.stderr.write('\n====== Emulator output ======\n')
sys.stderr.write(log.getvalue())
sys.stderr.write('\n=============================\n')
sys.exit(1)
else:
sys.exit(0)
|
