Lee-W commented on code in PR #57474:
URL: https://github.com/apache/airflow/pull/57474#discussion_r2485653001
##########
airflow-core/tests/unit/api_fastapi/execution_api/versions/head/test_task_instances.py:
##########
@@ -436,6 +443,71 @@ def task_3():
)
assert response.json()["upstream_map_indexes"] == {"tg.task_2": [0, 1,
2, 3, 4, 5]}
+ def test_dynamic_task_mapping_with_trigger_rule(
+ self, client: Client, dag_maker: DagMaker, session: Session
+ ):
+ """
+ Test that the Task Instance upstream_map_indexes is not populuated but
+ the downstream task should still be run due to trigger rule.
+ """
+
+ with dag_maker(session=session, serialized=True):
+
+ @task
+ def task_1():
+ raise AirflowSkipException()
+
+ @task_group
+ def tg(x):
+ @task
+ def task_2():
+ raise AirflowSkipException()
+
+ task_2()
+
+ @task(trigger_rule=TriggerRule.ALL_DONE)
+ def task_3():
+ pass
+
+ @task
+ def task_4():
+ pass
+
+ tg.expand(x=task_1()) >> [task_3(), task_4()]
+
+ dr = dag_maker.create_dagrun()
+
+ decision = dr.task_instance_scheduling_decisions(session=session)
+
+ # Simulate task_1 skipped
+ (ti_1,) = decision.schedulable_tis
+ ti_1.state = TaskInstanceState.SKIPPED
+ session.flush()
+
+ # Now task_2 in mapped tagk group is not expanded and also skipped..
Review Comment:
updated
--
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]
