Add test_switch_to_workspaces_widget helper

Scille · Jul 21, 2020 · ec7e095 · ec7e095
1 parent a5d53fd
commit ec7e095
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/core/gui/conftest.py b/tests/core/gui/conftest.py
@@ -362,8 +362,17 @@ async def test_switch_to_users_widget(error=False):
             await aqtbot.mouse_click(central_widget.menu.button_users, QtCore.Qt.LeftButton)
         return u_w
 
+    async def test_switch_to_workspaces_widget(error=False):
+        central_widget = gui.test_get_central_widget()
+        f_w = gui.test_get_workspaces_widget()
+        signal = f_w.list_error if error else f_w.list_success
+        async with aqtbot.wait_exposed(f_w), aqtbot.wait_signal(signal):
+            await aqtbot.mouse_click(central_widget.menu.button_files, QtCore.Qt.LeftButton)
+        return f_w
+
     gui.test_switch_to_devices_widget = test_switch_to_devices_widget
     gui.test_switch_to_users_widget = test_switch_to_users_widget
+    gui.test_switch_to_workspaces_widget = test_switch_to_workspaces_widget
 
     return gui