Wait until populated with non-empty value

Author: ax3l

tests/python/dashboard/test_python_import.py

@@ -24,6 +24,10 @@ def test_python_import(dashboard):
     """
     dashboard.load_example("testdata/example.py", manual=True)
 
+    # Wait for the file to finish loading and processing
+    # The importing_file state is True while loading, False when done
+    dashboard.assert_state("importing_file", False)
+
     BEAM_PARAMETERS = [
         ("tracking_mode", "Particle Tracking"),
         ("space_charge", "false"),
@@ -76,6 +80,10 @@ def test_python_import(dashboard):
         # This is important for CI environments where rendering may be slower
         dashboard.sb.wait_for_element_present(element_id, timeout=10)
 
+        # Wait for the element's value to be populated
+        # DOM elements may take time to reflect updated state values after file loading
+        dashboard.wait_for_element_value(element_id, timeout=10)
+
         # Get the value attribute - this works even if the element isn't visible
         actual_value = float(dashboard.sb.get_attribute(element_id, "value"))
         assert actual_value == pytest.approx(expected_value, **APPROX_TOL), (

tests/python/dashboard/utils.py

@@ -275,6 +275,39 @@ def get_state(self, state_name):
         """
         return self.sb.execute_script(js_script, state_name)
 
+    def wait_for_element_value(self, element_id: str, timeout=TIMEOUT):
+        """
+        Wait for an element's value attribute to be populated (non-empty).
+
+        This is useful after loading files, as DOM elements may take time to
+        reflect the updated state values.
+
+        :param element_id: ID of the input element to check (with or without # prefix).
+        :param timeout: Maximum time to wait in seconds.
+        """
+        for i in range(timeout):
+            try:
+                value = self.sb.get_attribute(element_id, "value")
+                if value and value.strip():  # Non-empty value
+                    return value
+            except Exception:
+                pass
+
+            time.sleep(1)
+
+        # If we get here, try one more time to get the value (even if empty)
+        # to provide a better error message
+        try:
+            final_value = self.sb.get_attribute(element_id, "value")
+            raise TimeoutError(
+                f"Element '{element_id}' value never populated after {timeout} seconds "
+                f"(last value: '{final_value}')"
+            )
+        except Exception as e:
+            raise TimeoutError(
+                f"Element '{element_id}' value never populated after {timeout} seconds"
+            ) from e
+
 
 def save_failure_screenshot(
     dashboard, request, directory: str | None = None