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)