summaryrefslogtreecommitdiff
diff options
authorŁukasz 'sil2100' Zemczak <lukasz.zemczak@canonical.com>2013-05-21 14:16:54 +0200
committerŁukasz 'sil2100' Zemczak <lukasz.zemczak@canonical.com>2013-05-21 14:16:54 +0200
commitaced6268f63a2f6a0e85295fe1703ef8e2cb1ec6 (patch)
tree262a83636828c9bc2b4c0d9841fe71b755710524
parentff7c1c0c9b59d02c46f9246e71b2ab8d2ba836d9 (diff)
Same fix for the test_search HUD tests
(bzr r3335.1.4)
-rw-r--r--tests/autopilot/unity/tests/test_search.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/autopilot/unity/tests/test_search.py b/tests/autopilot/unity/tests/test_search.py
index 77d6caa57..a7856d9b5 100644
--- a/tests/autopilot/unity/tests/test_search.py
+++ b/tests/autopilot/unity/tests/test_search.py
@@ -146,7 +146,12 @@ class HudSearchTestBase(SearchTestsBase):
def input_and_check_result(self, string, expected):
self.keyboard.type(string)
self.assertThat(self.unity.hud.search_string, Eventually(Equals(string), timeout=30))
- hud_query_check = lambda: self.unity.hud.selected_hud_button.label_no_formatting
+ def hud_query_check():
+ button = self.unity.hud.selected_hud_button
+ if not button:
+ return
+ return button.label_no_formatting
+
self.assertThat(hud_query_check, Eventually(Equals(expected), timeout=30))