コード例 #1
0
 def test_check_and_change_state_before_execution_dep_not_met(self):
     dag = models.DAG(dag_id='test_check_and_change_state_before_execution')
     task = DummyOperator(task_id='task', dag=dag, start_date=DEFAULT_DATE)
     task2 = DummyOperator(task_id='task2', dag=dag, start_date=DEFAULT_DATE)
     task >> task2
     ti = TI(
         task=task2, execution_date=timezone.utcnow())
     self.assertFalse(ti._check_and_change_state_before_execution())
コード例 #2
0
 def test_check_and_change_state_before_execution_dep_not_met(self):
     dag = models.DAG(dag_id='test_check_and_change_state_before_execution')
     task = DummyOperator(task_id='task', dag=dag, start_date=DEFAULT_DATE)
     task2 = DummyOperator(task_id='task2', dag=dag, start_date=DEFAULT_DATE)
     task >> task2
     ti = TI(
         task=task2, execution_date=timezone.utcnow())
     self.assertFalse(ti._check_and_change_state_before_execution())
コード例 #3
0
 def test_check_and_change_state_before_execution(self):
     dag = models.DAG(dag_id='test_check_and_change_state_before_execution')
     task = DummyOperator(task_id='task', dag=dag, start_date=DEFAULT_DATE)
     ti = TI(task=task, execution_date=timezone.utcnow())
     self.assertEqual(ti._try_number, 0)
     self.assertTrue(ti._check_and_change_state_before_execution())
     # State should be running, and try_number column should be incremented
     self.assertEqual(ti.state, State.RUNNING)
     self.assertEqual(ti._try_number, 1)
コード例 #4
0
 def test_check_and_change_state_before_execution(self):
     dag = models.DAG(dag_id='test_check_and_change_state_before_execution')
     task = DummyOperator(task_id='task', dag=dag, start_date=DEFAULT_DATE)
     ti = TI(
         task=task, execution_date=timezone.utcnow())
     self.assertEqual(ti._try_number, 0)
     self.assertTrue(ti._check_and_change_state_before_execution())
     # State should be running, and try_number column should be incremented
     self.assertEqual(ti.state, State.RUNNING)
     self.assertEqual(ti._try_number, 1)