如何在JUnit中Mock ClassB.methodB的SOAP调用以测试ClassA.methodA

嘿，我完全懂你现在卡在哪了——你在测试ClassA.methodA()的时候，希望被Mock的ClassB.methodB()每次被调用时能动态返回不同结果，而不是提前把返回值固定死，对吧？这种场景Mockito其实有现成的解决方案，我给你拆解几种实用的方式：

1. 用Answer接口实时生成返回值 #

Mockito的when(...).thenAnswer()方法就是专门干这个的——它允许你在Mock方法被触发的那一刻，根据调用参数、当前上下文甚至外部状态来动态生成返回值，完美匹配你需要模拟SOAP服务动态响应的场景：

// 先Mock出ClassB实例
ClassB mockClassB = Mockito.mock(ClassB.class);

// 给methodB设置动态响应逻辑
Mockito.when(mockClassB.methodB(Mockito.anyString())).thenAnswer(invocation -> {
    // 获取methodB被调用时传入的参数
    String requestParam = invocation.getArgument(0);
    
    // 这里可以根据实际需求写逻辑，比如模拟SOAP服务的不同返回
    if ("criticalRequest".equals(requestParam)) {
        return "SOAP服务返回的紧急响应";
    } else {
        return "SOAP服务返回的常规响应";
    }
});

// 把Mock的ClassB注入到ClassA里（假设ClassA用构造函数注入）
ClassA classAUnderTest = new ClassA(mockClassB);

// 调用methodA，此时methodB会根据参数动态返回对应结果
classAUnderTest.methodA();

2. 用doAnswer()处理void方法或特殊触发逻辑 #

如果methodB是void方法，但你需要在它被调用时执行一些更新操作（比如修改某个全局状态，间接影响后续流程），就可以用doAnswer()：

Mockito.doAnswer(invocation -> {
    // 获取methodB的调用参数
    Object[] callArgs = invocation.getArguments();
    // 在这里执行你需要的更新逻辑，比如记录调用次数、修改Mock内部状态
    System.out.println("methodB被调用了，参数是：" + Arrays.toString(callArgs));
    // void方法返回null即可
    return null;
}).when(mockClassB).methodB(Mockito.any());

3. 多次调用时返回不同序列值 #

如果methodB会被methodA多次调用，你需要每次返回不同的值，可以用两种方式：

方式一：链式thenReturn() #

// 第一次调用返回第一个值，第二次返回第二个，之后都返回第三个
Mockito.when(mockClassB.methodB()).thenReturn("第一次SOAP响应")
                                  .thenReturn("第二次SOAP响应")
                                  .thenReturn("默认SOAP响应");

方式二：结合计数器的thenAnswer() #

如果需要更灵活的序列逻辑，比如根据调用次数计算返回值：

AtomicInteger callCounter = new AtomicInteger(0);
Mockito.when(mockClassB.methodB()).thenAnswer(invocation -> {
    int callTimes = callCounter.getAndIncrement();
    switch (callTimes) {
        case 0: return "首次调用的响应";
        case 1: return "第二次调用的响应";
        default: return "后续所有调用的默认响应";
    }
});

几个关键提醒 #

• 一定要确保ClassA使用的是你Mock出来的ClassB实例！比如通过构造函数注入、Setter注入，不然ClassA会调用真实的ClassB，你的Mock逻辑根本不会生效。
• 如果测试过程中需要临时修改Mock行为，也可以中途重新设置when()规则，但尽量少用Mockito.reset(mockClassB)重置Mock，容易导致测试逻辑混乱，优先用动态Answer来处理。

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