Python DagRun.run_id示例

示例#1

文件： test_dagrun_id_dep.py 项目： zjasonliang/airflow

    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_id = 'notbackfill_something'
        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))

示例#2

文件： test_dagrun_id_dep.py 项目： zjasonliang/airflow

 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 = f"{DagRunType.BACKFILL_JOB.value}__something"
     ti = Mock(get_dagrun=Mock(return_value=dagrun))
     self.assertFalse(DagrunIdDep().is_met(ti=ti))

示例#3

 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))
     assert not DagrunIdDep().is_met(ti=ti)

示例#4

    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))
        assert DagrunIdDep().is_met(ti=ti)

        dagrun = DagRun()
        dagrun.run_id = None
        ti = Mock(get_dagrun=Mock(return_value=dagrun))
        assert DagrunIdDep().is_met(ti=ti)