Files
OpenCellular/util/run_host_test
Vic Yang e1c0b58c96 Show run time of unit tests
This is useful for finding out which test runs for too long.

BUG=chrome-os-partner:19235
TEST=Run tests and see run time logged.
BRANCH=None

Change-Id: I5cb6727b2e1017fce1107e4892c1898e66598492
Signed-off-by: Vic Yang <victoryang@chromium.org>
Reviewed-on: https://gerrit.chromium.org/gerrit/51105
Reviewed-by: Vincent Palatin <vpalatin@chromium.org>
2013-05-14 09:24:59 -07:00

54 lines
1.4 KiB
Python
Executable File

#!/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 sys
import time
TIMEOUT=10
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()
log = StringIO()
test_name = sys.argv[1]
start_time = time.time()
child = pexpect.spawn('build/host/{0}/{0}.exe'.format(test_name),
timeout=TIMEOUT)
child.logfile = Tee(log)
result_id = child.expect([pexpect.TIMEOUT, 'Pass!', 'Fail!'])
elapsed_time = time.time() - start_time
failed = False
if result_id == 0:
sys.stderr.write('Test %s timed out after %d seconds!\n' %
(test_name, TIMEOUT))
failed = True
elif result_id == 1:
sys.stderr.write('Test %s passed! (%.3f seconds)\n' %
(test_name, elapsed_time))
elif result_id == 2:
sys.stderr.write('Test %s failed! (%.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)