From 7a8f8865582583314c884e77a607f5b9dc9cf8b6 Mon Sep 17 00:00:00 2001 From: Stephen Warren Date: Thu, 16 Jun 2016 12:59:34 -0600 Subject: test/py: fix printenv signon message disable code CONFIG_VERSION_VARIABLE isn't always defined, so we can't simply look up its value directly, or an exception will occur if it isn't defined. Instead, we must use .get() to supply a default value if the variable isn't defined. Fixes: da37f006e7c5 ("tests: py: disable main_signon check for printenv cmd") Acked-by: Heiko Schocher Signed-off-by: Stephen Warren --- test/py/tests/test_env.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/py/tests/test_env.py b/test/py/tests/test_env.py index 22a22d1d53..035dbf5cac 100644 --- a/test/py/tests/test_env.py +++ b/test/py/tests/test_env.py @@ -39,7 +39,8 @@ class StateTestEnv(object): Nothing. """ - if self.u_boot_console.config.buildconfig['config_version_variable'] == 'y': + if self.u_boot_console.config.buildconfig.get( + 'config_version_variable', 'n') == 'y': with self.u_boot_console.disable_check('main_signon'): response = self.u_boot_console.run_command('printenv') else: -- cgit