Reject async callables in @task.virtualenv and @task.external_python - #73081

export DEMO_DIR="${TMPDIR:-/tmp}/airflow-async-venv-demo"
export AIRFLOW_HOME="$DEMO_DIR/airflow-home"
export AIRFLOW__CORE__DAGS_FOLDER="$DEMO_DIR/dags"
export AIRFLOW__CORE__LOAD_EXAMPLES=False
mkdir -p "$AIRFLOW__CORE__DAGS_FOLDER"
uv run --frozen --no-dev --project providers/standard --with "pandas>=3,<4" airflow db migrate
uv venv --allow-existing "$DEMO_DIR/pandas2-venv"
uv pip install --python "$DEMO_DIR/pandas2-venv/bin/python" "pandas>=2.2,<3"

cat > "$AIRFLOW__CORE__DAGS_FOLDER/virtualenv_example.py" <<'EOF'
from __future__ import annotations

from airflow.sdk import dag, task


@dag(schedule=None)
def virtualenv_example():
    @task.virtualenv(requirements=["pandas>=2.2,<3"], system_site_packages=False)
    def virtualenv_sync():
        import sys

        import pandas as pd

        return {"python": sys.executable, "pandas": pd.__version__}

    @task.virtualenv(requirements=["pandas>=2.2,<3"], system_site_packages=False)
    async def virtualenv_async():
        import sys

        import pandas as pd

        return {"python": sys.executable, "pandas": pd.__version__}

    virtualenv_sync()
    virtualenv_async()


example_dag = virtualenv_example()

if __name__ == "__main__":
    dag_run = example_dag.test()
    for ti in sorted(dag_run.get_task_instances(), key=lambda ti: ti.task_id):
        print(f"EXAMPLE STATE {ti.task_id} {ti.state}")
EOF

cat > "$AIRFLOW__CORE__DAGS_FOLDER/external_python_example.py" <<'EOF'
from __future__ import annotations

import os

from airflow.sdk import dag, task

PANDAS2_PYTHON = os.path.join(os.environ["DEMO_DIR"], "pandas2-venv", "bin", "python")


@dag(schedule=None)
def external_python_example():
    @task.external_python(python=PANDAS2_PYTHON)
    def external_python_sync():
        import sys

        import pandas as pd

        return {"python": sys.executable, "pandas": pd.__version__}

    @task.external_python(python=PANDAS2_PYTHON)
    async def external_python_async():
        import sys

        import pandas as pd

        return {"python": sys.executable, "pandas": pd.__version__}

    external_python_sync()
    external_python_async()


example_dag = external_python_example()

if __name__ == "__main__":
    dag_run = example_dag.test()
    for ti in sorted(dag_run.get_task_instances(), key=lambda ti: ti.task_id):
        print(f"EXAMPLE STATE {ti.task_id} {ti.state}")
EOF

uv run --frozen --no-dev --project providers/standard --with "pandas>=3,<4" python "$AIRFLOW__CORE__DAGS_FOLDER/virtualenv_example.py" 2>&1 | grep -oE "running task <TaskInstance: [^ ]+|Returned value was: \{[^}]*\}|EXAMPLE STATE .*"
uv run --frozen --no-dev --project providers/standard --with "pandas>=3,<4" python "$AIRFLOW__CORE__DAGS_FOLDER/external_python_example.py" 2>&1 | grep -oE "running task <TaskInstance: [^ ]+|Returned value was: \{[^}]*\}|EXAMPLE STATE .*"

uv run --frozen --no-dev --project providers/standard --with "pandas>=3,<4" python "$AIRFLOW__CORE__DAGS_FOLDER/virtualenv_example.py" 2>&1 | grep ValueError
uv run --frozen --no-dev --project providers/standard --with "pandas>=3,<4" python "$AIRFLOW__CORE__DAGS_FOLDER/external_python_example.py" 2>&1 | grep ValueError