def test_dagrun_id_is_backfill(self): """ Task instances whose dagrun ID is a backfill dagrun ID should fail this dep. """ dagrun = DagRun() dagrun.run_id = "anything" dagrun.run_type = DagRunType.BACKFILL_JOB ti = Mock(get_dagrun=Mock(return_value=dagrun)) self.assertFalse(DagrunIdDep().is_met(ti=ti))
def test_dagrun_id_is_not_backfill(self): """ Task instances whose dagrun ID is not a backfill dagrun ID should pass this dep. """ dagrun = DagRun() dagrun.run_type = 'custom_type' ti = Mock(get_dagrun=Mock(return_value=dagrun)) self.assertTrue(DagrunIdDep().is_met(ti=ti)) dagrun = DagRun() dagrun.run_id = None ti = Mock(get_dagrun=Mock(return_value=dagrun)) self.assertTrue(DagrunIdDep().is_met(ti=ti))