Add test

astronomer · Aug 31, 2023 · 0f61f4c · 0f61f4c
1 parent b0c2d9b
commit 0f61f4c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/cosmos/dbt/graph.py b/cosmos/dbt/graph.py
@@ -255,7 +255,7 @@ def load_via_custom_parser(self) -> None:
             dbt_models_dir=self.project.models_dir.stem if self.project.models_dir else None,
             dbt_seeds_dir=self.project.seeds_dir.stem if self.project.seeds_dir else None,
             project_name=self.project.name,
-            operator_args=operator_args,
+            operator_args=self.operator_args,
         )
         nodes = {}
         models = itertools.chain(project.models.items(), project.snapshots.items(), project.seeds.items())

diff --git a/tests/dbt/parser/test_project.py b/tests/dbt/parser/test_project.py
@@ -190,3 +190,13 @@ def test_dbtmodelconfig_with_sources(tmp_path):
 
     dbt_model = DbtModel(name="some_name", type=DbtModelType.DBT_MODEL, path=path_with_sources)
     assert "sample_source" not in dbt_model.config.upstream_models
+
+
+def test_dbtmodelconfig_with_vars(tmp_path):
+    model_sql = SAMPLE_MODEL_SQL_PATH.read_text()
+    model_with_sources_sql = model_sql.replace("ref('stg_customers_'~ var('country_code'))")
+    path_with_sources = tmp_path / "customers_with_sources.sql"
+    path_with_sources.write_text(model_with_sources_sql)
+
+    dbt_model = DbtModel(name="some_name", type=DbtModelType.DBT_MODEL, path=path_with_sources, operator_args={"vars": {"country_code": "us"}})
+    assert "stg_customers_us" in dbt_model.config.upstream_models