)
from airflow.models.baseoperator import BaseOperator
from airflow.models.dagrun import DagRunNote
from airflow.models.taskinstance import TaskInstanceNote
from airflow.models.taskmap import TaskMap
from airflow.models.taskreschedule import TaskReschedule
from airflow.operators.empty import EmptyOperator
from airflow.operators.python import ShortCircuitOperator
from airflow.serialization.serialized_objects import SerializedDAG
from airflow.stats import Stats
from airflow.utils import timezone
from airflow.utils.state import DagRunState, State, TaskInstanceState
from airflow.utils.trigger_rule import TriggerRule
from airflow.utils.types import DagRunType
from tests.models import DEFAULT_DATE as _DEFAULT_DATE
from tests.test_utils import db
from tests.test_utils.mock_operators import MockOperator
DEFAULT_DATE = pendulum.instance(_DEFAULT_DATE)
class TestDagRun:
    dagbag = DagBag(include_examples=True)
    @staticmethod
    def clean_db():
        db.clear_db_runs()
        db.clear_db_pools()
        db.clear_db_dags()
        db.clear_db_variables()
        db.clear_db_datasets()
        db.clear_db_xcom()
        db.clear_db_task_fail()
    def setup_class(self) -> None: