diff --git a/.github/workflows/test_kube.yml b/.github/workflows/test_kube.yml index 5be2b207d..16f35db5b 100644 --- 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 \ No newline at end of file + 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 new file mode 100644 index 000000000..aa9c1057e --- /dev/null +++ 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 \ No newline at end of file diff --git a/tests/test_example_dags.py b/tests/test_example_dags.py index af45191c9..e791ab878 100644 --- 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)