✨ feat: add fixture and update secondary database cleaning for tests

Author: axellpadilla

tests/functional/adapter/dbt/test_utils.py

@@ -1,5 +1,10 @@
 import pytest
-from dbt.tests.adapter.utils import fixture_cast_bool_to_text, fixture_dateadd, fixture_listagg
+from dbt.tests.adapter.utils import (
+    fixture_cast_bool_to_text,
+    fixture_dateadd,
+    fixture_listagg,
+    fixture_split_part,
+)
 from dbt.tests.adapter.utils.test_any_value import BaseAnyValue
 from dbt.tests.adapter.utils.test_array_append import BaseArrayAppend
 from dbt.tests.adapter.utils.test_array_concat import BaseArrayConcat
@@ -340,7 +345,18 @@ class TestSafeCast(BaseSafeCast):
 
 
 class TestSplitPart(BaseSplitPart):
-    pass
+    @pytest.fixture(scope="class")
+    def models(self):
+        model_sql = """
+-- depends_on: {{ ref('data_split_part') }}
+""" + self.interpolate_macro_namespace(
+            fixture_split_part.models__test_split_part_sql, "split_part"
+        )
+
+        return {
+            "test_split_part.yml": fixture_split_part.models__test_split_part_yml,
+            "test_split_part.sql": model_sql,
+        }
 
 
 class TestStringLiteral(BaseStringLiteral):

tests/functional/adapter/mssql/test_cross_db.py

@@ -55,7 +55,15 @@ def create_secondary_db(self, project):
             )
 
     def cleanup_secondary_database(self, project):
-        drop_sql = "DROP DATABASE IF EXISTS secondary_db"
+        drop_sql = """
+        USE [master]
+
+        IF EXISTS (SELECT * FROM sys.databases WHERE name = 'secondary_db')
+        BEGIN
+            ALTER DATABASE [secondary_db] SET SINGLE_USER WITH ROLLBACK IMMEDIATE
+            DROP DATABASE [secondary_db]
+        END
+        """
         with get_connection(project.adapter):
             project.adapter.execute(
                 drop_sql.format(database=project.database),