Java的ListenableFuture和addListener回调的单元测试

Question

我有一个模块，它有一个方法，可以将数据写入动觉流。我们称之为反馈流

我想对我的分析服务的函数运行一个验证单元测试，这些函数在ListenableFuture的结果中被调用。我尝试过设置模拟，但遇到了ClassCast异常。我想我可能设置的模拟不正确

我想进行单元测试的方法：

public void writeData(KinesisRecord kinesisRecord, Response response) {

    FeedbackRecord feedbackRecord = new 
    FeedbackRecord(kinesisRecord.getVersion(), kinesisRecord.getNotificationId(), response.name(), 
kinesisRecord.getDestination());

    Gson g = new Gson();
    byte[] dataBytes = g.toJson(feedbackRecord).getBytes(StandardCharsets.UTF_8);
    ListenableFuture<UserRecordResult> future = kinesisProducer.addUserRecord(streamName, kinesisRecord.getNotificationId(), ByteBuffer.wrap(dataBytes));

    future.addListener(() -> {
    try {
        UserRecordResult userRecordResult = future.get();
        analyticsService.trackSuccess(AnalyticsConstants.FeedbackProducer.WRITE_SUCCESS, AnalyticsConstants.FeedbackProducer.WRITEDATA_LOCATION + " " + userRecordResult.getSequenceNumber());
        log.debug("Write to Kinesis Feedback successful, record sequence number: " + userRecordResult.getSequenceNumber());
    } catch (InterruptedException e) {
        e.printStackTrace();
        analyticsService.trackError(AnalyticsConstants.FeedbackProducer.WRITE_ERROR, AnalyticsConstants.FeedbackProducer.WRITEDATA_LOCATION);
        log.error("Feedback Kinesis Write failed: " + e);
    } catch (ExecutionException e) {
        e.printStackTrace();
        analyticsService.trackError(AnalyticsConstants.FeedbackProducer.WRITE_ERROR, AnalyticsConstants.FeedbackProducer.WRITEDATA_LOCATION);
        log.error("Feedback Kinesis Write failed: " + e);
    }
}, MoreExecutors.directExecutor());

} 单元测试尝试：

@Test
public void addUserRecordFutureTest() throws ExecutionException, InterruptedException {

    ListenableFuture<UserRecordResult> responseFuture = mock(ListenableFuture.class);

    Attempt attempt = new Attempt(1, 2, "AttemptError", "5xx", true);
    List<Attempt> attempts = new ArrayList<>();
    attempts.add(attempt);
    UserRecordResult mockUserRecordResult = mock(UserRecordResult.class);

    when(kinesisProducer.addUserRecord(any(), any(), any())).thenReturn(responseFuture);
    doAnswer(invocation -> {
        ListenableFutureCallback listenableFutureCallback = invocation.getArgument(0);
        listenableFutureCallback.onSuccess(mockUserRecordResult);
    // do asserts here
        return null;
    }).when(responseFuture).addListener(any(), any());

    feedbackProducer.writeData(mockRecord, FeedbackProducer.Response.invalid_channel);
}

堆栈跟踪：