有人可以幫助為下面的代碼部分（尤其是添加回調部分）編寫 JUnit 測試嗎？我不知道如何使用回調為 Listenablefuture 編寫單元測試用例 private void handleResponse(final ListenableFuture<UserRecordResult> response, CompletableFuture future) { Futures.addCallback(response, new FutureCallback<UserRecordResult>() { @Override public void onFailure(@Nonnull Throwable throwable) { future.completeExceptionally(new Exception("Fail to put record" + throwable.getMessage())); } @Override public void onSuccess(UserRecordResult result) { if(result.isSuccessful()) { future.complete(true); } else { future.completeExceptionally(new Exception("Fail to put record")); } } }); }
1 回答
楊__羊羊
TA貢獻1943條經驗 獲得超7個贊
您可以通過存根ListeneableFuture并按CompletableFuture以下方式進行測試：
@Test
public void completeExceptionallyOnCallbackFailure() {
ListenableFuture<UserRecordResult> failureResponse = Futures.immediateFailedFuture(new RuntimeException());
CompletableFuture future = new CompletableFuture();
handleResponse(failureResponse, future);
assertThat(future.isCompletedExceptionally(), equalTo(true));
}
@Test
public void completeExceptionallyIfUserRecordResultIsNotSuccessful() {
UserRecordResult mockResult = Mockito.mock(UserRecordResult.class);
Mockito.when(mockResult.isSuccessful()).thenReturn(false);
ListenableFuture<UserRecordResult> failureResponse = Futures.immediateFuture(mockResult);
CompletableFuture future = new CompletableFuture();
handleResponse(failureResponse, future);
assertThat(future.isCompletedExceptionally(), equalTo(true));
}
@Test
public void completeSuccessfully() throws ExecutionException, InterruptedException {
UserRecordResult mockResult = Mockito.mock(UserRecordResult.class);
Mockito.when(mockResult.isSuccessful()).thenReturn(true);
ListenableFuture<UserRecordResult> successResponse = Futures.immediateFuture(mockResult);
CompletableFuture future = new CompletableFuture();
handleResponse(successResponse, future);
assertThat(future.get(), equalTo(true));
}
此外，您可以使用以下方法斷言實際的錯誤消息assertj：
@Test
public void verifyMessageIfCompletedExceptionallyAfterUserRecordResultIsNotSuccessful() {
UserRecordResult mockResult = Mockito.mock(UserRecordResult.class);
Mockito.when(mockResult.isSuccessful()).thenReturn(false);
ListenableFuture<UserRecordResult> failureResponse = Futures.immediateFuture(mockResult);
CompletableFuture future = new CompletableFuture();
Assertions.assertThatThrownBy(() -> {
handleResponse(failureResponse, future);
future.get();
})
.isInstanceOf(Exception.class)
.hasMessage("java.lang.Exception: Fail to put record");
}
添加回答
舉報
0/150
提交
取消