Mockito模拟retrytemplate.execute并返回模拟响应

时间:2020-09-21 06:27:08

标签: java spring-boot unit-testing mockito

下面是我要测试的代码:

@Autowired
RetryTemplate retryTemplate;

MyResponse response;

response = retryTemplate.execute(new RetryCallback<Mono<MyResponse>, RuntimeException>() {
    public Mono<MyResponse> doRetry(RetryContext context) {
        return webclient.post.body(Mono.just(requestBodyObj), RequestBodyObj.class).retrieve().bodyToMono(MyResponse.class);
    }
});

下面是我正在尝试的测试用例,但是得到了无效使用Matchers异常。另外,我不确定下面的代码是否可以工作。请指导:

MyResponse myResponseObj = new MyResponse();

when(retryTemplate.execute(any(RetryCallback.class))).thenAnswer(invocation -> {
        RetryCallback retry = invocation.getArgument(0);
        when(retry.doWithRetry(Mockito.any())).thenReturn(Mockito.eq(Mono.just(myResponseObj)))
        return retry.doWithRetry(null);
    });

请指导这里的错误是什么,每当调用myResponseObj时如何返回retry.doWithRetry

1 个答案:

答案 0 :(得分:0)

请查看下面的 retryTemplate 执行方法:

public final <T, E extends Throwable> T execute(RetryCallback<T, E> retryCallback) throws E {
        return this.doExecute(retryCallback, (RecoveryCallback)null, (RetryState)null);
}

当我们调用retryTemplate.execute(retryCallback)时,doExecute方法中的另外两个参数为null,所以基本上这就是你得到Invalid use of matchers异常的原因。