1
0
mirror of https://github.com/RIOT-OS/RIOT.git synced 2024-12-29 04:50:03 +01:00
RIOT/dist/pythonlibs/testrunner/__init__.py
sanjuprakashk d30bd7ae47 dist/pythinlibs/testrunner: dynamic changing of test timeout
Few flashers require a longer time to flash code and reset the device and
requires the test script to wait for a longer time before timing out.
This commit adds a environment variable "RIOT_TEST_TIMEOUT"  that can be
set by the user to vary the timeout in accordance to the requirements
of the system

on-behalf-of: @sparkmeter <sanju.kannioth@sparkmeter.io>
2019-07-17 06:25:28 -07:00

46 lines
1.5 KiB
Python
Executable File

# Copyright (C) 2018-19 Freie Universität Berlin
# 2017 Cenk Gündoğan <cenk.guendogan@haw-hamburg.de>
# 2016 Kaspar Schleiser <kaspar@schleiser.de>
# 2014 Martine Lenders <mlenders@inf.fu-berlin.de>
#
# This file is subject to the terms and conditions of the GNU Lesser
# General Public License v2.1. See the file LICENSE in the top level
# directory for more details.
import os
import sys
from traceback import print_tb
import pexpect
from .spawn import find_exc_origin, setup_child, teardown_child
from .unittest import PexpectTestCase # noqa, F401 expose to users
# Timeout for tests can be changed by setting RIOT_TEST_TIMEOUT to the desired
# value in the environment variables
# default value (10)
TIMEOUT = int(os.environ.get('RIOT_TEST_TIMEOUT') or 10)
def run(testfunc, timeout=TIMEOUT, echo=True, traceback=False):
child = setup_child(timeout, env=os.environ,
logfile=sys.stdout if echo else None)
try:
testfunc(child)
except pexpect.TIMEOUT:
trace = find_exc_origin(sys.exc_info()[2])
print("Timeout in expect script at \"%s\" (%s:%d)" % trace)
if traceback:
print_tb(sys.exc_info()[2])
return 1
except pexpect.EOF:
trace = find_exc_origin(sys.exc_info()[2])
print("Unexpected end of file in expect script at \"%s\" (%s:%d)" %
trace)
if traceback:
print_tb(sys.exc_info()[2])
return 1
finally:
print("")
teardown_child(child)
return 0