python - Airflow Python单元测试？

Question

我想为我们的 DAG 添加一些单元测试，但找不到。是否有用于 DAG 的单元测试框架？存在一个端到端测试框架，但我猜它已经死了:https://issues.apache.org/jira/browse/AIRFLOW-79 .请推荐，谢谢!

Answer 1

像这样测试你的操作符:

class TestMyOperator(TestCase):

    def test_execute(self):
        with DAG(dag_id="foo", start_date=datetime.now()):
            task = MyOperator(task_id="foo")
        ti = TaskInstance(task=task, execution_date=datetime.now())
        result = task.execute(ti.get_template_context())
        self.assertEqual(result, "foo")

Source