如何编写单元测试验证Buffer等内部资源是否被正确释放

要解决方法内部私有Buffer资源的生命周期校验问题，核心思路是拦截资源分配入口，给资源添加可观测的释放标记，不需要侵入业务逻辑就能实现泄漏校验，具体可按场景选择以下方案：

方案1：依赖注入改造（推荐新业务使用） #

不要在业务方法中硬编码静态资源分配逻辑，将Buffer分配能力抽象为可注入的工厂接口，测试时即可直接注入带监控能力的Mock Buffer实例：

// 1. 抽象Buffer工厂接口
public interface BufferFactory {
    Buffer allocate(int capacity);
}

// 2. 业务类通过构造器注入工厂，不再直接调用静态分配方法
public class BizService {
    private final BufferFactory bufferFactory;
    public BizService(BufferFactory bufferFactory) {
        this.bufferFactory = bufferFactory;
    }

    public void doRead() {
        Buffer bytebuf = bufferFactory.allocate(1024);
        try {
            read(bytebuf);
        } catch (SomeException e) {
            // 异常处理逻辑
        } finally {
            bytebuf.release();
        }
    }
}

单元测试时实现带状态标记的Mock Buffer：

public class MockBuffer implements Buffer {
    public boolean isReleased = false;
    // 省略Buffer接口其他方法的Mock实现
    @Override
    public void release() {
        isReleased = true;
    }
}

测试用例覆盖所有执行分支，最后断言所有分配的Mock Buffer的isReleased为true即可：

• 正常执行无异常的场景
• 抛出SomeException进入catch块的场景
• 抛出未捕获异常的场景（如果存在）
  如果有人删除了finally块的release逻辑，断言会直接失败触发告警。

方案2：静态方法打桩（适配无法改造的旧项目） #

如果是历史旧项目没法做依赖注入改造，可以用PowerMock、Mockito这类测试框架给SomeUtility.allocateBuffer静态方法打桩，直接返回带监控的Mock Buffer，原理和方案1一致，不需要修改业务代码就能拦截所有内部分配的Buffer实例，最后校验释放状态即可。

方案3：全局资源计数器校验 #

如果用的是自带泄漏检测能力的Buffer实现（比如Netty的ByteBuf），可以在单元测试运行时开启最高级别的泄漏检测：

JVM启动参数添加：-Dio.netty.leakDetection.level=PARANOID

测试执行后扫描日志是否存在泄漏告警，直接将泄漏检测结果关联到测试用例的成功/失败状态。
如果是自定义Buffer实现，可以在底层分配、释放逻辑中添加全局的活跃Buffer计数器，单元测试执行前后对比计数器差值，差值不为0即说明存在泄漏，直接让测试失败。

所有方案的核心逻辑是：只要资源的分配入口是可拦截的，就能在测试阶段拿到方法内分配的所有资源实例，从流程上避免漏写finally块导致的泄漏问题。

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