fix(smoke-tests) Increases sleep timeout in rollback test to prevent …

…flakiness (#4979)
datahub-project · May 24, 2022 · 483f66e · 483f66e
1 parent 669160a
commit 483f66e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/smoke-test/tests/cli/datahub-cli.py b/smoke-test/tests/cli/datahub-cli.py
@@ -30,7 +30,7 @@ def test_setup():
 
     ingested_dataset_run_id = ingest_file_via_rest("tests/cli/cli_test_data.json").config.run_id
     print("Setup ingestion id: " + ingested_dataset_run_id)
-    sleep(2)
+    sleep(5)
 
     assert "browsePaths" in get_aspects_for_entity(entity_urn=dataset_urn, aspects=["browsePaths"], typed=False)
 
@@ -41,7 +41,7 @@ def test_setup():
 
     session.post(rollback_url, data=json.dumps({"runId": ingested_editable_run_id, "dryRun": False, "hardDelete": True}))
     session.post(rollback_url, data=json.dumps({"runId": ingested_dataset_run_id, "dryRun": False, "hardDelete": True}))
-    sleep(2)
+    sleep(5)
 
     assert "browsePaths" not in get_aspects_for_entity(entity_urn=dataset_urn, aspects=["browsePaths"], typed=False)
     assert "editableDatasetProperties" not in get_aspects_for_entity(entity_urn=dataset_urn, aspects=["editableDatasetProperties"], typed=False)
@@ -75,7 +75,7 @@ def test_rollback_editable():
     session.post(rollback_url, data=json.dumps({"runId": ingested_dataset_run_id, "dryRun": False, "hardDelete": False}))
 
     # Allow async MCP processor to handle ingestions & rollbacks
-    sleep(2)
+    sleep(5)
 
     # EditableDatasetProperties should still be part of the entity that was soft deleted.
     assert "editableDatasetProperties" in get_aspects_for_entity(entity_urn=dataset_urn, aspects=["editableDatasetProperties"], typed=False)