90 lines
2.4 KiB
Plaintext
90 lines
2.4 KiB
Plaintext
|
#!/usr/bin/env python2
|
||
|
|
|
||
|
|
# Copyright 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 os
|
||
|
|
import pexpect
|
||
|
|
import signal
|
||
|
|
import subprocess
|
||
|
|
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]
|
||
|
|
|
||
|
|
PS_ARGS = ['ps', '--no-headers', '-o', 'stat', '--pid']
|
||
|
|
|
||
|
|
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, env):
|
||
|
|
child = pexpect.spawn('build/host/{0}/{0}.exe'.format(test_name),
|
||
|
|
timeout=TIMEOUT, env=env)
|
||
|
|
child.logfile = log
|
||
|
|
try:
|
||
|
|
return child.expect(EXPECT_LIST)
|
||
|
|
finally:
|
||
|
|
if child.isalive():
|
||
|
|
ps_cmd = PS_ARGS + ['%d' % (child.pid)]
|
||
|
|
ps_stat = subprocess.check_output(ps_cmd).strip()
|
||
|
|
log.write('\n*** test %s process %d in state %s ***\n' %
|
||
|
|
(test_name, child.pid, ps_stat))
|
||
|
|
child.kill(signal.SIGTERM)
|
||
|
|
child.read()
|
||
|
|
|
||
|
|
# ASAN_OPTIONS environment variable is only required when test is built with
|
||
|
|
# ASan, but is otherwise harmless.
|
||
|
|
env = dict(os.environ)
|
||
|
|
env["ASAN_OPTIONS"] = "log_path=stderr"
|
||
|
|
|
||
|
|
log = StringIO()
|
||
|
|
tee_log = Tee(log)
|
||
|
|
test_name = sys.argv[1]
|
||
|
|
start_time = time.time()
|
||
|
|
|
||
|
|
result_id = RunOnce(test_name, tee_log, env)
|
||
|
|
|
||
|
|
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)
|