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()