如何用 mockito 模拟 builder

2022-01-14 00:00:00 unit-testing mockito java

我有一个建造者:

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 模拟每个函数本身.

Mocking the builder in mockito will gives me null for every method. So is there an easy way to get the builder return itself on every function call, without mocking every function itself using when().thenReturn.

推荐答案

你可以使用RETURN_DEEP_STUBS 模拟链接 API.

You can use RETURN_DEEP_STUBS to mock a chaining API.

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

If you know the exact order your builder will be called, here's an example of how you would use it:

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

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

Unfortunately this won't give you a generic way of mocking "all the various builder methods" so that they always return this, see the other answer is you need that.

相关文章