Update snowflake generator with relationships path (#97)

* Update PathsModel with relationships * Update tests --------- Co-authored-by: Valerii Mironchenko <vmironchenko@provectus.com>
opendatadiscovery · Feb 18, 2024 · 5fd8b07 · 5fd8b07
1 parent 022ea5c
commit 5fd8b07
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/oddrn_generator/path_models.py b/oddrn_generator/path_models.py
@@ -132,6 +132,7 @@ class SnowflakePathsModel(BasePathsModel):
     tables_columns: Optional[str] = Field(alias="columns")
     views_columns: Optional[str] = Field(alias="columns")
     pipes: Optional[str]
+    relationships: Optional[str]
 
     class Config:
         dependencies_map = {
@@ -142,6 +143,7 @@ class Config:
             "tables_columns": ("databases", "schemas", "tables", "tables_columns"),
             "views_columns": ("databases", "schemas", "views", "views_columns"),
             "pipes": ("pipes",),
+            "relationships": ("databases", "schemas", "tables", "relationships"),
         }
         data_source_path = "databases"
 

diff --git a/tests/params.py b/tests/params.py
@@ -75,6 +75,7 @@
                 "tables_columns": "some_table_column",
                 "views_columns": "some_view_column",
                 "pipes": "pipe",
+                "relationships": "some_relationship",
             },
             "aliases": {
                 "tables_columns": "columns",