Updated to 1.8 and rebuilt test suite

Author: cody-scott

.devcontainer/devcontainer.json

@@ -5,7 +5,7 @@
         "ghcr.io/devcontainers/features/docker-in-docker:2": {}
     },
     "forwardPorts": [1433],
-    "postStartCommand": "sudo bash .devcontainer/setup_odbc.sh && bash .devcontainer/install_pyenv.sh && bash .devcontainer/setup_env.sh",
+    "postStartCommand": "/bin/bash ./.devcontainer/setup_odbc.sh & /bin/bash ./.devcontainer/setup_env.sh",
     "containerEnv": {
         "SQLSERVER_TEST_DRIVER": "ODBC Driver 18 for SQL Server",
         "SQLSERVER_TEST_HOST": "127.0.0.1",

.devcontainer/setup_env.sh

@@ -1,8 +1,8 @@
 cp test.env.sample test.env
 
-pyenv install 3.10.7
-pyenv virtualenv 3.10.7 dbt-sqlserver
-pyenv activate dbt-sqlserver
+docker compose build
+docker compose up -d
 
-make dev
-make server
+pip install -e .
+
+pip install -r dev_requirements.txt

dbt/adapters/sqlserver/__init__.py

@@ -1,10 +1,10 @@
 from dbt.adapters.base import AdapterPlugin
 
-from dbt.adapters.sqlserver.sql_server_adapter import SQLServerAdapter
-from dbt.adapters.sqlserver.sql_server_column import SQLServerColumn
-from dbt.adapters.sqlserver.sql_server_configs import SQLServerConfigs
-from dbt.adapters.sqlserver.sql_server_connection_manager import SQLServerConnectionManager
-from dbt.adapters.sqlserver.sql_server_credentials import SQLServerCredentials
+from dbt.adapters.sqlserver.sqlserver_adapter import SQLServerAdapter
+from dbt.adapters.sqlserver.sqlserver_column import SQLServerColumn
+from dbt.adapters.sqlserver.sqlserver_configs import SQLServerConfigs
+from dbt.adapters.sqlserver.sqlserver_connections import SQLServerConnectionManager  # noqa
+from dbt.adapters.sqlserver.sqlserver_credentials import SQLServerCredentials
 from dbt.include import sqlserver
 
 Plugin = AdapterPlugin(

dbt/adapters/sqlserver/__version__.py

@@ -1 +1 @@
-version = "1.7.4"
+version = "1.8.0"

dbt/adapters/sqlserver/relation_configs/__init__.py

@@ -0,0 +1,13 @@
+from dbt.adapters.sqlserver.relation_configs.policies import (
+    MAX_CHARACTERS_IN_IDENTIFIER,
+    SQLServerIncludePolicy,
+    SQLServerQuotePolicy,
+    SQLServerRelationType,
+)
+
+__all__ = [
+    "MAX_CHARACTERS_IN_IDENTIFIER",
+    "SQLServerIncludePolicy",
+    "SQLServerQuotePolicy",
+    "SQLServerRelationType",
+]

dbt/adapters/sqlserver/relation_configs/policies.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass
+
+from dbt.adapters.contracts.relation import Policy
+from dbt_common.dataclass_schema import StrEnum
+
+MAX_CHARACTERS_IN_IDENTIFIER = 127
+
+
+class SQLServerRelationType(StrEnum):
+    Table = "table"
+    View = "view"
+    CTE = "cte"
+
+
+class SQLServerIncludePolicy(Policy):
+    database: bool = True
+    schema: bool = True
+    identifier: bool = True
+
+
+@dataclass
+class SQLServerQuotePolicy(Policy):
+    database: bool = True
+    schema: bool = True
+    identifier: bool = True

dbt/adapters/sqlserver/sql_server_adapter.py

@@ -0,0 +1,61 @@
+from typing import Optional
+
+import dbt.exceptions
+from dbt.adapters.base.impl import ConstraintSupport
+from dbt.adapters.fabric import FabricAdapter
+from dbt.contracts.graph.nodes import ConstraintType
+
+from dbt.adapters.sqlserver.sqlserver_column import SQLServerColumn
+from dbt.adapters.sqlserver.sqlserver_connections import SQLServerConnectionManager
+from dbt.adapters.sqlserver.sqlserver_relation import SQLServerRelation
+
+
+class SQLServerAdapter(FabricAdapter):
+    """
+    Controls actual implmentation of adapter, and ability to override certain methods.
+    """
+
+    ConnectionManager = SQLServerConnectionManager
+    Column = SQLServerColumn
+    Relation = SQLServerRelation
+
+    CONSTRAINT_SUPPORT = {
+        ConstraintType.check: ConstraintSupport.ENFORCED,
+        ConstraintType.not_null: ConstraintSupport.ENFORCED,
+        ConstraintType.unique: ConstraintSupport.ENFORCED,
+        ConstraintType.primary_key: ConstraintSupport.ENFORCED,
+        ConstraintType.foreign_key: ConstraintSupport.ENFORCED,
+    }
+
+    @classmethod
+    def render_model_constraint(cls, constraint) -> Optional[str]:
+        constraint_prefix = "add constraint "
+        column_list = ", ".join(constraint.columns)
+
+        if constraint.name is None:
+            raise dbt.exceptions.DbtDatabaseError(
+                "Constraint name cannot be empty. Provide constraint name  - column "
+                + column_list
+                + " and run the project again."
+            )
+
+        if constraint.type == ConstraintType.unique:
+            return constraint_prefix + f"{constraint.name} unique nonclustered({column_list})"
+        elif constraint.type == ConstraintType.primary_key:
+            return constraint_prefix + f"{constraint.name} primary key nonclustered({column_list})"
+        elif constraint.type == ConstraintType.foreign_key and constraint.expression:
+            return (
+                constraint_prefix
+                + f"{constraint.name} foreign key({column_list}) references "
+                + constraint.expression
+            )
+        elif constraint.type == ConstraintType.check and constraint.expression:
+            return f"{constraint_prefix} {constraint.name} check ({constraint.expression})"
+        elif constraint.type == ConstraintType.custom and constraint.expression:
+            return f"{constraint_prefix} {constraint.name} {constraint.expression}"
+        else:
+            return None
+
+    @classmethod
+    def date_function(cls):
+        return "getdate()"

dbt/adapters/sqlserver/sql_server_column.py

@@ -0,0 +1,22 @@
+from dbt.adapters.fabric import FabricColumn
+
+
+class SQLServerColumn(FabricColumn):
+    def is_integer(self) -> bool:
+        return self.dtype.lower() in [
+            # real types
+            "smallint",
+            "integer",
+            "bigint",
+            "smallserial",
+            "serial",
+            "bigserial",
+            # aliases
+            "int2",
+            "int4",
+            "int8",
+            "serial2",
+            "serial4",
+            "serial8",
+            "int",
+        ]