[U-Boot] [PATCH v2] test/py: Support setting up specific timeout

[Stephen Warren <swarren@wwwdotorg.org>]

On 05/18/2016 11:00 AM, 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(...)

> diff --git a/test/py/u_boot_console_base.py b/test/py/u_boot_console_base.py

> +class ConsoleSetupTimeout(object):

> +    def __init__(self, console, timeout):
> +        self.p = console.p
> +        self.orig_timeout = self.p.timeout
> +        self.p.timeout = timeout

> +    def __exit__(self, extype, value, traceback):
> +        if not self.p:
> +            return

That test can't fail, since __init__ already used self.p in a way that 
would have triggered an exception during the constructor, which I 
believe would cause neither __enter__ nor __exit__ to ever be called 
since the object would not exist.

Still, this does no harm, so either way,
Reviewed-by: Stephen Warren <swarren@nvidia.com>