如何编写自动化测试验证对象是否可被GC回收以检测内存泄漏？

我有一个包含事件订阅内部逻辑的对象，希望编写自动化测试来检查该对象在执行特定步骤后能否被GC回收。示例代码如下：

public interface IEventPublisher {
    event Action Evt;
}

public class MyClass : IDisposable {
    private IEventPublisher _evtPub; // 修正拼写错误：IEventPubliser → IEventPublisher
    public MyClass(IEventPublisher evtPub) {
        _evtPub = evtPub;
    }
    public void Initialize() {
        _evtPub.Evt += EvtHandler; // 修正：EvtHandler() → EvtHandler，事件订阅应传入方法组而非调用方法
    }
    private void EvtHandler() {
        // 一些业务逻辑
    }
    public void Dispose() {
        _evtPub.Evt -= EvtHandler; // 同样修正：EvtHandler() → EvtHandler
    }
}

[Test]
public void MemoryLeakTest() {
    WeakReference weakRef = null;
    {
        // Arrange 
        var evtPub = Substitute.For<IEventPublisher>(); // 使用NSubstitute创建Mock
        var myObj = new MyClass(evtPub);
        weakRef = new WeakReference(myObj);
        
        // Act
        myObj.Initialize();
        // 执行一些操作
        myObj.Dispose();
    } // 此处myObj的局部作用域结束，没有强引用指向它
    
    // Assert
    // 触发GC并等待回收完成
    GC.Collect();
    GC.WaitForPendingFinalizers();
    // 再次触发GC，确保终结器队列处理完毕
    GC.Collect();
    
    Assert.IsFalse(weakRef.IsAlive, "MyClass实例未被GC回收，存在内存泄漏风险");
}

核心问题解答 #

一、断言部分的实现代码 #

上述测试中的断言逻辑已给出，核心思路是：

• 用WeakReference跟踪目标对象，弱引用不会阻止GC回收该对象
• 手动触发GC（GC.Collect()）并等待终结器执行完成（GC.WaitForPendingFinalizers()），二次触发GC确保终结器队列处理完毕
• 检查weakRef.IsAlive是否为false，若是则说明对象已被成功回收

二、这种测试是否可行？ #

可行，但需要注意几个关键点：

• GC的非确定性：手动触发GC只是建议CLR执行回收，无法100%保证立即完成，测试偶尔可能出现不稳定情况。可以增加重试逻辑，或多次触发GC后再检查。
• 编译模式影响：Debug模式下CLR优化较少，局部变量的生命周期可能被延长，建议在Release模式下运行这类测试，或给测试方法添加[MethodImpl(MethodImplOptions.NoOptimization)]特性，避免编译器优化保留强引用。
• Mock对象的潜在问题：确保使用的Mock框架（如NSubstitute）不会意外持有对MyClass的强引用，成熟框架一般不会出现此问题，但需提前确认。
• 自定义终结器的影响：如果MyClass有自定义终结器，要确保终结器逻辑不会将对象重新赋值给强引用，否则会导致对象无法被回收，影响测试结果。

另外需注意示例代码中的两个错误：

1. IEventPubliser拼写错误，应为IEventPublisher
2. 事件订阅/取消订阅时，EvtHandler()写法错误（会编译失败），正确写法是传入方法组EvtHandler

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