changed the shock checkbox test

magnus-haw · Mar 18, 2024 · 13a552f · 13a552f
1 parent 36deb0f
commit 13a552f
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tests/test_gui.py b/tests/test_gui.py
@@ -89,12 +89,22 @@ def test_toggle_show_shock_checkbox(app, qtbot):
     """
     assert app.ui.checkBox_display_shock.isChecked()
 
+    # Click on the checkbox to toggle it off
     qtbot.mouseClick(app.ui.checkBox_display_shock, Qt.LeftButton)
-    qtbot.wait(100)  # event might be asynchronous
+
+    # Wait for the checkbox state to update
+    qtbot.waitUntil(lambda: not app.ui.checkBox_display_shock.isChecked())
+
+    # Assert that the checkbox is now unchecked
     assert not app.ui.checkBox_display_shock.isChecked()
 
+    # Click on the checkbox again to toggle it on
     qtbot.mouseClick(app.ui.checkBox_display_shock, Qt.LeftButton)
-    qtbot.wait(100)
+
+    # Wait for the checkbox state to update
+    qtbot.waitUntil(lambda: app.ui.checkBox_display_shock.isChecked())
+
+    # Assert that the checkbox is now checked
     assert app.ui.checkBox_display_shock.isChecked()