Remove license and add test for dependencies.

astronomer · Sep 11, 2023 · 41784c0 · 41784c0
1 parent c856309
commit 41784c0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 201 deletions.
diff --git a/dev/dags/dbt/model_version/LICENSE b/dev/dags/dbt/model_version/LICENSE
diff --git a/tests/dbt/test_graph.py b/tests/dbt/test_graph.py
@@ -350,3 +350,19 @@ def test_load_dbt_ls_and_manifest_with_model_version(load_method):
     for unique_id, name in expected_dbt_nodes.items():
         assert unique_id in dbt_graph.nodes
         assert name == dbt_graph.nodes[unique_id].name
+
+    # Test dependencies
+    assert {
+        "model.jaffle_shop.stg_customers.v1",
+        "model.jaffle_shop.stg_orders.v1",
+        "model.jaffle_shop.stg_payments",
+    } == set(dbt_graph.nodes["model.jaffle_shop.customers.v1"].depends_on)
+    assert {
+        "model.jaffle_shop.stg_customers.v2",
+        "model.jaffle_shop.stg_orders.v1",
+        "model.jaffle_shop.stg_payments",
+    } == set(dbt_graph.nodes["model.jaffle_shop.customers.v2"].depends_on)
+    assert {
+        "model.jaffle_shop.stg_orders.v1",
+        "model.jaffle_shop.stg_payments",
+    } == set(dbt_graph.nodes["model.jaffle_shop.orders"].depends_on)