使用Mockito测试Mono和Flux

Question

使用Mockito测试Mono和Flux

javaunit-testingmockingmockitospring-webflux

8

我目前正在为我的REST端点编写一些基本的单元测试。我使用Mockito进行测试。下面是一个示例：

@MockBean
private MyService service;

@Test
public void getItems() {
    Flux<Item> result = Flux.create(sink -> {
        sink.next(new Item("1"));
        sink.next(new Item("2"));
        sink.complete();
    });

    Mono<ItemParams> params = Mono.just(new ItemParams("1"));

    Mockito.when(this.service.getItems(params)).thenReturn(result);

    this.webClient.post().uri("/items")
            .accept(MediaType.APPLICATION_STREAM_JSON)
            .contentType(MediaType.APPLICATION_STREAM_JSON)
            .body(BodyInserters.fromPublisher(params, ItemParams.class))
            .exchange()
            .expectStatus().isOk()
            .expectBodyList(Item.class).isEqualTo(Objects.requireNonNull(result.collectList().block()));
}

这个实现会导致以下错误：

java.lang.AssertionError: Response body expected:<[Item(name=1), Item(name=2)]> but was:<[]>

> POST /items
> WebTestClient-Request-Id: [1]
> Accept: [application/stream+json]
> Content-Type: [application/stream+json]

Content not available yet

< 200 OK
< Content-Type: [application/stream+json;charset=UTF-8]

No content

当我将Mockito语句中的参数替换为Mockito.any()时

Mockito.when(this.service.getItems(Mockito.any())).thenReturn(result);

测试成功运行。这意味着由于我在Mockito语句中放入的“params”与我放入BodyInserters.fromPublisher（params，ItemParams.class）中的“params”对象不相等，因此出现了这种情况。那么我应该如何测试我的功能呢？编辑 REST端点

@PostMapping(path = "/items", consumes = MediaType.APPLICATION_STREAM_JSON_VALUE, produces = MediaType.APPLICATION_STREAM_JSON_VALUE)
public Flux<Item> getItems(@Valid @RequestBody Mono<ItemParams> itemParms) {
    return this.service.getItems(itemParms);
}

- Mulgard

断言在哪里？你能补充一下你是如何模拟服务的吗？ - Maciej Kowalski

我添加了 MockBean，并且断言是 .expectBodyList(Item.class).isEqualTo(Objects.requireNonNull(result.collectList().block()));。 - Mulgard

尝试展示端点实现...至少是重要部分。 - Maciej Kowalski

我已添加了 REST 端点。 - Mulgard

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Maciej Kowalski · Accepted Answer

实际对象@RequestBody Mono<ItemParams> itemParms会不会和你在测试中创建和传递的对象不同？

你可以利用thenAnswer来验证实际传递给服务的对象内容：

Mockito.when(this.service.getItems(Mockito.any()))
       .thenAnswer(new Answer<Flux<Item>>() {

    @Override
    public Flux<Item> answer(InvocationOnMock invocation) throws Throwable {
        Mono<ItemParams> mono = (Mono<ItemParams>)invocation.getArgument(0);

        if(/* verify that paseed mono contains new ItemParams("1")*/){
          return result;
        }

        return null;
    }
});