单元测试场景下实现线程等待对象更新的最佳实践

我完全理解你在单元测试里的需求——让多个工作线程在执行到某个节点时暂停，等主线程完成状态检查后再继续，而且不想用那种轮询布尔值的笨办法（毕竟轮询既浪费CPU，代码也不够优雅）。这里有几个更合适的.NET线程同步方案，完美匹配你的场景：

1. 使用 ManualResetEventSlim（轻量级单信号同步） #

这是最直接的方案，适合你需要主线程手动触发“继续”信号的场景。它是传统ManualResetEvent的轻量级版本，性能更好，适合短等待场景。

public void Test()
{
    // 初始化一个未触发的事件（false表示初始状态为等待）
    using var continueSignal = new ManualResetEventSlim(false);

    var task1 = Task.Run(() =>
    {
        // 第一阶段操作
        Console.WriteLine("Task 1 completed first stage");
        // 等待主线程的继续信号
        continueSignal.Wait();
        // 第二阶段操作
        Console.WriteLine("Task 1 completed second stage");
    });

    var task2 = Task.Run(() =>
    {
        // 第一阶段操作
        Console.WriteLine("Task 2 completed first stage");
        // 等待主线程的继续信号
        continueSignal.Wait();
        // 第二阶段操作
        Console.WriteLine("Task 2 completed second stage");
    });

    // 主线程执行状态检查
    Console.WriteLine("Main thread verifying object state...");
    // 这里添加你的状态验证逻辑，比如Assert断言

    // 触发信号，让所有等待的线程继续执行
    continueSignal.Set();

    // 等待所有任务完成
    Task.WaitAll(task1, task2);
    // 执行后续检查
    Console.WriteLine("Main thread completed post-checks");
}

优势：API简单直观，资源占用少，支持using自动释放，非常适合单元测试的短周期场景。

2. 使用 Barrier（多线程同步屏障） #

如果你需要确保所有线程都到达指定执行阶段后，主线程再进行状态检查，Barrier会是更省心的选择。它会自动计数等待的线程，当所有参与者都到达屏障时，才会一起继续执行。

public void Test()
{
    // 参与者数量：2个工作线程 + 主线程自己
    using var syncBarrier = new Barrier(3, barrier =>
    {
        // 这个回调会在所有参与者到达屏障时执行
        Console.WriteLine("All threads reached the barrier! Main thread checking state...");
        // 这里添加状态验证逻辑，确保所有线程都处于预期阶段
    });

    var task1 = Task.Run(() =>
    {
        // 第一阶段操作
        Console.WriteLine("Task 1 completed first stage");
        // 通知屏障自己已到达，然后等待其他线程
        syncBarrier.SignalAndWait();
        // 第二阶段操作
        Console.WriteLine("Task 1 completed second stage");
    });

    var task2 = Task.Run(() =>
    {
        // 第一阶段操作
        Console.WriteLine("Task 2 completed first stage");
        // 通知屏障自己已到达，然后等待其他线程
        syncBarrier.SignalAndWait();
        // 第二阶段操作
        Console.WriteLine("Task 2 completed second stage");
    });

    // 主线程也到达屏障，触发回调并继续
    syncBarrier.SignalAndWait();

    // 等待所有任务完成
    Task.WaitAll(task1, task2);
    // 执行后续检查
    Console.WriteLine("Main thread completed post-checks");
}

优势：自动处理线程同步的计数逻辑，不需要手动确认每个线程是否到位，完美匹配你“所有线程处于同一执行阶段”的需求。

3. 使用 TaskCompletionSource<T>（异步优先方案） #

如果你的单元测试采用异步编程模型，TaskCompletionSource<T>是更符合异步风格的选择，它通过任务的方式实现等待，避免阻塞线程。

public void Test()
{
    var continueTcs = new TaskCompletionSource<bool>();

    var task1 = Task.Run(async () =>
    {
        // 第一阶段操作
        Console.WriteLine("Task 1 completed first stage");
        // 异步等待信号
        await continueTcs.Task;
        // 第二阶段操作
        Console.WriteLine("Task 1 completed second stage");
    });

    var task2 = Task.Run(async () =>
    {
        // 第一阶段操作
        Console.WriteLine("Task 2 completed first stage");
        // 异步等待信号
        await continueTcs.Task;
        // 第二阶段操作
        Console.WriteLine("Task 2 completed second stage");
    });

    // 主线程执行状态检查
    Console.WriteLine("Main thread verifying object state...");
    // 这里添加状态验证逻辑

    // 发送继续信号
    continueTcs.SetResult(true);

    // 等待所有任务完成
    Task.WaitAll(task1, task2);
    // 执行后续检查
    Console.WriteLine("Main thread completed post-checks");
}

优势：完全适配异步编程模型，不会造成线程阻塞，适合现代异步单元测试场景。

方案选择建议 #

• 如果只是需要简单的“主线程发信号让所有线程继续”：选ManualResetEventSlim，简单高效。
• 如果需要确保所有线程都到达等待点再检查：选Barrier，自动同步更省心。
• 如果是异步单元测试：选TaskCompletionSource<T>，符合异步编程习惯。

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