Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions dbt/adapters/sqlserver/sqlserver_adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,9 @@ class SQLServerAdapter(SQLAdapter):

def __init__(self, config, mp_context=None):
super().__init__(config, mp_context)
SQLServerRelation.disable_empty_relation_aliases = (
self.behavior.dbt_sqlserver_disable_empty_relation_aliases
)
if self.behavior.dbt_sqlserver_use_native_string_types:
self.Column = SQLServerColumnNative

Expand All @@ -76,6 +79,15 @@ def _behavior_flags(self) -> List[BehaviorFlag]:
"macro in your project instead."
),
},
{
"name": "dbt_sqlserver_disable_empty_relation_aliases",
"default": True,
"description": (
"When True, SQL Server limited relations used by --empty and sample mode "
"do not automatically receive dbt-generated aliases. Set this false to opt "
"out of alias generation temporarily for testing."
),
},
{
"name": "dbt_sqlserver_use_native_string_types",
"default": False,
Expand Down
13 changes: 10 additions & 3 deletions dbt/adapters/sqlserver/sqlserver_relation.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from dataclasses import dataclass, field
from typing import Optional, Type
from typing import ClassVar, Optional, Type

from dbt_common.exceptions import DbtRuntimeError

Expand All @@ -20,19 +20,26 @@ class SQLServerRelation(BaseRelation):
default_factory=lambda: SQLServerIncludePolicy()
)
quote_policy: SQLServerQuotePolicy = field(default_factory=lambda: SQLServerQuotePolicy())
disable_empty_relation_aliases: ClassVar[bool] = True

@classproperty
def get_relation_type(cls) -> Type[SQLServerRelationType]:
return SQLServerRelationType

def _render_limited_alias(self) -> str:
if self.disable_empty_relation_aliases:
return ""

return super()._render_limited_alias()

def render_limited(self) -> str:
rendered = self.render()
if self.limit is None:
return rendered
elif self.limit == 0:
return f"(select * from {rendered} where 1=0) AS {self._render_limited_alias()}"
return f"(select * from {rendered} where 1=0){self._render_limited_alias()}"
else:
return f"(select TOP {self.limit} * from {rendered}) AS {self._render_limited_alias()}"
return f"(select TOP {self.limit} * from {rendered}){self._render_limited_alias()}"

def __post_init__(self):
# Check for length of Redshift table/view names.
Expand Down
56 changes: 53 additions & 3 deletions tests/functional/adapter/dbt/test_empty.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,14 +12,21 @@

model_sql_sqlserver = """
select *
from {{ ref('model_input') }}
from {{ ref('model_input') }} as model_input_alias
union all
select *
from {{ source('seed_sources', 'raw_source') }}
from {{ source('seed_sources', 'raw_source') }} as raw_source_alias
"""

model_inline_sql_sqlserver = """
select * from {{ source('seed_sources', 'raw_source') }}
select * from {{ source('seed_sources', 'raw_source') }} as raw_source_alias
"""

model_sql_user_alias_sqlserver = """
select user_alias.id as id
from {{ ref('model_input') }} as user_alias
inner join {{ source('seed_sources', 'raw_source') }} as source_alias
on user_alias.id = source_alias.id
"""


Expand Down Expand Up @@ -47,6 +54,49 @@ def test_run_with_empty(self, project):
self.assert_row_count(project, "model", 0)


class TestEmptyWithUserAlias(BaseTestEmpty):
@pytest.fixture(scope="class")
def models(self):
return {
"model_input.sql": model_input_sql,
"model.sql": model_sql_user_alias_sqlserver,
"sources.yml": schema_sources_yml,
}

def test_run_with_empty(self, project):
run_dbt(["seed"])

run_dbt(["run", "--empty"])
self.assert_row_count(project, "model", 0)


@pytest.mark.xfail(
reason="Upstream dbt empty-mode alias handling needs to be contextual aware.",
)
class TestEmptyWithUserAliasAndNoAliasFlag(BaseTestEmpty):
@pytest.fixture(scope="class")
def models(self):
return {
"model_input.sql": model_input_sql,
"model.sql": model_sql_user_alias_sqlserver,
"sources.yml": schema_sources_yml,
}

@pytest.fixture(scope="class")
def project_config_update(self):
return {
"flags": {
"dbt_sqlserver_disable_empty_relation_aliases": False,
}
}

def test_run_with_empty(self, project):
run_dbt(["seed"])

run_dbt(["run", "--empty"])
self.assert_row_count(project, "model", 0)


class TestemptyInlineSourceRef(BaseTestEmptyInlineSourceRef):
@pytest.fixture(scope="class")
def models(self):
Expand Down