我有以下服务方法:
public CommandDTO update(UUID uuid, EmployeeRequest request) {
final List<Product> productList = productRepository
.findAllByUuidOrderByOrderNumber(uuid);
final List<UUID> uuidList = request.getUuidList();
for (int i = 0; i < uuidList.size(); i++) {
Product product = productList.get(i);
product.setOrderNumber(i + 1);
// how can I get the product value in this line?
}
return CommandDTO.builder().uuid(uuid).build();
}
通常,我使用 ArgumentCaptor
来让 values 传递到存储库或服务。然而,在这个例子中,我想在 product.setOrderNumber(i + 1);
之后获得的产品 value 没有通过服务或存储库。出于这个原因,我无法得到它。我不确定是否可以使用 @Spy
或 doAnswer
,但据我所知,这些方法不能使用,因为 value 没有传递给服务或 repo。任何想法?
回答1
您可以模拟您的存储库以返回模拟列表:
public test() {
Product product1 = mock(Product.class);
Product product2 = mock(Product.class);
List list = Arraylist();
list.add(product1);
list.add(product2);
when(productRepositoryMock.findAllByUuidOrderByOrderNumber(uuid))
.thenReturn(list)
when(product1).setOrderNumber(captor.capture)
在这种情况下,您可以在这里使用 ArgumentCaptor,但我相信在这种情况下它是多余的,简单的验证调用就足够了
service.update(.....);
verify.(product1).setOrderNumber(value);
回答2
product
变量来自 productList
,而 productList
又来自 repository.findAllByUuidOrderByOrderNumber()
方法。因此,如果您模拟您的存储库,您可以编写如下内容:
Product givenProduct = new Product();
UUID givenUUID = UUID.randomUUID();
// Create an EmployeeRequest with a `uuidList`
EmployeeRequest givenRequest = new EmployeeRequest(/* TODO */);
// Use Mockito to return a `productList` you choose
// This uses a static import of `Mockito.when()`
when(productRepository.findAllByUuidOrderByOrderNumber(givenUUID)).thenReturn(List.of(givenProduct));
service.updateRequest(givenUUID, givenRequest);
从理论上讲,您甚至可以在上面的测试逻辑中返回一个模拟产品,但我建议使用真正的 Product
对象。
现在,无论该产品发生什么,都可以在您的测试中断言。
例如:
// `givenProduct` is element 0, and thus will get an orderNumber of 1
assertThat(givenProduct.getOrderNumber()).isEqualTo(1);