单元测试中模拟工作负载是否有更优的实现方式？

问题背景 #

团队刚引入单元测试，现有一段测试TimeoutHandler超时逻辑的代码，通过局部random变量结合三角函数运算模拟工作负载，并且需要调用GC.KeepAlive防止JIT优化掉无实际输出的循环逻辑。由于该模拟逻辑需要在多处复用，已构思了初步优化实现，现确认方案优劣并寻求更简洁的方式。

原测试核心代码：

// Act
timeoutHandler = new TimeoutHandler(TimeSpan.FromMilliseconds(1));
preTimeout = timeoutHandler.IsTimeout;
double random = new Random().NextDouble();
while(!timeoutHandler.IsTimeout)
{
    // Simulate some work
    random = Math.Sin(random) * Math.Cos(random);
}
GC.KeepAlive(random);
postTimeout = timeoutHandler.IsTimeout;

初步优化实现：

// Act
timeoutHandler = new TimeoutHandler(TimeSpan.FromMilliseconds(1));
preTimeout = timeoutHandler.IsTimeout;
double? rnd = null;
while(!timeoutHandler.IsTimeout)
    rnd = SimulateWork(rnd);
postTimeout = timeoutHandler.IsTimeout;

// 复用的模拟方法
public static double SimulateWork(double? rnd = null)
{
    rnd ??= new Random().NextDouble();
    return Math.Sin(rnd.Value) * Math.Cos(rnd.Value);
}

初步实现的优缺点分析 #

• 优点：
  • 成功将模拟工作负载的逻辑抽离为可复用的静态方法，符合DRY原则
  • 用double?结合??=初始化随机值，避免了每次调用创建Random实例的问题
  • 无需再调用GC.KeepAlive：因为循环中持续给局部变量rnd赋值，JIT编译器不会将这段逻辑判定为无意义并优化掉
• 缺点：
  • 调用处仍需维护rnd局部变量，增加了测试代码的冗余，不够简洁

更优的优化方案 #

可以将「等待超时+模拟工作」的完整逻辑封装为独立方法，彻底消除调用处的局部变量，让测试代码聚焦于核心断言逻辑：

方案1：封装固定逻辑的等待方法

如果模拟工作负载的逻辑不需要变动，直接封装成专用方法：

public static void WaitForTimeout(TimeoutHandler timeoutHandler)
{
    double state = new Random().NextDouble();
    while (!timeoutHandler.IsTimeout)
    {
        state = Math.Sin(state) * Math.Cos(state);
    }
}

测试代码中调用时只需一行：

// Act
timeoutHandler = new TimeoutHandler(TimeSpan.FromMilliseconds(1));
preTimeout = timeoutHandler.IsTimeout;
WaitForTimeout(timeoutHandler); // 直接调用封装方法
postTimeout = timeoutHandler.IsTimeout;

方案2：支持自定义工作负载的通用方法

如果需要在不同测试中使用不同的工作负载模拟逻辑，可以通过委托实现通用化：

public static void WaitForTimeout(TimeoutHandler timeoutHandler, Func<double, double> workSimulator)
{
    double state = new Random().NextDouble();
    while (!timeoutHandler.IsTimeout)
    {
        state = workSimulator(state);
    }
}

// 默认的工作负载模拟方法
public static double DefaultWorkSimulator(double state)
{
    return Math.Sin(state) * Math.Cos(state);
}

调用时可使用默认逻辑，也可传入自定义逻辑：

// 使用默认模拟
WaitForTimeout(timeoutHandler, DefaultWorkSimulator);

// 自定义模拟逻辑示例
WaitForTimeout(timeoutHandler, s => Math.Tan(s) * Math.Log(Math.Abs(s) + 1));

总结 #

你的初步实现已经比原代码更优，解决了逻辑复用和GC.KeepAlive的问题。在此基础上，通过封装完整的等待超时方法，可以进一步简化测试代码，让测试逻辑更清晰、更易维护。

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