使用Mockito模拟另一个类中的类方法 [英] Using Mockito to mock a class method inside another class

问题描述

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

I'm trying to write unit tests with Mockito / JUnit for a function like this:

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

在Mockito中有什么方法来存根class2.method的结果？我正在尝试改进class1的代码覆盖率，所以我需要调用它真正的生产方法。

Is there any way in Mockito to stub the result of class2.method? I'm trying to improve code coverage for class1 so I need to call its real production methods.

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

I looked into the Mockito API at its spy method but that would overwrite the whole method and not the part that I want.

推荐答案

我想我理解你的问题。让我重新说一下，你有一个函数，你试图测试并想要模拟在该函数内调用的函数的结果，但是在另一个类中。我已按以下方式处理。

I think I am understanding your question. Let me re-phrase, you have a function that you are trying to test and want to mock the results of a function called within that function, but in a different class. I have handled that in the following way.

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中的方法（或方法）得到的输出。

What this is doing is that within the function wrapped with the @Before annotation, I am setting up how I want the functions in class2 to respond given specific inputs. Then, from within the actual test, I am just calling the function that I am trying to test in the class I want to test. In this case, the myClass.functionCall() is running through as normal and you are not overwriting any of its methods, but you are just mocking the outputs that it gets from the methods (or method) within MyClass2.

这篇关于使用Mockito模拟另一个类中的类方法的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！