Visual Studio中SQL Server单元测试如何在运行前自动清空数据库

方案1：配置SQL Server项目发布规则（推荐） #

无需修改代码，直接通过数据库项目的发布配置即可实现自动清空旧库：

• 右键你的SQL Server数据库项目，选择「发布」
• 在弹出的发布窗口点击右下角的「高级」按钮
• 在高级部署设置的「常规」标签页，勾选 如果可能则重新创建数据库，同时可勾选 删除目标中存在但项目中不存在的对象
• 保存该发布配置为profile，单元测试的数据库部署流程会自动沿用该配置，每次发布前都会直接删除原有数据库，重新创建全新的库，无需手动写清理逻辑。

方案2：手动创建master库连接执行删库操作 #

你之前报错的原因是TestContext.DataConnection默认连接的是目标测试库，在程序集初始化阶段还未完成初始化，且要删除数据库本身也需要连接到master库才能操作，你可以自行构造master库的连接来执行删库逻辑：

using System.Data.SqlClient;

[TestClass()]
public class SqlDatabaseSetup
{
    [AssemblyInitialize()]
    public static void InitializeAssembly(TestContext ctx)
    {
        // 连接字符串可从配置文件读取，无需硬编码，只需保证连接到master库、有删库权限即可
        var masterConnString = "Data Source=你的SQL实例地址;Initial Catalog=master;Integrated Security=True;";
        using (var conn = new SqlConnection(masterConnString))
        {
            conn.Open();
            var cmd = conn.CreateCommand();
            // 先切单用户模式杀所有活跃连接，避免删库时被占用报错
            cmd.CommandText = @"
                IF EXISTS(SELECT 1 FROM sys.databases WHERE name = 'MyDb')
                BEGIN
                    ALTER DATABASE MyDb SET SINGLE_USER WITH ROLLBACK IMMEDIATE;
                    DROP DATABASE MyDb;
                END";
            cmd.ExecuteNonQuery();
        }

        // 执行原有部署逻辑
        SqlDatabaseTestClass.TestService.DeployDatabaseProject();
        SqlDatabaseTestClass.TestService.GenerateData();
    }
}

方案3：单测试用例级别的干净环境（可选） #

如果你需要每个测试方法执行完都重置数据，不需要每次都重新部署整个数据库，可以用事务回滚的方案：

• 封装一个测试基类，所有测试类继承该基类
• 在基类的[TestInitialize]方法中开启事务，所有测试的增删改操作都在该事务中执行
• 在基类的[TestCleanup]方法中回滚事务，所有测试产生的数据都会被清空，速度远高于重新部署数据库，适合测试用例较多的场景。

内容的提问来源于stack exchange，提问作者WynDiesel