def setUp(self):
clear_db_runs()
clear_db_pools()
clear_db_dags()
clear_db_sla_miss()
clear_db_errors()
clear_db_event_model()
def setUp(self) -> None:
self.default_time = DEFAULT_DATETIME_1
self.ti_init = {
"execution_date": self.default_time,
"state": State.RUNNING,
}
self.ti_extras = {
"start_date": self.default_time + dt.timedelta(days=1),
"end_date": self.default_time + dt.timedelta(days=2),
"pid": 100,
"duration": 10000,
"pool": "default_pool",
"queue": "default_queue",
"job_id": 0,
}
self.client = self.app.test_client() # type:ignore
clear_db_runs()
clear_db_sla_miss()
self.dagbag = DagBag(include_examples=True)
def setup_attrs(self, configured_app) -> None:
self.default_time = DEFAULT_DATETIME_1
self.ti_init = {
"execution_date": self.default_time,
"state": State.RUNNING,
}
self.ti_extras = {
"start_date": self.default_time + dt.timedelta(days=1),
"end_date": self.default_time + dt.timedelta(days=2),
"pid": 100,
"duration": 10000,
"pool": "default_pool",
"queue": "default_queue",
"job_id": 0,
}
self.app = configured_app
self.client = self.app.test_client() # type:ignore
clear_db_runs()
clear_db_sla_miss()
DagBag(include_examples=True, read_dags_from_db=False).sync_to_db()
self.dagbag = DagBag(include_examples=True, read_dags_from_db=True)
