如何用 mockito 模拟 builder

Question

我有一个建造者：

class Builder{
     private String name;
     private String address;
     public Builder setName(String name){
         this.name = name;
         return this;
    }
     public Builder setAddress(String address){
         this.address = address;
         return this;
    }

}

在 mockito 中模拟构建器会给我每个方法的 null。那么有没有一种简单的方法可以让构建器在每次函数调用时返回自身，而无需使用when().thenReturn 模拟每个函数本身。

Answer 1

使用 RETURN_DEEP_STUBS 的问题在于，每次调用方法时都会得到不同的模拟。我认为从您的问题来看，您想使用一个默认的答案，该答案实际上返回了调用它的模拟，对于每个具有正确返回类型的方法。这可能如下所示。请注意，我没有对此进行测试，因此它可能包含拼写错误，但我希望无论如何意图都清楚。

import static org.mockito.Mockito.RETURNS_DEFAULTS;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

public class SelfReturningAnswer implements Answer<Object> {

    public Object answer(InvocationOnMock invocation) throws Throwable {
        Object mock = invocation.getMock();
        if(invocation.getMethod().getReturnType().isInstance(mock)){
            return mock;
        }
        return RETURNS_DEFAULTS.answer(invocation);
    }
}

然后，当您创建模拟时，将其指定为您的默认答案。这将使您的模拟从它可以的每个方法中返回自己；但是当你调用一个返回类型对 mock 错误的方法时，它的行为就像一个普通的 mock。

像这样创建你的模拟

Builder mockBuilder = mock( Builder.class, new SelfReturningAnswer());

或为此类创建一个常量并编写类似

@Mock(answer = SELF_RETURNING) private Builder mockBuilder;

希望对您有所帮助。

Answer 2

从 Mockito 2.0（测试版）开始，RETURNS_SELF 有一个新的默认答案，其行为几乎与David Wallace's answer 相同。 Mockito 文档中的示例：

@Test
public void use_full_builder_with_terminating_method() {
    HttpBuilder builder = mock(HttpBuilder.class, RETURNS_SELF);
    HttpRequesterWithHeaders requester = new HttpRequesterWithHeaders(builder);
    String response = "StatusCode: 200";

    when(builder.request()).thenReturn(response);

    assertThat(requester.request("URI")).isEqualTo(response);
}

请注意，它同时出现在Mockito 类和Answers 枚举中，因此它也与@Mock(answer = RETURNS_SELF) 语法兼容。

Answer 3

您可以使用RETURN_DEEP_STUBS 模拟链接 API。

如果您知道调用构建器的确切顺序，以下是您将如何使用它的示例：

Builder b = Mockito.mock(Builder.class, RETURNS_DEEP_STUBS);
when(b.setName("a name").setAddress("an address")).thenReturn(b);
assert b.setName("a name").setAddress("an address") == b; // this passes

不幸的是，这不会为您提供一种模拟“所有各种构建器方法”的通用方式，因此它们总是返回这个，看看另一个答案是你需要那个。