如何在SQL单元测试执行完成后运行自定义代码更新数据库

核心结论 #

• 不存在仅在所有单元测试100%执行成功时才触发的内置专属事件
• Visual Studio SQL单元测试底层基于MSTest框架实现，框架提供了覆盖所有执行场景的生命周期钩子，无论测试成功、断言失败、数据库连接超时、测试初始化报错，对应的钩子方法都会被触发，完全可以实现自定义逻辑注入

为什么SqlExecutionResult[]会返回null #

你在TestMethod内部获取的testResults属于SQL执行环节的局部返回对象，只有在数据库连接正常、SQL批处理完整走完执行流程的场景下才会被赋值。如果出现连接超时、SQL语法严重报错、测试初始化失败等中断执行流的场景，代码会直接跳出测试方法，根本不会走到给testResults赋值的逻辑，依赖这个对象判断测试结果必然会出现漏判。

稳定捕获全场景测试结果的实现方案 #

不需要安装第三方插件，直接用MSTest原生特性即可实现，所有SQL单元测试类继承公共基类就能接入逻辑，对现有测试代码侵入性极低：

• 单个测试执行完成（无论成败）的通用入口：给基类方法添加[TestCleanup]特性
• 整个测试程序集所有测试执行完成的全局入口：给静态方法添加[AssemblyCleanup]特性
• 结果判断不要依赖testResults.hasError，直接读取框架注入的TestContext.CurrentTestOutcome属性，这个属性由测试框架本身在全流程维护，覆盖所有成功/失败场景，不会出现空值漏判

参考实现代码 #

using Microsoft.VisualStudio.TestTools.UnitTesting;
using System.Data.SqlClient;

/// <summary>
/// 所有SQL单元测试类继承该基类，即可自动接入SPROC测试状态跟踪逻辑
/// </summary>
[TestClass]
public class BaseSqlTest
{
    // 框架自动注入当前测试的上下文对象，无需手动赋值
    public TestContext TestContext { get; set; }

    [TestCleanup]
    public void TrackSingleTestResult()
    {
        // 读取当前测试最终执行结果，覆盖所有成功、失败、异常、超时场景
        bool testPassed = TestContext.CurrentTestOutcome == UnitTestOutcome.Passed;
        string currentTestMethod = TestContext.TestName;
        string errorInfo = TestContext.TestRunException?.Message ?? string.Empty;
        
        // 按照团队的测试方法命名规则，从方法名提取关联的SPROC名称
        string targetSprocName = ExtractSprocName(currentTestMethod);
        
        // 执行数据库更新逻辑，写入你已有的SPROC跟踪表
        UpdateTestStatusToTable(targetSprocName, testPassed, errorInfo, DateTime.Now);
    }

    [AssemblyCleanup]
    public static void AfterAllTestsFinished()
    {
        // 此处可编写全量测试执行完成后的批量逻辑，比如统计测试通过率、生成SPROC变更测试报告
        // 该方法在测试资源管理器单次触发的所有测试执行完毕后，仅运行一次
    }

    private string ExtractSprocName(string testMethodName)
    {
        // 示例规则：测试方法统一命名为{SPROC名称}_Test，去除后缀即可得到SPROC名，可按团队实际规则调整
        return testMethodName.Replace("_Test", string.Empty);
    }

    private void UpdateTestStatusToTable(string sprocName, bool isPassed, string errorMsg, DateTime execTime)
    {
        // 编写数据库操作逻辑，连接业务库更新SPROC对应的测试状态字段即可
        using (var conn = new SqlConnection("你的业务跟踪库连接字符串"))
        {
            // 具体UPDATE语句逻辑省略
        }
    }
}

方案说明 #

• 现有测试逻辑不需要做任何修改，只要把所有SQL单元测试类的父类改成BaseSqlTest即可自动生效
• 不会漏判任何异常场景，哪怕测试在初始化阶段就抛出异常，[TestCleanup]方法依然会执行，CurrentTestOutcome也会准确标记为失败状态
• 支持单测试粒度实时更新、全量测试跑完后批量更新两种模式，可以根据你现有跟踪表的结构灵活调整

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