Я пытаюсь написать модульный тест для ListenableFuture, добавляя обратный вызов, но я не уверен, как это сделать. Ничего полезного в интернете не нашел.
@Test
public void can_publish_data_to_kafka() {
String topic = someString(10);
String key = someAlphanumericString(5);
String data = someString(50);
SendResult sendResult = mock(SendResult.class);
ListenableFuture<SendResult<String, Object>> future = mock(ListenableFuture.class);
given(kafkaTemplate.send(topic, key, data)).willReturn(future);
doAnswer(new Answer() {
@Override
public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
return invocationOnMock.getArguments()[1];
}
});
service.method(key, topic, data);
}
Код, для которого я хочу написать тестовый пример
ListenableFuture<SendResult<String, Object>> future = kafkaTemplate.send(topicName, key, data);
future.addCallback(new ListenableFutureCallback<SendResult<String, Object>>() {
@Override
public void onSuccess(SendResult<String, Object> stringKafkaBeanSendResult) {
RecordMetadata recordMetadata = stringKafkaBeanSendResult.getRecordMetadata();
LOGGER.info(String.format("sent message %s to topic %s partition %s with offset %s" + data.toString(), recordMetadata.topic(), recordMetadata.partition(), recordMetadata.offset()));
}
@Override
public void onFailure(Throwable throwable) {
LOGGER.error(String.format("unable to send message = %s to topic %s because of error %s" + data.toString(), topicName, throwable.getMessage()));
}
});
Я ожидаю направления, в котором я должен двигаться для написания UT с использованием mockito.
Вы можете написать тестовый пример следующим образом.
@Test
public void can_publishDataToKafka() {
String key = someAlphanumericString();
String topic = someAlphaString(10);
long offset = somePositiveLong();
int partition = somePositiveInteger();
SiebelRecord siebelRecord = mock(SiebelRecord.class);
SendResult<String, Object> sendResult = mock(SendResult.class);
ListenableFuture<SendResult<String, Object>> responseFuture = mock(ListenableFuture.class);
RecordMetadata recordMetadata = new RecordMetadata(new TopicPartition(topic, partition), offset, 0L, 0L, 0L, 0, 0);
given(sendResult.getRecordMetadata()).willReturn(recordMetadata);
when(kafkaTemplate.send(topic, key, siebelRecord)).thenReturn(responseFuture);
doAnswer(invocationOnMock -> {
ListenableFutureCallback listenableFutureCallback = invocationOnMock.getArgument(0);
listenableFutureCallback.onSuccess(sendResult);
assertEquals(sendResult.getRecordMetadata().offset(), offset);
assertEquals(sendResult.getRecordMetadata().partition(), partition);
return null;
}).when(responseFuture).addCallback(any(ListenableFutureCallback.class));
service.publishDataToKafka(key, topic, siebelRecord);
verify(kafkaTemplate, times(1)).send(topic, key, siebelRecord);
}
@Test(expected = KafkaException.class)
public void can_capture_failure_publishDataToKafka() {
String key = someAlphanumericString();
String topic = someAlphaString(10);
String message = someString(20);
SiebelRecord siebelRecord = mock(SiebelRecord.class);
ListenableFuture<SendResult<String, Object>> responseFuture = mock(ListenableFuture.class);
Throwable throwable = mock(Throwable.class);
given(throwable.getMessage()).willReturn(message);
when(kafkaTemplate.send(topic, key, siebelRecord)).thenReturn(responseFuture);
doAnswer(invocationOnMock -> {
ListenableFutureCallback listenableFutureCallback = invocationOnMock.getArgument(0);
listenableFutureCallback.onFailure(throwable);
return null;
}).when(responseFuture).addCallback(any(ListenableFutureCallback.class));
service.publishDataToKafka(key, topic, siebelRecord);
}
И я не хочу использовать PowerMockito.
@AdityaSharma Проверьте, полезно ли это ListeningScheduledExecutorService executor = MoreExecutors.listeningDecorator(Executors.newSingleThreadScheduledExecutor()); ListenableFuture<T> futureResponse = executor.schedule(call, n, unit);
Я не уверен, почему, но блок doAnswer не действует и не выполняется.
can_capture_failure по какой-то причине не вызывает исключение KafkaException. Я вижу, что он активирует обратный вызов onFailure, но тест завершается с ошибкой с ожидаемым исключением:...
Как я могу проверить это, если я использую Futures.addCallback.