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)