java - 使用 Mockito 模拟另一个类中的类方法

Question

我正在尝试使用 Mockito/JUnit 为这样的函数编写单元测试:

class1 {
 method {
  object1 = class2.method // method that I want to fake the return value
  // some code that I still want to run
 }
}

在 Mockito 中有什么方法可以 stub class2.method 的结果吗？我正在尝试提高 class1 的代码覆盖率，因此我需要调用其真正的生产方法。

我查看了 Mockito API 的 spy 方法，但这会覆盖整个方法，而不是我想要的部分。

Answer 1

我想我理解你的问题。让我换句话说，你有一个你正在尝试测试的函数，并且想要模拟在该函数内调用的函数的结果，但在不同的类中。我已通过以下方式处理了该问题。

public MyUnitTest {
    private static final MyClass2 class2 = mock(MyClass2.class);

    @Begin
    public void setupTests() {
        when(class2.get(1000)).thenReturn(new User(1000, "John"));
        when(class2.validateObject(anyObj()).thenReturn(true);
    }

    @Test
    public void testFunctionCall() {
        String out = myClass.functionCall();
        assertThat(out).isEqualTo("Output");
    }
}

这样做是在用@Before 注释包装的函数内，我正在设置我希望 class2 中的函数如何响应给定的特定输入。然后，在实际测试中，我只是在要测试的类中调用我要测试的函数。在这种情况下，myClass.functionCall() 正常运行并且您没有覆盖它的任何方法，但您只是模拟它从 MyClass2 中的方法(或方法)获得的输出。