From 79dac436ec218dff796c114b94b87699e57beabc Mon Sep 17 00:00:00 2001 From: Valerii Mironchenko Date: Mon, 15 Jul 2024 14:20:09 +0300 Subject: [PATCH] Update Neo4jPathsModel with relationships (#100) * Update Neo4jPathsModel with relationships --------- Co-authored-by: Valerii Mironchenko --- oddrn_generator/path_models.py | 2 ++ tests/params.py | 1 + 2 files changed, 3 insertions(+) diff --git a/oddrn_generator/path_models.py b/oddrn_generator/path_models.py index 5ae31e2..25fedca 100644 --- a/oddrn_generator/path_models.py +++ b/oddrn_generator/path_models.py @@ -529,6 +529,7 @@ class Neo4jPathsModel(BasePathsModel): databases: str nodes: Optional[str] = None fields: Optional[str] = None + relationships: Optional[str] = None @classmethod def _dependencies_map_factory(cls): @@ -536,6 +537,7 @@ def _dependencies_map_factory(cls): "databases": ("databases",), "nodes": ("databases", "nodes"), "fields": ("databases", "nodes", "fields"), + "relationships": ("databases", "nodes", "relationships"), } data_source_path: str = "databases" diff --git a/tests/params.py b/tests/params.py index 482518e..fc91920 100644 --- a/tests/params.py +++ b/tests/params.py @@ -258,6 +258,7 @@ "databases": "some_database", "nodes": "some_node", "fields": "some_field", + "relationships": "some_relationship", }, "aliases": {}, },