test: add test test_import_annotations

tests/providers/docker/decorators/_with_annotations.py

@@ -0,0 +1,18 @@
+from typing import TYPE_CHECKING
+
+from airflow.decorators import task
+
+if TYPE_CHECKING:
+    from airflow.decorators.base import Task
+
+__all__ = []
+
+
+def create_task_factory(image: str, **kwargs: "Any") -> "Task[..., Any]":  # type: ignore  # noqa: F821
+    kwargs.setdefault("multiple_outputs", False)
+
+    @task.docker(image=image, auto_remove="force", **kwargs)
+    def func(value: "dict[str, Any]") -> "Any":  # type: ignore  # noqa: F821
+        assert isinstance(value, dict)
+
+    return func

tests/providers/docker/decorators/test_docker.py

@@ -284,3 +284,25 @@ def f():
             ret = f()
 
         assert ret.operator.docker_url == "unix://var/run/docker.sock"
+
+    def test_import_annotations(self, dag_maker):
+        from typing import Any
+
+        from airflow.models.dagrun import DagRun  # noqa: TCH001
+        from airflow.utils.state import DagRunState
+
+        from ._with_annotations import create_task_factory  # noqa: TID252
+
+        task_factory = create_task_factory("python:3.9-slim")
+
+        with dag_maker():
+
+            @task.python(multiple_outputs=False)
+            def create_dummy_value() -> dict[str, Any]:
+                return {}
+
+            value = create_dummy_value()
+            _ = task_factory(value)
+
+        dagrun: DagRun = dag_maker.create_dagrun()
+        assert DagRunState(dagrun.state) == DagRunState.SUCCESS