Fix tests

wri · Jul 31, 2024 · 9830d30 · 9830d30
1 parent 71dc93b
commit 9830d30
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/tests_v2/unit/app/routes/datasets/test_query.py b/tests_v2/unit/app/routes/datasets/test_query.py
@@ -682,6 +682,7 @@ async def test_query_batch_feature_collection(
     payload = {
         "sql": "select count(*) from data",
         "feature_collection": FEATURE_COLLECTION,
+        "id_field": "id",
     }
 
     response = await async_client.post(
@@ -727,6 +728,7 @@ async def test_query_batch_uri(
     payload = {
         "sql": "select count(*) from data",
         "uri": "s3://path/to/files",
+        "id_field": "id",
     }
 
     response = await async_client.post(

diff --git a/tests_v2/unit/app/routes/jobs/test_job.py b/tests_v2/unit/app/routes/jobs/test_job.py
@@ -9,7 +9,7 @@
 TEST_JOB_ID = "f3caa6c8-09d7-43a8-823f-e7528344a169"
 
 
-def _get_sfn_execution_mocked_pending(job_id):
+async def _get_sfn_execution_mocked_pending(job_id):
     return {
         "executionArn": "arn::fake_execution_arn",
         "stateMachineArn": "arn::fake_state_machine_arn",
@@ -20,7 +20,7 @@ def _get_sfn_execution_mocked_pending(job_id):
     }
 
 
-def _get_sfn_execution_mocked_success(job_id):
+async def _get_sfn_execution_mocked_success(job_id):
     return {
         "executionArn": "arn::fake_execution_arn",
         "stateMachineArn": "arn::fake_state_machine_arn",
@@ -36,7 +36,7 @@ def _get_sfn_execution_mocked_success(job_id):
     }
 
 
-def _get_sfn_execution_mocked_failed(job_id):
+async def _get_sfn_execution_mocked_failed(job_id):
     return {
         "executionArn": "arn::fake_execution_arn",
         "stateMachineArn": "arn::fake_state_machine_arn",
@@ -47,7 +47,7 @@ def _get_sfn_execution_mocked_failed(job_id):
     }
 
 
-def _get_map_run_mocked(job_id):
+async def _get_map_run_mocked(job_id):
     return {
         "executionArn": "arn::fake_execution_arn",
         "mapRunArn": "arn::fake_map_run_arn",