如何模拟类的属性调用？

Question

public class MyClass {
    protected OtherClass other;
    public MyClass(String arg) {
        this.other = new OtherClass(arg);
    }

    public void doSomething() {
        String response = other.sendRequest(); // mock this call
        // ...
    }
}

我想测试doSomething()，为此，我需要模拟调用sendrequest()。为此，我尝试了以下方法，但没有成功：

@Test
public void doSomethingTest() {
    MyClass myClass = new MyClass("test");
    when(myClass.other.sendRequest()).thenReturn("response");
    myClass.doSomething();
}

仍然在调用真正的sendrequest()而不是返回模拟响应。

如何使用Mockito模拟对sendrequest()的调用？

Answer 1

如果在MyClass类中实例化OtherClass对象，那么实际上不可能模拟它，您需要在MyClass之外实例化它，然后将它注入到MyClass中（例如通过其构造函数）。

然后在测试中所需要做的就是模拟对mockOtherClass对象的调用。并使用它实例化MyClass对象

将MyClasss更改为

public class MyClass {

    protected OtherClass other;
    
    public MyClass(OtherClass otherClass) {
        this.other = otherClass;
    }

    public void doSomething() {
        String response = other.sendRequest(); 
    }

}

您的代码将从

        MyClass myClass = new MyClass("s");

        OtherClass other = new OtherClass("s");
        MyClass myClass = new MyClass(other);

@Test
public void doSomethingTest() {
    OtherClass other = mock(OtherClass.class);
    when(other.sendRequest()).thenReturn("response");
    
    MyClass myClass = new MyClass(other);
    myClass.doSomething();

   // .. test assertions
}