如何用Mockito和JUnit测试公共方法中调用的私有方法

核心原则先说明 #

单元测试的核心是验证公共方法的行为输出/副作用，而非内部私有方法的实现细节。但如果确实需要验证私有方法的调用或逻辑，反射或PowerMock是可行方案，以下是具体实现思路：

方案1：用Java反射直接调用私有方法测试逻辑 #

如果要单独验证imprimeDocumento的业务逻辑，可通过反射绕过访问限制：

1. 获取私有方法的Method对象
   需严格匹配私有方法的参数类型（注意你的代码里私有方法签名有误，实际调用传了6个参数：Long、String、int、int、int、String）：

   // 获取目标类的Class实例
   Class<?> targetClass = TuClase.class;
   // 获取指定名称和参数类型的私有方法
   Method privateMethod = targetClass.getDeclaredMethod(
       "imprimeDocumento",
       Long.class, String.class, int.class, int.class, int.class, String.class
   );
   // 取消访问检查，允许调用私有方法
   privateMethod.setAccessible(true);
   

2. 调用私有方法并验证
   创建类实例后，通过invoke方法传入参数调用，随后验证方法执行的副作用（比如文件生成、数据状态变更等）：

   TuClase instance = new TuClase();
   // 调用私有方法，参数对应方法定义
   privateMethod.invoke(instance, 0L, "", 12, 2024, 31, "0");
   
   // 这里添加断言，比如验证生成的文件是否存在、内容是否符合预期
   // 示例：Assert.assertTrue(new File("目标文件路径").exists());
   

方案2：用PowerMock验证公共方法对私有方法的调用 #

如果只需验证generarArchivoProd是否正确触发了imprimeDocumento（含参数正确性），可使用PowerMock扩展Mockito的能力：

1. 依赖准备
   确保项目中引入PowerMock和Mockito的兼容依赖（Java 8环境推荐搭配JUnit 4）。

2. 编写测试代码

   import org.junit.Test;
   import org.junit.runner.RunWith;
   import org.powermock.api.mockito.PowerMockito;
   import org.powermock.core.classloader.annotations.PrepareForTest;
   import org.powermock.modules.junit4.PowerMockRunner;
   
   @RunWith(PowerMockRunner.class)
   @PrepareForTest(TuClase.class) // 指定要增强的类
   public class TuClaseTest {
   
       @Test
       public void generarArchivoProd_DebeLlamarImprimeDocumento() throws Exception {
           // 创建被测类的Spy对象，保留真实方法逻辑
           TuClase spy = PowerMockito.spy(new TuClase());
           
           // 调用公共方法
           spy.generarArchivoProd(2024, 12, 31, "0");
           
           // 验证私有方法是否被调用，参数是否匹配
           PowerMockito.verifyPrivate(spy)
               .invoke("imprimeDocumento", 0L, "", 12, 2024, 31, "0");
       }
   }
   

注意事项 #

• 反射会破坏类的封装性，仅在必要时使用，优先通过公共方法的行为来覆盖私有方法的逻辑
• PowerMock会修改类加载机制，可能导致测试复杂度上升，升级依赖时需注意版本兼容性
• 使用反射时，参数类型必须与私有方法定义完全一致，否则会抛出NoSuchMethodException

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