C#单元测试中Mock DateTime.UtcNow的问题与解决方案

我在.NET 6环境下尝试用Pose库Mock DateTime.UtcNow时踩了坑——执行到PoseContext.Isolate()时测试直接卡住，终止测试时Visual Studio还会冻结甚至强制重启。

我清楚引入接口、依赖注入是Mock时间的常规操作，但不想为了Mock一个时间就额外加服务依赖，希望找个不用大幅改动业务代码的替代方案。

最终用Ambient Context模式解决了问题，只需要把业务代码里的DateTime.UtcNow替换成DateTimeProvider.UtcNow，再实现两个轻量辅助类就行，完全不用新增服务。

实现代码 #

首先是核心的时间提供类和上下文类：

public static class DateTimeProvider
{
    private static readonly AsyncLocal<Func<DateTime>> _utcNowOverride = new();

    public static DateTime UtcNow => _utcNowOverride.Value?.Invoke() ?? DateTime.UtcNow;

    public static IDisposable SetUtcNow(Func<DateTime> utcNowFunc)
    {
        var original = _utcNowOverride.Value;
        _utcNowOverride.Value = utcNowFunc;
        return new Disposable(() => _utcNowOverride.Value = original);
    }

    private class Disposable : IDisposable
    {
        private readonly Action _disposeAction;

        public Disposable(Action disposeAction) => _disposeAction = disposeAction;

        public void Dispose() => _disposeAction();
    }
}

// 封装成上下文类，测试时用起来更直观
public class DateTimeProviderContext : IDisposable
{
    private readonly IDisposable _scope;

    public DateTimeProviderContext(DateTime fixedUtcNow)
    {
        _scope = DateTimeProvider.SetUtcNow(() => fixedUtcNow);
    }

    public DateTimeProviderContext(Func<DateTime> utcNowFunc)
    {
        _scope = DateTimeProvider.SetUtcNow(utcNowFunc);
    }

    public void Dispose() => _scope.Dispose();
}

业务代码修改 #

只需要把所有直接调用DateTime.UtcNow的地方替换成DateTimeProvider.UtcNow：

// 原来的代码
var currentTime = DateTime.UtcNow;

// 修改后
var currentTime = DateTimeProvider.UtcNow;

单元测试示例 #

测试时用using块包裹，就能在指定范围内固定时间：

[Fact]
public void DoSomething_WithFixedTime_ReturnsCorrectTimestamp()
{
    var fixedTestTime = new DateTime(2024, 5, 20, 14, 30, 0, DateTimeKind.Utc);

    using (new DateTimeProviderContext(fixedTestTime))
    {
        // 执行待测试的业务逻辑
        var service = new MyBusinessService();
        var result = service.DoSomething();

        // 断言结果中的时间与固定测试时间一致
        Assert.Equal(fixedTestTime, result.Timestamp);
    }
}

这种方案改动极小，不需要引入依赖注入容器，也不会出现Pose库在.NET 6中的卡顿、IDE冻结问题，完美满足了不想新增服务的需求。

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