Add test cases for mermaid

datnguye · Feb 8, 2024 · 349b1cc · 349b1cc
1 parent 3c64a87
commit 349b1cc
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/tests/unit/adapters/targets/mermaid/test_mermaid_test_relationship.py b/tests/unit/adapters/targets/mermaid/test_mermaid_test_relationship.py
@@ -25,6 +25,7 @@ class TestMermaidTestRelationship:
                 [],
                 [],
                 ["model"],
+                False,
                 """erDiagram
                   "MODEL.DBT_RESTO.TABLE1" {
                       name1-type name1
@@ -73,6 +74,7 @@ class TestMermaidTestRelationship:
                 [],
                 [],
                 ["model", "source"],
+                False,
                 """erDiagram
                   "MODEL.DBT_RESTO.TABLE1" {
                       name1-type name1
@@ -118,6 +120,7 @@ class TestMermaidTestRelationship:
                 ["schema:--schema--"],
                 [],
                 ["model", "source"],
+                False,
                 """erDiagram
                     "MODEL.DBT_RESTO.TABLE1" {
                         name1-type name1
@@ -139,6 +142,7 @@ class TestMermaidTestRelationship:
                 [],
                 ["model.dbt_resto.table1"],
                 ["model"],
+                False,
                 """erDiagram
                 """,
             ),
@@ -165,6 +169,7 @@ class TestMermaidTestRelationship:
                 ["model.dbt_resto"],
                 ["model.dbt_resto.table2"],
                 ["model"],
+                False,
                 """erDiagram
                     "MODEL.DBT_RESTO.TABLE1" {
                         name1-type name1
@@ -186,6 +191,7 @@ class TestMermaidTestRelationship:
                 ["schema:", "wildcard:", ""],
                 [],
                 ["model"],
+                False,
                 """erDiagram
                     "MODEL.DBT_RESTO.TABLE1" {
                         name1-type name1
@@ -215,12 +221,34 @@ class TestMermaidTestRelationship:
                 ["schema:--schema--,wildcard:*dbt_resto.table*"],
                 ["wildcard:*table2"],
                 ["model"],
+                False,
                 """erDiagram
                     "MODEL.DBT_RESTO.TABLE1" {
                         name1-type name1
                     }
                 """,
             ),
+            (
+                [
+                    Table(
+                        name="model.dbt_resto.table1",
+                        node_name="model.dbt_resto.table1",
+                        database="--database--",
+                        schema="--schema--",
+                        columns=[Column(name="name1", data_type="name1-type")],
+                        raw_sql="--irrelevant--",
+                    )
+                ],
+                [],
+                [],
+                [],
+                ["model"],
+                True,
+                """erDiagram
+                  "MODEL.DBT_RESTO.TABLE1" {
+                  }
+                """,
+            ),
         ],
     )
     def test_parse(
@@ -239,6 +267,7 @@ def test_parse(
                     catalog="--catalog--",
                     select=select,
                     exclude=exclude,
+                    omit_columns=omit_columns,
                     resource_type=resource_type,
                 )
                 print("mermaid ", mermaid.replace(" ", "").replace("\n", ""))