Add dag test

astronomer · Jul 22, 2024 · 9384c7c · 9384c7c
1 parent ea775fe
commit 9384c7c
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/.github/workflows/test_kube.yml b/.github/workflows/test_kube.yml
@@ -10,4 +10,6 @@ jobs:
       - name: Kubernetes KinD Cluster
         uses: container-tools/kind-action@v1
       - name: Run setup
-        run: sh ./scripts/test/kubernetes-test.sh
+        run: |
+          sh ./scripts/test/kubernetes-test.sh
+          sh ./scripts/test/integration-kube.sh
diff --git a/scripts/test/integration-kube.sh b/scripts/test/integration-kube.sh
@@ -0,0 +1,6 @@
+#!/bin/bash
+
+set -x
+set -e
+
+pytest tests/test_example_dags.py::test_example_dag_k8s
diff --git a/tests/test_example_dags.py b/tests/test_example_dags.py
@@ -91,6 +91,16 @@ def get_dag_ids() -> list[str]:
 @pytest.mark.integration
 @pytest.mark.parametrize("dag_id", get_dag_ids())
 def test_example_dag(session, dag_id: str):
+    if dag_id == "jaffle_shop_kubernetes":
+        return
     dag_bag = get_dag_bag()
     dag = dag_bag.get_dag(dag_id)
     test_utils.run_dag(dag)
+
+
+@pytest.mark.integration
+def test_example_dag_k8s(session):
+    dag_bag = get_dag_bag()
+    dag_id = "jaffle_shop_kubernetes"
+    dag = dag_bag.get_dag(dag_id)
+    test_utils.run_dag(dag)