amoghrajesh commented on PR #44843:
URL: https://github.com/apache/airflow/pull/44843#issuecomment-2538393429
FYI, I ran the DAG from ash that found this issue:
```
from __future__ import annotations
import sys
import time
from datetime import datetime
from airflow.decorators import dag, task
@dag(
# every minute on the 30-second mark
catchup=False,
tags=[],
schedule=None,
start_date=datetime(2021, 1, 1),
)
def hello_dag():
"""
### TaskFlow API Tutorial Documentation
This is a simple data pipeline example which demonstrates the use of
the TaskFlow API using three simple tasks for Extract, Transform, and
Load.
Documentation that goes along with the Airflow TaskFlow API tutorial is
located
[here](https://airflow.apache.org/docs/apache-airflow/stable/tutorial_taskflow_api.html)
"""
@task()
def hello():
print("hello")
time.sleep(3)
print("goodbye")
print("err mesg", file=sys.stderr)
hello()
hello_dag()
```
using this UT:
```
def test_startup_dag_with_no_templates_mixed_types(mocked_parse,
test_dags_dir):
"""Test startup of a simple DAG."""
what = StartupDetails(
ti=TaskInstance(id=uuid7(), task_id="hello", dag_id="hello_dag",
run_id="c", try_number=1),
file=str(test_dags_dir / "mydag.py"),
requests_fd=0,
)
ti = parse(what)
with mock.patch(
"airflow.sdk.execution_time.task_runner.SUPERVISOR_COMMS",
create=True
) as mock_supervisor_comms:
mock_supervisor_comms.get_message.return_value = what
startup()
run(ti, log=mock.ANY)
```
Result:
```
Home of the user: /Users/amoghdesai
Airflow home /Users/amoghdesai/airflow
PASSED [100%][2024-12-12T15:16:57.104+0530] {dagbag.py:535} INFO - Filling
up the DagBag from
/Users/amoghdesai/Documents/OSS/airflow/task_sdk/tests/dags/mydag.py
[2024-12-12T15:16:57.175+0530] {dagbag.py:535} INFO - Filling up the DagBag
from /Users/amoghdesai/Documents/OSS/airflow/task_sdk/tests/dags/mydag.py
2024-12-12 15:16:57 [debug ] DAG file parsed [task]
file=/Users/amoghdesai/Documents/OSS/airflow/task_sdk/tests/dags/mydag.py
hello
goodbye
[2024-12-12T15:17:00.178+0530] {python.py:197} INFO - Done. Returned value
was: None
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
