Add missing awaits

inngest · Mar 5, 2025 · a9b1fca · a9b1fca
1 parent 0a87273
commit a9b1fca
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_inngest_encryption/cases/step_and_fn_output.py b/tests/test_inngest_encryption/cases/step_and_fn_output.py
@@ -97,15 +97,15 @@ async def run_test(self: base.TestClass) -> None:
         print("running step and fn output test")
         self.client.send_sync(inngest.Event(name=event_name))
 
-        run_id = state.wait_for_run_id()
-        run = test_core.helper.client.wait_for_run_status(
+        run_id = await state.wait_for_run_id()
+        run = await test_core.helper.client.wait_for_run_status(
             run_id,
             test_core.helper.RunStatus.COMPLETED,
         )
 
         # Ensure that step_1 output is encrypted and its value is correct
         output = json.loads(
-            test_core.helper.client.get_step_output(
+            await test_core.helper.client.get_step_output(
                 run_id=run_id,
                 step_id="step_1",
             )
@@ -117,7 +117,7 @@ async def run_test(self: base.TestClass) -> None:
 
         # Ensure that step_2 output is encrypted and its value is correct
         output = json.loads(
-            test_core.helper.client.get_step_output(
+            await test_core.helper.client.get_step_output(
                 run_id=run_id,
                 step_id="step_2",
             )