español - Prueba de la unidad de flujo de aire Python?

apache airflow tutorial

Question

Me gustaría agregar algunas pruebas unitarias para nuestros DAG, pero no pude encontrar ninguna. ¿Hay un marco para la prueba de unidad para DAGs? Existe un marco de pruebas de extremo a extremo, pero supongo que está muerto: https://issues.apache.org/jira/browse/AIRFLOW-79 . Por favor sugerir, gracias!

Answer 1

Actualmente no pude encontrar nada mejor que simplemente usar BashOperator :

with DAG(''platform-test'', start_date=datetime(2017, 8, 29)) as dag:
    test_command = "python3 -m unittest --verbose {}".format(platform_test_fname)
    op = BashOperator(
        task_id="platform-test",
        bash_command=test_command,
    )

Answer 2

Prueba tus operadores de esta manera:

class TestMyOperator(TestCase):

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

Source