
On 18.5.2016 15:11, Michal Simek wrote:
Large file transfers, flash erasing and more complicated tests requires more time to finish. Provide a way to setup specific timeout directly in test.
For example description for 50s test: timeout = 50000 with u_boot_console.temporary_timeout(timeout): u_boot_console.run_command(...)
Signed-off-by: Michal Simek michal.simek@xilinx.com
test/py/u_boot_console_base.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+)
diff --git a/test/py/u_boot_console_base.py b/test/py/u_boot_console_base.py index f7434363fbcb..82bbba43463e 100644 --- a/test/py/u_boot_console_base.py +++ b/test/py/u_boot_console_base.py @@ -56,6 +56,24 @@ class ConsoleDisableCheck(object): self.console.disable_check_count[self.check_type] -= 1 self.console.eval_bad_patterns()
+class ConsoleSetupTimeout(object):
- """Context manager (for Python's with statement) that temporarily setup
- timeout for specific comnand. This is useful when execution time is greater
- then default 30s."""
- def __init__(self, console, timeout):
self.console = console
self.console.orig_timeout = self.console.p.timeout
self.console.p.timeout = timeout
- def __enter__(self):
return self
- def __exit__(self, extype, value, traceback):
self.console.p = self.console.get_spawn()
This line should be there. It is resetting console. I will do more testing on real hw not on qemu.
Thanks, Michal
self.console.p.timeout = self.console.orig_timeout
class ConsoleBase(object): """The interface through which test functions interact with the U-Boot console. This primarily involves executing shell commands, capturing their @@ -391,3 +409,18 @@ class ConsoleBase(object): """
return ConsoleDisableCheck(self, check_type)
- def temporary_timeout(self, timeout):
"""Temporarily setup different timeout for commands.
Create a new context manager (for use with the "with" statement) which
temporarily change timeout.
Args:
timeout: Time in miliseconds.
Returns:
A context manager object.
"""
return ConsoleSetupTimeout(self, timeout)