Skip to content

Commit 1d30b38

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent cb773fe commit 1d30b38

1 file changed

Lines changed: 27 additions & 11 deletions

File tree

tests/unit/adapters/mssql/test_generate_schema_name.py

Lines changed: 27 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
7475
class 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+
114123
class 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

Comments
 (0)