apache-airflow-providers-openlineage 2.7.2rc1__tar.gz → 2.7.3__tar.gz

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-openlineage
-Version: 2.7.2rc1
+Version: 2.7.3
 Summary: Provider package apache-airflow-providers-openlineage for Apache Airflow
 Keywords: airflow-provider,openlineage,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -20,15 +20,15 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: System :: Monitoring
-Requires-Dist: apache-airflow>=2.10.0rc1
-Requires-Dist: apache-airflow-providers-common-sql>=1.20.0rc1
-Requires-Dist: apache-airflow-providers-common-compat>=1.4.0rc1
+Requires-Dist: apache-airflow>=2.10.0
+Requires-Dist: apache-airflow-providers-common-sql>=1.20.0
+Requires-Dist: apache-airflow-providers-common-compat>=1.8.0
 Requires-Dist: attrs>=22.2
-Requires-Dist: openlineage-integration-common>=1.36.0
-Requires-Dist: openlineage-python>=1.36.0
+Requires-Dist: openlineage-integration-common>=1.38.0
+Requires-Dist: openlineage-python>=1.38.0
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.staged.apache.org/docs/apache-airflow-providers-openlineage/2.7.2/changelog.html
-Project-URL: Documentation, https://airflow.staged.apache.org/docs/apache-airflow-providers-openlineage/2.7.2
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3
 Project-URL: Mastodon, https://fosstodon.org/@airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
@@ -59,7 +59,7 @@ Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
 
 Package ``apache-airflow-providers-openlineage``
 
-Release: ``2.7.2``
+Release: ``2.7.3``
 
 
 `OpenLineage <https://openlineage.io/>`__
@@ -72,7 +72,7 @@ This is a provider package for ``openlineage`` provider. All classes for this pr
 are in ``airflow.providers.openlineage`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3/>`_.
 
 Installation
 ------------
@@ -91,10 +91,10 @@ PIP package                                 Version required
 ==========================================  ==================
 ``apache-airflow``                          ``>=2.10.0``
 ``apache-airflow-providers-common-sql``     ``>=1.20.0``
-``apache-airflow-providers-common-compat``  ``>=1.4.0``
+``apache-airflow-providers-common-compat``  ``>=1.8.0``
 ``attrs``                                   ``>=22.2``
-``openlineage-integration-common``          ``>=1.36.0``
-``openlineage-python``                      ``>=1.36.0``
+``openlineage-integration-common``          ``>=1.38.0``
+``openlineage-python``                      ``>=1.38.0``
 ==========================================  ==================
 
 Cross provider package dependencies
@@ -118,5 +118,5 @@ Dependent package
 ==================================================================================================================  =================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3/changelog.html>`_.
 

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/README.rst

@@ -23,7 +23,7 @@
 
 Package ``apache-airflow-providers-openlineage``
 
-Release: ``2.7.2``
+Release: ``2.7.3``
 
 
 `OpenLineage <https://openlineage.io/>`__
@@ -36,7 +36,7 @@ This is a provider package for ``openlineage`` provider. All classes for this pr
 are in ``airflow.providers.openlineage`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.2/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3/>`_.
 
 Installation
 ------------
@@ -55,10 +55,10 @@ PIP package                                 Version required
 ==========================================  ==================
 ``apache-airflow``                          ``>=2.10.0``
 ``apache-airflow-providers-common-sql``     ``>=1.20.0``
-``apache-airflow-providers-common-compat``  ``>=1.4.0``
+``apache-airflow-providers-common-compat``  ``>=1.8.0``
 ``attrs``                                   ``>=22.2``
-``openlineage-integration-common``          ``>=1.36.0``
-``openlineage-python``                      ``>=1.36.0``
+``openlineage-integration-common``          ``>=1.38.0``
+``openlineage-python``                      ``>=1.38.0``
 ==========================================  ==================
 
 Cross provider package dependencies
