Add mssql integration tests and relocate existing unit tests

.github/workflows/integration-tests.yml

@@ -140,6 +140,11 @@ jobs:
           breeze testing integration-tests --integration qdrant
           breeze down
         if: inputs.is-airflow-runner != 'true' && matrix.backend == 'postgres'
+      - name: "Integration Tests Postgres: mssql"
+        run: |
+          breeze testing integration-tests --integration mssql
+          breeze down
+        if: inputs.is-airflow-runner != 'true' && matrix.backend == 'postgres'
       # For Self Hosted runners we run complete set of integration tests
       # in one big test - as we have enough resources to start all integrations
       # and we do that for both - postgres and mysql

airflow/providers/microsoft/mssql/hooks/mssql.py

@@ -23,6 +23,7 @@
 from typing import TYPE_CHECKING, Any
 
 import pymssql
+from pymssql import Connection as PymssqlConnection
 
 from airflow.providers.common.sql.hooks.sql import DbApiHook
 
@@ -104,23 +105,23 @@ def get_sqlalchemy_connection(
         engine = self.get_sqlalchemy_engine(engine_kwargs=engine_kwargs)
         return engine.connect(**(connect_kwargs or {}))
 
-    def get_conn(self) -> pymssql.connect:
+    def get_conn(self) -> PymssqlConnection:
         """Return ``pymssql`` connection object."""
         conn = self.connection
         return pymssql.connect(
             server=conn.host,
             user=conn.login,
             password=conn.password,
             database=self.schema or conn.schema,
-            port=conn.port,
+            port=str(conn.port),
         )
 
     def set_autocommit(
         self,
-        conn: pymssql.connect,
+        conn: PymssqlConnection,
         autocommit: bool,
     ) -> None:
         conn.autocommit(autocommit)
 
-    def get_autocommit(self, conn: pymssql.connect):
+    def get_autocommit(self, conn: PymssqlConnection):
         return conn.autocommit_state

contributing-docs/testing/integration_tests.rst

@@ -58,6 +58,8 @@ The following integrations are available:
 +--------------+----------------------------------------------------+
 | mongo        | Integration required for MongoDB hooks.            |
 +--------------+----------------------------------------------------+
+| mssql        | Integration required for mssql hooks.              |
++--------------+----------------------------------------------------+
 | openlineage  | Integration required for Openlineage hooks.        |
 +--------------+----------------------------------------------------+
 | otel         | Integration required for OTEL/opentelemetry hooks. |