unittests: stabilize TaskbarTestCases.testClickVisibleIcon

Fulmo · Jan 17, 2020 · 977f0a0 · 977f0a0
1 parent 3352a63
commit 977f0a0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pywinauto/unittests/test_taskbar.py b/pywinauto/unittests/test_taskbar.py
@@ -257,7 +257,12 @@ def testClickVisibleIcon(self):
         def _show_popup_menu():
             taskbar.explorer_app.wait_cpu_usage_lower(threshold=5, timeout=self.tm)
             taskbar.RightClickSystemTrayIcon('MFCTrayDemo')
-            menu = self.app.top_window().children()[0]
+            children = self.app.top_window().children()
+            if not children:
+                # Somehow, we ended up with a dialog - parent of the menu
+                menu = self.app.windows(visible=True)[0].children()[0]
+            else:
+                menu = children[0]
             res = isinstance(menu, ToolbarWrapper) and menu.is_visible()
             menu_window[0] = menu
             return res