Added more tests to composer

navikt · May 15, 2024 · 73fdd28 · 73fdd28
1 parent 29c5dce
commit 73fdd28
Showing 1 changed file with 54 additions and 0 deletions.
diff --git a/tests-integration/composer.py b/tests-integration/composer.py
@@ -14,6 +14,16 @@
         startup_timeout_seconds=60,
     )
 
+    py_op_uv = python_operator(
+        dag=dag,
+        name="python-op-uv",
+        script_path="tests-integration/notebooks/script.py",
+        requirements_path="tests-integration/notebooks/requirements.txt",
+        retries=0,
+        startup_timeout_seconds=60,
+        use_uv_pip_install=True,
+    )
+
     nb_op = notebook_operator(
         dag=dag,
         name="nb-op",
@@ -23,6 +33,16 @@
         startup_timeout_seconds=60,
     )
 
+    nb_op_uv = notebook_operator(
+        dag=dag,
+        name="nb-op-uv",
+        nb_path="tests-integration/notebooks/mynb.ipynb",
+        requirements_path="tests-integration/notebooks/requirements.txt",
+        retries=0,
+        startup_timeout_seconds=60,
+        use_uv_pip_install=True,
+    )
+
     quarto_op = quarto_operator(
         dag=dag,
         name="quarto-op",
@@ -37,6 +57,21 @@
         startup_timeout_seconds=60,
     )
 
+    quarto_op_uv = quarto_operator(
+        dag=dag,
+        name="quarto-op-uv",
+        quarto={
+            "path": "tests-integration/notebooks/quarto.ipynb",
+            "env": "dev",
+            "id": "bf48d8a4-05ca-47a5-a360-bc24171baf62",
+            "token": Variable.get("quarto_token"),
+        },
+        requirements_path="tests-integration/notebooks/requirements.txt",
+        retries=0,
+        startup_timeout_seconds=60,
+        use_uv_pip_install=True,
+    )
+
     quarto_book_op = quarto_operator(
         dag=dag,
         name="quarto-book-op",
@@ -51,7 +86,26 @@
         startup_timeout_seconds=60,
     )
 
+    quarto_book_op_uv = quarto_operator(
+        dag=dag,
+        name="quarto-book-op-uv",
+        quarto={
+            "folder": "tests-integration/notebooks/quartobook",
+            "env": "dev",
+            "id": "757da08e-031e-4fac-a5f0-fffe6d2d96b6",
+            "token": Variable.get("quarto_token"),
+        },
+        requirements_path="tests-integration/notebooks/requirements.txt",
+        retries=0,
+        startup_timeout_seconds=60,
+        use_uv_pip_install=True,
+    )
+
     py_op
+    py_op_uv
     nb_op
+    nb_op_uv
     quarto_op
+    quarto_op_uv
     quarto_book_op
+    quarto_book_op_uv