Add test for get_prior_knowledge

z3z1ma · Sep 17, 2023 · 61fe557 · 61fe557
1 parent 2a773d9
commit 61fe557
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_osmosis.py b/tests/test_osmosis.py
@@ -0,0 +1,40 @@
+from dbt.contracts.graph.manifest import Manifest
+from dbt_osmosis.core.osmosis import DbtYamlManager
+
+
+class TestDbtYamlManager:
+    def test_get_prior_knowledge(test):
+        knowledge = {
+            "myColumn": {
+                "progenitor": "source.my_model.source.Order",
+                "generation": "generation_0",
+                "name": "my_column",
+            },
+            "my_column": {
+                "progenitor": "model.my_model.mart.Order",
+                "generation": "generation_0",
+                "name": "my_column",
+            },
+        }
+        assert (
+            DbtYamlManager.get_prior_knowledge(knowledge, "my_column")["progenitor"]
+            == "source.my_model.source.Order"
+        )
+
+    def test_get_prior_knowledge_with_camel_case(test):
+        knowledge = {
+            "myColumn": {
+                "progenitor": "model.my_model.dwh.Order",
+                "generation": "generation_1",
+                "name": "myColumn",
+            },
+            "my_column": {
+                "progenitor": "model.my_model.mart.Order",
+                "generation": "generation_0",
+                "name": "my_column",
+            },
+        }
+        assert (
+            DbtYamlManager.get_prior_knowledge(knowledge, "my_column")["progenitor"]
+            == "model.my_model.dwh.Order"
+        )