.NET Core单元测试中如何设置System.Diagnostics.Activity.Current

嘿，这个问题我太熟悉了！之前帮团队排查过类似的单元测试坑——核心原因就是单元测试环境里没有Web请求的HttpContext，System.Diagnostics.Activity.Current不会自动被初始化，而你之前只是new了Activity但没启动它，所以还是null。给你三个靠谱的解决办法，按需选：

解决方案1：手动启动Activity（快速修复） #

你之前的测试代码只是创建了Activity对象，但没有调用Start()方法——这个方法才会把当前Activity设置为Activity.Current。而且最好用using包裹，确保测试结束后自动清理，避免影响其他测试：

[TestMethod]
public void MyTest() {
    // 创建并启动Activity，自动绑定到Activity.Current
    using var testActivity = new Activity("MyUnitTests").Start();
    
    Company c = new Company();
    Assert.AreNotEqual(string.Empty, c.CorrelationId);
}

这样运行测试时，Company.CorrelationId就能正常拿到Activity的Id了，空引用异常直接消失。

解决方案2：解耦依赖（优雅长期方案） #

从领域驱动设计的角度看，领域类直接依赖System.Diagnostics.Activity这种框架级API其实不太合理，耦合度太高，也不利于测试。我们可以通过依赖注入解耦：

第一步：定义一个获取关联ID的接口 #

public interface ICorrelationIdProvider {
    string GetCorrelationId();
}

第二步：修改领域类，通过构造函数注入依赖 #

public class Company {
    public int CompanyId { get; private set; }
    public string Name { get; private set; }
    public string CorrelationId { get; }

    public Company(ICorrelationIdProvider correlationIdProvider) {
        CorrelationId = correlationIdProvider.GetCorrelationId() 
            ?? throw new ArgumentNullException(nameof(correlationIdProvider));
    }
}

第三步：在Web API中实现生产环境的Provider #

public class ActivityBasedCorrelationIdProvider : ICorrelationIdProvider {
    public string GetCorrelationId() {
        // 生产环境用Activity.Current.Id， fallback到Guid避免null
        return System.Diagnostics.Activity.Current?.Id ?? Guid.NewGuid().ToString();
    }
}

记得在API的DI容器里注册这个实现：services.AddScoped<ICorrelationIdProvider, ActivityBasedCorrelationIdProvider>();

第四步：单元测试中使用模拟Provider #

// 测试用的Provider，返回固定值或者自定义ID
public class TestCorrelationIdProvider : ICorrelationIdProvider {
    public string GetCorrelationId() => "test-correlation-id-12345";
}

[TestMethod]
public void MyTest() {
    var testProvider = new TestCorrelationIdProvider();
    Company c = new Company(testProvider);
    Assert.AreEqual("test-correlation-id-12345", c.CorrelationId);
}

这个方案不仅解决了测试问题，还让领域类更符合单一职责原则，后续如果要换关联ID的生成方式（比如用HttpContext的TraceIdentifier），只需要换Provider实现就行，完全不用改领域类。

解决方案3：全局测试初始化（批量处理） #

如果你的测试很多，不想每个测试都写启动Activity的代码，可以在测试类的初始化方法里全局启动一个Activity，测试结束后清理：

private Activity _globalTestActivity;

[TestInitialize]
public void TestSetup() {
    // 每个测试开始前启动Activity
    _globalTestActivity = new Activity("GlobalUnitTestActivity").Start();
}

[TestCleanup]
public void TestTeardown() {
    // 每个测试结束后停止并释放Activity
    _globalTestActivity?.Stop();
    _globalTestActivity?.Dispose();
}

[TestMethod]
public void MyTest() {
    Company c = new Company();
    Assert.AreNotEqual(string.Empty, c.CorrelationId);
}

注意：这个方式要确保测试之间的隔离，比如不要在测试间共享Activity，否则可能出现测试污染的问题。

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