@@ -82,4 +82,4 @@ Dependent package
 ==================================================================================================================  =================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.2/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3/changelog.html>`_.

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/docs/changelog.rst

@@ -26,6 +26,32 @@
 Changelog
 ---------
 
+2.7.3
+.....
+
+Bug Fixes
+~~~~~~~~~
+
+* ``Fix openlineage dag state event emit on timed out dag (#56542)``
+* ``Only import OpenLineage plugin listeners/hooks if plugin is enabled (#56266)``
+
+Misc
+~~~~
+
+* ``nit: Bump required OL client for Openlineage provider (#56302)``
+* ``chore: safeguard external call in OL sqlparser (#55692)``
+
+Doc-only
+~~~~~~~~
+
+* ``Correct 'Dag' to 'DAG' for code snippets in provider docs (#56727)``
+* ``Remove placeholder Release Date in changelog and index files (#56056)``
+
+.. Below changes are excluded from the changelog. Move them to
+   appropriate section above if needed. Do not delete the lines(!):
+   * ``chore: add sleep in ol test operator (#54807)``
+   * ``Fix DagBag imports in 3.2+ (#56109)``
+
 2.7.2
 .....
 
@@ -43,8 +69,6 @@ Misc
 .....
 
 
-Release Date: ``|PypiReleaseDate|``
-
 Bug Fixes
 ~~~~~~~~~
 
@@ -63,8 +87,6 @@ Misc
 .....
 
 
-Release Date: ``|PypiReleaseDate|``
-
 Features
 ~~~~~~~~
 

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/docs/guides/user.rst

@@ -396,7 +396,7 @@ While enabling lineage on a Dag implicitly enables it for all tasks within that
 
     from airflow.providers.openlineage.utils.selective_enable import disable_lineage, enable_lineage
 
-    with Dag(...) as dag:
+    with DAG(...) as dag:
         t1 = MyOperator(...)
         t2 = AnotherOperator(...)
 

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/docs/index.rst

@@ -81,7 +81,7 @@ apache-airflow-providers-openlineage package
 `OpenLineage <https://openlineage.io/>`__
 
 
-Release: 2.7.2
+Release: 2.7.3
 
 Provider package
 ----------------
@@ -106,10 +106,10 @@ PIP package                                 Version required
 ==========================================  ==================
 ``apache-airflow``                          ``>=2.10.0``
 ``apache-airflow-providers-common-sql``     ``>=1.20.0``
-``apache-airflow-providers-common-compat``  ``>=1.4.0``
+``apache-airflow-providers-common-compat``  ``>=1.8.0``
 ``attrs``                                   ``>=22.2``
-``openlineage-integration-common``          ``>=1.36.0``
-``openlineage-python``                      ``>=1.36.0``
+``openlineage-integration-common``          ``>=1.38.0``
+``openlineage-python``                      ``>=1.38.0``
 ==========================================  ==================
 
 Cross provider package dependencies
@@ -138,5 +138,5 @@ Downloading official packages
 You can download officially released packages and verify their checksums and signatures from the
 `Official Apache Download site <https://downloads.apache.org/airflow/providers/>`_
 
-* `The apache-airflow-providers-openlineage 2.7.2 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.2.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.2.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.2.tar.gz.sha512>`__)
-* `The apache-airflow-providers-openlineage 2.7.2 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.2-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.2-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.2-py3-none-any.whl.sha512>`__)
+* `The apache-airflow-providers-openlineage 2.7.3 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.3.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.3.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.3.tar.gz.sha512>`__)
+* `The apache-airflow-providers-openlineage 2.7.3 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.3-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.3-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_openlineage-2.7.3-py3-none-any.whl.sha512>`__)

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/provider.yaml

@@ -22,12 +22,13 @@ description: |
   `OpenLineage <https://openlineage.io/>`__
 
 state: ready
-source-date-epoch: 1758787152
+source-date-epoch: 1761117353
 # Note that those versions are maintained by release manager - do not update them manually
 # with the exception of case where other provider in sources has >= new provider version.
 # In such case adding >= NEW_VERSION and bumping to NEW_VERSION in a provider have
 # to be done in the same PR
 versions:
+  - 2.7.3
   - 2.7.2
   - 2.7.1
   - 2.7.0

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/pyproject.toml

