Fix tests

dagster-io · Oct 21, 2024 · 7bf9c80 · 7bf9c80
1 parent 41914e8
commit 7bf9c80
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/...rative_automation_tests/automation_condition_tests/fundamentals/test_result_value_hash.py b/...rative_automation_tests/automation_condition_tests/fundamentals/test_result_value_hash.py
@@ -26,6 +26,7 @@
 two_parents_daily = two_parents.with_asset_properties(partitions_def=daily_partitions)
 
 
+@pytest.mark.asyncio
 @pytest.mark.parametrize(
     ["expected_value_hash", "condition", "scenario_spec", "materialize_A"],
     [
@@ -51,16 +52,16 @@
         ("7f852ab7408c67e0830530d025505a37", SC.missing(), one_parent_daily, False),
     ],
 )
-def test_value_hash(
+async def test_value_hash(
     condition: SC, scenario_spec: ScenarioSpec, expected_value_hash: str, materialize_A: bool
 ) -> None:
     state = AutomationConditionScenarioState(
         scenario_spec, automation_condition=condition
     ).with_current_time("2024-01-01T00:00")
 
-    state, _ = state.evaluate("downstream")
+    state, _ = await state.evaluate("downstream")
     if materialize_A:
         state = state.with_runs(run_request("A"))
 
-    state, result = state.evaluate("downstream")
+    state, result = await state.evaluate("downstream")
     assert result.value_hash == expected_value_hash
diff --git a/...itions_tests/declarative_automation_tests/scenario_utils/automation_condition_scenario.py b/...itions_tests/declarative_automation_tests/scenario_utils/automation_condition_scenario.py
@@ -64,7 +64,7 @@ def _get_request_subsets_by_key(
             for asset_key, aps in ap_by_key.items()
         }
 
-    def evaluate(
+    async def evaluate(
         self, asset: CoercibleToAssetKey
     ) -> Tuple["AutomationConditionScenarioState", AutomationResult]:
         asset_key = AssetKey.from_coercible(asset)
@@ -100,7 +100,7 @@ def evaluate(
             )  # type: ignore
             context = AutomationContext.create(key=asset_key, evaluator=evaluator)
 
-            full_result = asset_condition.evaluate(context)
+            full_result = await asset_condition.evaluate(context)  # type: ignore
             new_state = dataclasses.replace(self, condition_cursor=full_result.get_new_cursor())
             result = full_result.child_results[0] if self.ensure_empty_result else full_result