jroachgolf84 commented on code in PR #73135:
URL: https://github.com/apache/airflow/pull/73135#discussion_r4200635914
##########
airflow-core/tests/unit/triggers/test_base_trigger.py:
##########
@@ -278,6 +278,40 @@ def
test_base_event_trigger_asset_state_store_independent_across_instances():
assert b.asset_state_store is None
+def test_base_trigger_task_state_store_initialized_to_none():
+ """task_state_store is None before the triggerer injects one."""
Review Comment:
Resolved locally.
##########
airflow-core/tests/unit/triggers/test_base_trigger.py:
##########
@@ -278,6 +278,40 @@ def
test_base_event_trigger_asset_state_store_independent_across_instances():
assert b.asset_state_store is None
+def test_base_trigger_task_state_store_initialized_to_none():
+ """task_state_store is None before the triggerer injects one."""
+ trigger = DummyTrigger(name="Dummy Trigger")
+
+ assert trigger.task_state_store is None
+
+
+def test_base_trigger_task_state_store_can_be_set():
+ """task_state_store can be set once the Trigger is initialized."""
+ trigger = DummyTrigger(name="Dummy Trigger")
+
+ mock_store = create_autospec(TaskStateStoreAccessor, instance=True)
+ trigger.task_state_store = mock_store
+
+ assert trigger.task_state_store is mock_store
+
+
+def test_base_trigger_task_state_store_independent_across_instances():
+ """a.task_state_store does not impact b.task_state_store."""
Review Comment:
Resolved locally.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
