fix: return empty data instead of None when OpenLineage on_start meth…

…od is missing (#41268) Signed-off-by: Kacper Muda <mudakacper@gmail.com>
apache · Aug 5, 2024 · a7353d2 · a7353d2
1 parent d6bc1f6
commit a7353d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/providers/openlineage/extractors/base.py b/airflow/providers/openlineage/extractors/base.py
@@ -113,7 +113,7 @@ def _execute_extraction(self) -> OperatorLineage | None:
                 "Operator %s does not have the get_openlineage_facets_on_start method.",
                 self.operator.task_type,
             )
-            return None
+            return OperatorLineage()
 
     def extract_on_complete(self, task_instance) -> OperatorLineage | None:
         failed_states = [TaskInstanceState.FAILED, TaskInstanceState.UP_FOR_RETRY]

diff --git a/tests/providers/openlineage/extractors/test_base.py b/tests/providers/openlineage/extractors/test_base.py
@@ -224,7 +224,7 @@ def test_extraction_without_on_start():
         task_instance=task_instance
     )
 
-    assert metadata is None
+    assert metadata == OperatorLineage()
 
     assert metadata_on_complete == OperatorLineage(
         inputs=INPUTS,