Skip to content

Commit 9f70af5

Browse files
authored
fix: Adjust OL system test to latest changes (#52971)
1 parent b379779 commit 9f70af5

File tree

3 files changed

+2
-9
lines changed

3 files changed

+2
-9
lines changed

providers/openlineage/tests/system/openlineage/example_openlineage_base_complex_dag.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -132,12 +132,7 @@ def execute(self, context):
132132
with TaskGroup("section_1", prefix_group_id=True) as tg:
133133
task_5 = CustomMappedOperator.partial(task_id="task_5", doc_md="md doc").expand(value=[1])
134134
with TaskGroup("section_2", parent_group=tg, tooltip="group_tooltip") as tg2:
135-
if AIRFLOW_VERSION.major == 3:
136-
add_args: dict[str, Any] = {
137-
"run_as_user": "some_user"
138-
} # Random user break task execution on AF2
139-
else:
140-
add_args = {"sla": timedelta(seconds=123)} # type: ignore[dict-item] # SLA is not present in AF3 yet
135+
add_args: dict[str, Any] = {"sla": timedelta(seconds=123)} if AIRFLOW_VERSION.major == 2 else {}
141136
task_6 = EmptyOperator(
142137
task_id="task_6",
143138
on_success_callback=lambda x: print(1),

providers/openlineage/tests/system/openlineage/expected_events/openlineage_base_complex_dag__af3.json

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1931,7 +1931,6 @@
19311931
"outlets": "[]",
19321932
"owner": "airflow",
19331933
"priority_weight": 1,
1934-
"run_as_user": "some_user",
19351934
"queue": "{{ result is string }}",
19361935
"retries": "{{ result is number }}",
19371936
"retry_exponential_backoff": false,
@@ -2105,7 +2104,6 @@
21052104
"outlets": "[]",
21062105
"owner": "airflow",
21072106
"priority_weight": 1,
2108-
"run_as_user": "some_user",
21092107
"queue": "{{ result is string }}",
21102108
"retries": "{{ result is number }}",
21112109
"retry_exponential_backoff": false,

providers/openlineage/tests/system/openlineage/expected_events/openlineage_schedule_timetable_dag__af3.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
],
1414
"restrict_to_events": false
1515
},
16-
"timetable_summary": "3 events"
16+
"timetable_summary": "My Team's Baseball Games"
1717
}
1818
}
1919
}

0 commit comments

Comments
 (0)