def setUp(self, session) -> None:
self.default_time = datetime(2020, 1, 1)
clear_db_runs()
clear_db_xcom()
self.dag = self._create_dag()
self.app.dag_bag.dags = {self.dag.dag_id: self.dag} # type: ignore # pylint: disable=no-member
self.app.dag_bag.sync_to_db() # type: ignore # pylint: disable=no-member
dr = DagRun(
dag_id=self.dag.dag_id,
run_id="TEST_DAG_RUN_ID",
execution_date=self.default_time,
run_type=DagRunType.MANUAL,
)
session.add(dr)
session.commit()
self.client = self.app.test_client() # type:ignore
def setup_attrs(self, configured_app, session) -> None:
self.default_time = datetime(2020, 1, 1)
clear_db_runs()
clear_db_xcom()
self.app = configured_app
self.dag = self._create_dag()
self.app.dag_bag = DagBag(os.devnull, include_examples=False)
self.app.dag_bag.dags = {self.dag.dag_id: self.dag} # type: ignore # pylint: disable=no-member
self.app.dag_bag.sync_to_db() # type: ignore # pylint: disable=no-member
dr = DagRun(
dag_id=self.dag.dag_id,
run_id="TEST_DAG_RUN_ID",
execution_date=self.default_time,
run_type=DagRunType.MANUAL,
)
session.add(dr)
session.commit()
self.client = self.app.test_client() # type:ignore
def tearDown(self) -> None:
super().tearDown()
clear_db_runs()
clear_db_xcom()
def teardown_method(self) -> None:
clear_db_runs()
clear_db_xcom()
def tearDown(self) -> None:
db.clear_db_xcom()
def setUp(self) -> None:
db.clear_db_xcom()
def clean_db():
db.clear_db_runs()
db.clear_db_task_reschedule()
db.clear_db_xcom()
def clean_db():
clear_db_runs()
clear_db_xcom()
