test_taskbar: verify combobox in _toggle_notification_area_icons

adevdash · Feb 20, 2017 · aa766ae · aa766ae
1 parent 22ad69f
commit aa766ae
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/pywinauto/unittests/test_taskbar.py b/pywinauto/unittests/test_taskbar.py
@@ -96,14 +96,14 @@ def _cabinetwclass_exist():
             explorer.WaitCPUUsageLower(threshold=2, timeout=_ready_timeout)
             window.type_keys(cmd_str, with_spaces=True, set_foreground=True)
             # verfiy the text in the address combobox after type_keys finished
-            texts = window.AddressBandRoot.ComboBoxEx.texts()
-            if texts and texts[0] == cmd_str:
-                break
-            else:
-                l = pywinauto.actionlogger.ActionLogger()
-                l.log(texts)
-                # Send ESCs to remove the invalid text
-                window.type_keys("{ESC}" * 3)
+            cmbx_spec = window.AddressBandRoot.ComboBoxEx
+            if cmbx_spec.exists(timeout=_ready_timeout, retry_interval=_retry_interval):
+                texts = cmbx_spec.texts()
+                if texts and texts[0] == cmd_str:
+                    break
+            # Send ESCs to remove the invalid text
+            window.type_keys("{ESC}" * 3)
+
         # Send 'ENTER' separately, this is to make sure
         # the window focus hasn't accidentally been lost
         window.type_keys(