From b88eeb48387a49206b8a985fa1895df8c0d929c8 Mon Sep 17 00:00:00 2001
From: Emily Rockman <emily.rockman@dbtlabs.com>
Date: Fri, 26 Jul 2024 11:11:09 -0500
Subject: [PATCH] make schema name dynamic

---
 integration_tests/macros/operations/create_source_table.sql | 2 +-
 integration_tests/tests/test_generate_model_struct_yaml.sql | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/integration_tests/macros/operations/create_source_table.sql b/integration_tests/macros/operations/create_source_table.sql
index 6794e51..f17f3fd 100644
--- a/integration_tests/macros/operations/create_source_table.sql
+++ b/integration_tests/macros/operations/create_source_table.sql
@@ -9,7 +9,7 @@ reset enable_case_sensitive_identifier;
 
 {% set target_schema=api.Relation.create(
     database=target.database,
-    schema="codegen_integration_tests__data_source_schema"
+    schema="{{ schema }}__data_source_schema"
 ) %}
 
 
diff --git a/integration_tests/tests/test_generate_model_struct_yaml.sql b/integration_tests/tests/test_generate_model_struct_yaml.sql
index 148323b..48378d6 100644
--- a/integration_tests/tests/test_generate_model_struct_yaml.sql
+++ b/integration_tests/tests/test_generate_model_struct_yaml.sql
@@ -3,7 +3,7 @@
 -- test all args
 {% set actual_source_yaml = codegen.generate_source(
     database_name=target.database,
-    schema_name='codegen_integration_tests__data_source_schema',
+    schema_name='{{ schema }}__data_source_schema',
     table_names=['codegen_integration_tests__data_source_table_nested_array'],
     generate_columns=True,
     include_descriptions=True