commit 7a8f8865582583314c884e77a607f5b9dc9cf8b6
author Stephen Warren <swarren@nvidia.com> Thu Jun 16 12:59:34 2016 -0600
committer Tom Rini <trini@konsulko.com> Sun Jun 19 09:50:56 2016 -0400
tree 8590a23bb295a46757cc08fc2fc5f03b0110ffec
parent 6beacfcff81bdc04c10a440971b0fb683ee57534

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 <hs@denx.de>
Signed-off-by: Stephen Warren <swarren@nvidia.com>

test/py/tests/test_env.py

diff --git a/test/py/tests/test_env.py b/test/py/tests/test_env.py
index 22a22d1..035dbf5 100644
--- a/test/py/tests/test_env.py
+++ b/test/py/tests/test_env.py
@@ -39,7 +39,8 @@
             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: