robustify store test_data_lifecycle

Author: alexcjohnson

tests/integration/store/test_data_lifecycle.py

@@ -1,3 +1,6 @@
+import dash.testing.wait as wait
+
+
 def test_stdl001_data_lifecycle_with_different_condition(store_app, dash_dcc):
     dash_dcc.start_server(store_app)
 
@@ -31,7 +34,7 @@ def test_stdl001_data_lifecycle_with_different_condition(store_app, dash_dcc):
     ), "memory storage should contain the initial data in new tab"
 
     dash_dcc.multiple_click("#btn", 2)
-    assert dash_dcc.get_session_storage() == {"n_clicks": 2}
+    wait.until(lambda: dash_dcc.get_session_storage() == {"n_clicks": 2}, timeout=1)
     assert (
         '"n_clicks": 2' in dash_dcc.wait_for_element("#output").text
     ), "memory storage should reflect to the new clicks"