@@ -25,7 +25,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-openlineage"
-version = "2.7.2rc1"
+version = "2.7.3"
 description = "Provider package apache-airflow-providers-openlineage for Apache Airflow"
 readme = "README.rst"
 authors = [
@@ -57,12 +57,12 @@ requires-python = ">=3.10"
 # Make sure to run ``prek update-providers-dependencies --all-files``
 # After you modify the dependencies, and rebuild your Breeze CI image with ``breeze ci-image build``
 dependencies = [
-    "apache-airflow>=2.10.0rc1",
-    "apache-airflow-providers-common-sql>=1.20.0rc1",
-    "apache-airflow-providers-common-compat>=1.4.0rc1",
+    "apache-airflow>=2.10.0",
+    "apache-airflow-providers-common-sql>=1.20.0",
+    "apache-airflow-providers-common-compat>=1.8.0",
     "attrs>=22.2",
-    "openlineage-integration-common>=1.36.0",
-    "openlineage-python>=1.36.0",
+    "openlineage-integration-common>=1.38.0",
+    "openlineage-python>=1.38.0",
 ]
 
 [dependency-groups]
@@ -105,8 +105,8 @@ apache-airflow-providers-common-sql = {workspace = true}
 apache-airflow-providers-standard = {workspace = true}
 
 [project.urls]
-"Documentation" = "https://airflow.staged.apache.org/docs/apache-airflow-providers-openlineage/2.7.2"
-"Changelog" = "https://airflow.staged.apache.org/docs/apache-airflow-providers-openlineage/2.7.2/changelog.html"
+"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3"
+"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-openlineage/2.7.3/changelog.html"
 "Bug Tracker" = "https://github.com/apache/airflow/issues"
 "Source Code" = "https://github.com/apache/airflow"
 "Slack Chat" = "https://s.apache.org/airflow-slack"

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/src/airflow/providers/openlineage/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "2.7.2"
+__version__ = "2.7.3"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.10.0"

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/src/airflow/providers/openlineage/plugins/openlineage.py

@@ -18,16 +18,20 @@ from __future__ import annotations
 
 from airflow.plugins_manager import AirflowPlugin
 from airflow.providers.openlineage import conf
-from airflow.providers.openlineage.plugins.listener import get_openlineage_listener
-from airflow.providers.openlineage.plugins.macros import (
-    lineage_job_name,
-    lineage_job_namespace,
-    lineage_parent_id,
-    lineage_root_job_name,
-    lineage_root_parent_id,
-    lineage_root_run_id,
-    lineage_run_id,
-)
+
+# Conditional imports - only load expensive dependencies when plugin is enabled
+if not conf.is_disabled():
+    from airflow.lineage.hook import HookLineageReader
+    from airflow.providers.openlineage.plugins.listener import get_openlineage_listener
+    from airflow.providers.openlineage.plugins.macros import (
+        lineage_job_name,
+        lineage_job_namespace,
+        lineage_parent_id,
+        lineage_root_job_name,
+        lineage_root_parent_id,
+        lineage_root_run_id,
+        lineage_run_id,
+    )
 
 
 class OpenLineageProviderPlugin(AirflowPlugin):
@@ -50,8 +54,6 @@ class OpenLineageProviderPlugin(AirflowPlugin):
             lineage_root_parent_id,
         ]
         listeners = [get_openlineage_listener()]
-        from airflow.lineage.hook import HookLineageReader
-
         hook_lineage_readers = [HookLineageReader]
     else:
         macros = []

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/src/airflow/providers/openlineage/sqlparser.py

@@ -232,8 +232,8 @@ class SQLParser(LoggingMixin):
             else None,
         )
 
