@@ -71,6 +71,7 @@ def _render(custom_schema_name, target_schema="my_target_schema", use_default_co
7171# Tests - flag=False (legacy behaviour, the default)
7272# ---------------------------------------------------------------------------
7373
74+
7475class TestLegacyBehaviour :
7576 """When flag is False (or absent), uses legacy adapter behaviour."""
7677
@@ -83,14 +84,21 @@ def test_custom_schema_returned_directly_without_prefix(self):
8384 Key difference from dbt-core: custom_schema_name is NOT prefixed
8485 with target.schema. "reporting" stays "reporting".
8586 """
86- assert _render ("reporting" , target_schema = "dbt_dev" , use_default_concat = False ) == "reporting"
87+ assert (
88+ _render ("reporting" , target_schema = "dbt_dev" , use_default_concat = False ) == "reporting"
89+ )
8790
8891 def test_custom_schema_whitespace_is_trimmed (self ):
89- assert _render (" analytics " , target_schema = "dbt_dev" , use_default_concat = False ) == "analytics"
92+ assert (
93+ _render (" analytics " , target_schema = "dbt_dev" , use_default_concat = False )
94+ == "analytics"
95+ )
9096
9197 def test_flag_absent_defaults_to_legacy (self ):
9298 """var() returning its default (False) gives the same legacy result."""
93- env = jinja2 .Environment (trim_blocks = True , lstrip_blocks = True , extensions = ["jinja2.ext.do" ])
99+ env = jinja2 .Environment (
100+ trim_blocks = True , lstrip_blocks = True , extensions = ["jinja2.ext.do" ]
101+ )
94102 tmpl = env .from_string (
95103 _DEFAULT_GENERATE_SCHEMA_NAME
96104 + "\n "
@@ -111,6 +119,7 @@ def test_flag_absent_defaults_to_legacy(self):
111119# Tests - flag=True (dbt-core default concatenation)
112120# ---------------------------------------------------------------------------
113121
122+
114123class TestDefaultConcatBehaviour :
115124 """When flag is True, delegates to default__generate_schema_name."""
116125
@@ -119,25 +128,32 @@ def test_no_custom_schema_returns_target_schema(self):
119128
120129 def test_custom_schema_is_prefixed_with_target_schema (self ):
121130 """dbt-core: "dbt_dev" + "_" + "reporting" -> "dbt_dev_reporting" """
122- assert _render ("reporting" , target_schema = "dbt_dev" , use_default_concat = True ) == "dbt_dev_reporting"
131+ assert (
132+ _render ("reporting" , target_schema = "dbt_dev" , use_default_concat = True )
133+ == "dbt_dev_reporting"
134+ )
123135
124136 def test_custom_schema_concatenation_uses_underscore_separator (self ):
125- assert _render ("finance" , target_schema = "analytics" , use_default_concat = True ) == "analytics_finance"
137+ assert (
138+ _render ("finance" , target_schema = "analytics" , use_default_concat = True )
139+ == "analytics_finance"
140+ )
126141
127142
128143# ---------------------------------------------------------------------------
129144# Parametrised matrix test
130145# ---------------------------------------------------------------------------
131146
147+
132148@pytest .mark .parametrize (
133149 "custom_schema_name, target_schema, use_default_concat, expected" ,
134150 [
135- (None , "dbt_dev" , False , "dbt_dev" ),
136- ("reporting" , "dbt_dev" , False , "reporting" ),
137- (" trimmed " , "dbt_dev" , False , "trimmed" ),
138- (None , "dbt_dev" , True , "dbt_dev" ),
139- ("reporting" , "dbt_dev" , True , "dbt_dev_reporting" ),
140- ("finance" , "analytics" , True , "analytics_finance" ),
151+ (None , "dbt_dev" , False , "dbt_dev" ),
152+ ("reporting" , "dbt_dev" , False , "reporting" ),
153+ (" trimmed " , "dbt_dev" , False , "trimmed" ),
154+ (None , "dbt_dev" , True , "dbt_dev" ),
155+ ("reporting" , "dbt_dev" , True , "dbt_dev_reporting" ),
156+ ("finance" , "analytics" , True , "analytics_finance" ),
141157 ],
142158 ids = [
143159 "legacy-no_custom" ,
0 commit comments