diff --git a/tests/integration/export_v2/test_legacy_export.py b/tests/integration/export_v2/test_legacy_export.py index 071e8254d..e4e1e595c 100644 --- a/tests/integration/export_v2/test_legacy_export.py +++ b/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] diff --git a/tests/integration/test_batch.py b/tests/integration/test_batch.py index 485bf308c..ad1c9eb89 100644 --- a/tests/integration/test_batch.py +++ b/tests/integration/test_batch.py @@ -240,8 +240,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] diff --git a/tests/integration/test_project.py b/tests/integration/test_project.py index 20b36533a..b9467e0dd 100644 --- a/tests/integration/test_project.py +++ b/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)