+    @staticmethod
     def get_metadata_from_parser(
-        self,
         inputs: list[DbTableMeta],
         outputs: list[DbTableMeta],
         database_info: DatabaseInfo,
@@ -315,6 +315,7 @@ class SQLParser(LoggingMixin):
         :param database_info: database specific information
         :param database: when passed it takes precedence over parsed database name
         :param sqlalchemy_engine: when passed, engine's dialect is used to compile SQL queries
+        :param use_connection: if call to db should be performed to enrich datasets (e.g., with schema)
         """
         job_facets: dict[str, JobFacet] = {"sql": sql_job.SQLJobFacet(query=self.normalize_sql(sql))}
         parse_result = self.parse(sql=self.split_sql_string(sql))
@@ -338,17 +339,28 @@ class SQLParser(LoggingMixin):
             )
 
         namespace = self.create_namespace(database_info=database_info)
+        inputs: list[Dataset] = []
+        outputs: list[Dataset] = []
         if use_connection:
-            inputs, outputs = self.parse_table_schemas(
-                hook=hook,
-                inputs=parse_result.in_tables,
-                outputs=parse_result.out_tables,
-                namespace=namespace,
-                database=database,
-                database_info=database_info,
-                sqlalchemy_engine=sqlalchemy_engine,
-            )
-        else:
+            try:
+                inputs, outputs = self.parse_table_schemas(
+                    hook=hook,
+                    inputs=parse_result.in_tables,
+                    outputs=parse_result.out_tables,
+                    namespace=namespace,
+                    database=database,
+                    database_info=database_info,
+                    sqlalchemy_engine=sqlalchemy_engine,
+                )
+            except Exception as e:
+                self.log.warning(
+                    "OpenLineage method failed to enrich datasets using db metadata. Exception: `%s`",
+                    e,
+                )
+                self.log.debug("OpenLineage failure details:", exc_info=True)
+
+        # If call to db failed or was not performed, use datasets from sql parsing alone
+        if not inputs and not outputs:
             inputs, outputs = self.get_metadata_from_parser(
                 inputs=parse_result.in_tables,
                 outputs=parse_result.out_tables,

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/src/airflow/providers/openlineage/utils/utils.py

@@ -751,16 +751,20 @@ def get_airflow_state_run_facet(
     dag_id: str, run_id: str, task_ids: list[str], dag_run_state: DagRunState
 ) -> dict[str, AirflowStateRunFacet]:
     tis = DagRun.fetch_task_instances(dag_id=dag_id, run_id=run_id, task_ids=task_ids)
+
+    def get_task_duration(ti):
+        if ti.duration is not None:
+            return ti.duration
+        if ti.end_date is not None and ti.start_date is not None:
+            return (ti.end_date - ti.start_date).total_seconds()
+        # Fallback to 0.0 for tasks with missing timestamps (e.g., skipped/terminated tasks)
+        return 0.0
+
     return {
         "airflowState": AirflowStateRunFacet(
             dagRunState=dag_run_state,
             tasksState={ti.task_id: ti.state for ti in tis},
-            tasksDuration={
-                ti.task_id: ti.duration
-                if ti.duration is not None
-                else (ti.end_date - ti.start_date).total_seconds()
-                for ti in tis
-            },
+            tasksDuration={ti.task_id: get_task_duration(ti) for ti in tis},
         )
     }
 

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/tests/system/openlineage/operator.py

@@ -21,6 +21,7 @@ import json
 import logging
 import os
 import re
+import time
 import uuid
 from typing import TYPE_CHECKING, Any
 from urllib.parse import urlparse
@@ -220,6 +221,7 @@ class OpenLineageTestOperator(BaseOperator):
             raise ValueError("Can't pass both event_templates and file_path")
 
     def execute(self, context: Context) -> None:
+        time.sleep(10)  # Wait for all variables to update properly
         if self.file_path is not None:
             self.event_templates = {}
             self.log.info("Reading OpenLineage event templates from file `%s`", self.file_path)

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/tests/unit/openlineage/plugins/test_execution.py

@@ -37,9 +37,9 @@ from airflow.utils.types import DagRunType
 
 from tests_common.test_utils.config import conf_vars
 from tests_common.test_utils.db import clear_db_runs
-from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS, AIRFLOW_V_3_1_PLUS
+from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS, AIRFLOW_V_3_2_PLUS
 
-if AIRFLOW_V_3_1_PLUS:
+if AIRFLOW_V_3_2_PLUS:
     from airflow.dag_processing.dagbag import DagBag
 else:
     from airflow.models.dagbag import DagBag  # type: ignore[attr-defined, no-redef]

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/tests/unit/openlineage/test_sqlparser.py

@@ -356,3 +356,53 @@ class TestSQLParser:
             }
         )
         assert metadata.job_facets["sql"].query.replace(" ", "") == formatted_sql.replace(" ", "")
+
+    def test_generate_openlineage_metadata_from_sql_with_db_error(self):
+        parser = SQLParser(default_schema="ANOTHER_SCHEMA")
+        db_info = DatabaseInfo(scheme="myscheme", authority="host:port")
+
+        hook = MagicMock()
+
+        sql = """INSERT INTO popular_orders_day_of_week (order_day_of_week)
+        SELECT EXTRACT(ISODOW FROM order_placed_on) AS order_day_of_week
+        FROM top_delivery_times -- irrelevant comment"""
+
+        hook.get_conn.side_effect = RuntimeError("Simulated DB error")
+
+        formatted_sql = """INSERT INTO popular_orders_day_of_week (order_day_of_week)
+        SELECT EXTRACT(ISODOW FROM order_placed_on) AS order_day_of_week
+        FROM top_delivery_times"""
+        expected_schema = "ANOTHER_SCHEMA"
+        metadata = parser.generate_openlineage_metadata_from_sql(
+            sql=sql, hook=hook, database_info=db_info, use_connection=True
+        )
+
+        assert metadata.inputs == [
+            Dataset(
+                namespace="myscheme://host:port",
+                name=f"{expected_schema}.top_delivery_times",
+                facets={},
+            )
+        ]
+        assert len(metadata.outputs) == 1
+        assert metadata.outputs[0].namespace == "myscheme://host:port"
+        assert metadata.outputs[0].name == f"{expected_schema}.popular_orders_day_of_week"
+        assert len(metadata.outputs[0].facets) == 1
+        assert metadata.outputs[0].facets[
+            "columnLineage"
+        ] == column_lineage_dataset.ColumnLineageDatasetFacet(
+            fields={
+                "order_day_of_week": column_lineage_dataset.Fields(
+                    inputFields=[
+                        column_lineage_dataset.InputField(
+                            namespace="myscheme://host:port",
+                            name=f"{expected_schema}.top_delivery_times",
+                            field="order_placed_on",
+                        )
+                    ],
+                    transformationDescription="",
+                    transformationType="",
+                )
+            }
+        )
+        assert metadata.job_facets["sql"].query.replace(" ", "") == formatted_sql.replace(" ", "")

{apache_airflow_providers_openlineage-2.7.2rc1 → apache_airflow_providers_openlineage-2.7.3}/tests/unit/openlineage/utils/test_utils.py

@@ -44,6 +44,7 @@ from airflow.providers.openlineage.utils.utils import (
     _truncate_string_to_byte_size,
     get_airflow_dag_run_facet,
     get_airflow_job_facet,
+    get_airflow_state_run_facet,
     get_dag_documentation,
     get_fully_qualified_class_name,
     get_job_name,
@@ -57,6 +58,7 @@ from airflow.serialization.serialized_objects import SerializedBaseOperator
 from airflow.timetables.events import EventsTimetable
 from airflow.timetables.trigger import CronTriggerTimetable
 from airflow.utils import timezone
+from airflow.utils.session import create_session
 from airflow.utils.state import DagRunState
 from airflow.utils.types import DagRunType
 
@@ -2054,3 +2056,65 @@ def test_get_operator_provider_version_for_mapped_operator(mock_providers_manage
     mapped_operator = BashOperator.partial(task_id="test_task").expand(bash_command=["echo 1", "echo 2"])
     result = get_operator_provider_version(mapped_operator)
     assert result == "1.2.0"
+
+
+class TestGetAirflowStateRunFacet:
+    @pytest.mark.db_test
+    def test_task_with_timestamps_defined(self, dag_maker):
+        """Test task instance with defined start_date and end_date."""
+        with dag_maker(dag_id="test_dag"):
+            BaseOperator(task_id="test_task")
+
+        dag_run = dag_maker.create_dagrun()
+        ti = dag_run.get_task_instance(task_id="test_task")
+
+        # Set valid timestamps
+        start_time = pendulum.parse("2024-01-01T10:00:00Z")
+        end_time = pendulum.parse("2024-01-01T10:02:30Z")  # 150 seconds difference
+        ti.start_date = start_time
+        ti.end_date = end_time
+        ti.state = TaskInstanceState.SUCCESS
+        ti.duration = None
+
+        # Persist changes to database
+        with create_session() as session:
+            session.merge(ti)
+            session.commit()
+
+        result = get_airflow_state_run_facet(
+            dag_id="test_dag",
+            run_id=dag_run.run_id,
+            task_ids=["test_task"],
+            dag_run_state=DagRunState.SUCCESS,
+        )
+
+        assert result["airflowState"].tasksDuration["test_task"] == 150.0
+
+    @pytest.mark.db_test
+    def test_task_with_none_timestamps_fallback_to_zero(self, dag_maker):
+        """Test task with None timestamps falls back to 0.0."""
+        with dag_maker(dag_id="test_dag"):
+            BaseOperator(task_id="terminated_task")
+
+        dag_run = dag_maker.create_dagrun()
+        ti = dag_run.get_task_instance(task_id="terminated_task")
+
+        # Set None timestamps (signal-terminated case)
+        ti.start_date = None
+        ti.end_date = None
+        ti.state = TaskInstanceState.SKIPPED
+        ti.duration = None
+
+        # Persist changes to database
+        with create_session() as session:
+            session.merge(ti)
+            session.commit()
+
+        result = get_airflow_state_run_facet(
+            dag_id="test_dag",
+            run_id=dag_run.run_id,
+            task_ids=["terminated_task"],
+            dag_run_state=DagRunState.FAILED,
+        )
+
+        assert result["airflowState"].tasksDuration["terminated_task"] == 0.0