def setUp(self):
db.clear_db_jobs()
db.clear_db_dags()
db.clear_db_serialized_dags()
db.clear_db_runs()
db.clear_db_task_execution()
db.clear_db_message()
self.scheduler = None
self.port = 50102
self.storage = MemoryEventStorage()
self.master = NotificationMaster(NotificationService(self.storage),
self.port)
self.master.run()
self.client = NotificationClient(server_uri="localhost:{}".format(
self.port),
default_namespace="test_namespace")
time.sleep(1)
def test_delete_dag_run_and_task_instance_does_not_raise_error(self):
clear_db_jobs()
clear_db_runs()
job_id = 22
dag = DAG(dag_id='test_delete_dag_run', start_date=days_ago(1))
_ = BashOperator(task_id='task1', dag=dag, bash_command="echo hi")
# Simulate DagRun is created by a job inherited by BaseJob with an id
# This is so that same foreign key exists on DagRun.creating_job_id & BaseJob.id
dag_run = self.create_dag_run(dag=dag, creating_job_id=job_id)
assert dag_run is not None
session = settings.Session()
job = BaseJob(id=job_id)
session.add(job)
# Simulate TaskInstance is created by a job inherited by BaseJob with an id
# This is so that same foreign key exists on TaskInstance.queued_by_job_id & BaseJob.id
ti1 = dag_run.get_task_instance(task_id="task1")
ti1.queued_by_job_id = job_id
session.merge(ti1)
session.commit()
# Test Deleting DagRun does not raise an error
session.delete(dag_run)
# Test Deleting TaskInstance does not raise an error
ti1 = dag_run.get_task_instance(task_id="task1")
session.delete(ti1)
session.commit()
# CleanUp
clear_db_runs()
clear_db_jobs()
def setUp(self) -> None:
db.clear_db_runs()
db.clear_db_jobs()
def tearDown(self) -> None:
db.clear_db_runs()
db.clear_db_jobs()
def tearDown(self) -> None:
clear_db_jobs()
clear_db_runs()
def setUp(self):
clear_db_jobs()
clear_db_runs()
patcher = patch('airflow.jobs.base_job.sleep')
self.addCleanup(patcher.stop)
self.mock_base_job_sleep = patcher.start()
def clean_db_helper():
yield
clear_db_jobs()
clear_db_runs()
def setUp(self) -> None:
clear_db_jobs()
def tearDown(self) -> None:
if self.scheduler_job and self.scheduler_job.processor_agent:
self.scheduler_job.processor_agent.end()
clear_db_jobs()
def setUp(self) -> None:
clear_db_jobs()
self.scheduler_job = None
