test: fill 100% cov

datnguye · Sep 7, 2023 · 0cc2cbd · 0cc2cbd
1 parent 26c8e19
commit 0cc2cbd
Show file tree

Hide file tree

Showing 2 changed files with 89 additions and 0 deletions.
diff --git a/tests/unit/adapters/algos/test_test_relationship.py b/tests/unit/adapters/algos/test_test_relationship.py
@@ -52,6 +52,11 @@ class ManifestNode:
     depends_on: ManifestNodeDependsOn = field(default_factory=ManifestNodeDependsOn)
 
 
+@dataclass
+class ManifestExposureNode:
+    depends_on: ManifestNodeDependsOn
+
+
 @dataclass
 class ManifestNodeColumn:
     name: str
@@ -184,6 +189,17 @@ class DummyManifestTable:
     }
 
 
+@dataclass
+class DummyManifestWithExposure:
+    exposures = {
+        "exposure.dbt_resto.dummy": ManifestExposureNode(
+            depends_on=ManifestNodeDependsOn(
+                nodes=["model.dbt_resto.table1", "model.dbt_resto.table2"]
+            ),
+        )
+    }
+
+
 @dataclass
 class CatalogNode:
     columns: dict
@@ -355,3 +371,22 @@ def test_get_relationships(self, manifest, algorithm, expected):
     )
     def test_get_relationship_type(self, meta, type):
         assert algo.get_relationship_type(meta=meta) == type
+
+    @pytest.mark.parametrize(
+        "manifest, expected",
+        [
+            (
+                DummyManifestWithExposure(),
+                [
+                    dict(table_name="model.dbt_resto.table1", exposure_name="dummy"),
+                    dict(table_name="model.dbt_resto.table2", exposure_name="dummy"),
+                ],
+            ),
+            (
+                DummyManifestTable(),
+                [],
+            ),
+        ],
+    )
+    def test_get_node_exposures(self, manifest, expected):
+        assert expected == base_algo.get_node_exposures(manifest=manifest)
diff --git a/tests/unit/adapters/targets/dbml/test_dbml_test_relationship.py b/tests/unit/adapters/targets/dbml/test_dbml_test_relationship.py
@@ -219,6 +219,60 @@ class TestDbmlTestRelationship:
                 //Refs (based on the DBT Relationship Tests)
                 """,
             ),
+            (
+                [
+                    Table(
+                        name="model.dbt_resto.table1",
+                        database="--database--",
+                        schema="--schema--",
+                        columns=[Column(name="name1", data_type="name1-type")],
+                        raw_sql="--irrelevant--",
+                        exposures=["dummy1"]
+                    )
+                ],
+                [],
+                ["exposure:dummy1", "exposure:"],
+                [],
+                ["model"],
+                """//Tables (based on the selection criteria)
+                //--configured at schema: --database--.--schema--
+                Table "model.dbt_resto.table1" {
+                    "name1" "name1-type"
+                }
+                //Refs (based on the DBT Relationship Tests)
+                """,
+            ),
+            (
+                [
+                    Table(
+                        name="model.dbt_resto.table1",
+                        database="--database--",
+                        schema="--schema--",
+                        columns=[Column(name="name1", data_type="name1-type")],
+                        raw_sql="--irrelevant--",
+                        exposures=["dummy1"]
+                    ),
+                    Table(
+                        name="model.dbt_resto.table23",
+                        database="--database--",
+                        schema="--schema--",
+                        columns=[Column(name="name23", data_type="name23-type")],
+                        raw_sql="--irrelevant--",
+                        exposures=["dummy2", "dummy3"]
+                    )
+                ],
+                [],
+                ["exposure:dummy2"],
+                [],
+                ["model"],
+                """//Tables (based on the selection criteria)
+                //--configured at schema: --database--.--schema--
+                Table "model.dbt_resto.table23" {
+                    "name23" "name23-type"
+                }
+                //Refs (based on the DBT Relationship Tests)
+                """,
+            ),
         ],
     )
     def test_parse(