Remove sleep between creating and exporting a batch (#1213)

Author: attila-papai

tests/integration/export_v2/test_legacy_export.py

@@ -171,7 +171,6 @@ def test_export_data_rows(project: Project, dataset: Dataset):
 
     data_rows = [dr.uid for dr in list(dataset.export_data_rows())]
     batch = project.create_batch("batch test", data_rows)
-
     result = list(batch.export_data_rows())
     exported_data_rows = [dr.uid for dr in result]
 

tests/integration/test_batch.py

@@ -211,8 +211,6 @@ def test_export_data_rows(project: Project, dataset: Dataset):
 
     data_rows = [dr.uid for dr in list(dataset.export_data_rows())]
     batch = project.create_batch("batch test", data_rows)
-    # allow time for catapult to sync changes to ES
-    time.sleep(5)
     result = list(batch.export_data_rows())
     exported_data_rows = [dr.uid for dr in result]
 

tests/integration/test_project.py

@@ -228,8 +228,6 @@ def test_create_batch_with_global_keys_sync(project: Project, data_rows):
     global_keys = [dr.global_key for dr in data_rows]
     batch_name = f'batch {uuid.uuid4()}'
     batch = project.create_batch(batch_name, global_keys=global_keys)
-    # allow time for catapult to sync changes to ES
-    time.sleep(5)
     # TODO: Move to export_v2
     batch_data_rows = set(batch.export_data_rows())
     assert batch_data_rows == set(data_rows)