@Test
@SuppressWarnings({"unchecked", "rawtypes"})
public void testProducerListenerSet() throws Exception {
KafkaProducerContext producerContext = new KafkaProducerContext();
Serializer serializer = Mockito.mock(Serializer.class);
ProducerMetadata<Object, Object> producerMetadata =
new ProducerMetadata<>("default", Object.class, Object.class, serializer, serializer);
Producer producer = Mockito.mock(Producer.class);
ProducerConfiguration<Object, Object> producerConfiguration =
new ProducerConfiguration<>(producerMetadata, producer);
ProducerListener producerListener = mock(ProducerListener.class);
producerConfiguration.setProducerListener(producerListener);
Map<String, ProducerConfiguration<?, ?>> producerConfigurations =
Collections.<String, ProducerConfiguration<?, ?>>singletonMap(
"default", producerConfiguration);
producerContext.setProducerConfigurations(producerConfigurations);
KafkaProducerMessageHandler handler = new KafkaProducerMessageHandler(producerContext);
handler.handleMessage(
MessageBuilder.withPayload("somePayload")
.setHeader(KafkaHeaders.PARTITION_ID, 2)
.setHeader(KafkaHeaders.MESSAGE_KEY, "someKey")
.build());
final ArgumentCaptor<Callback> argument = ArgumentCaptor.forClass(Callback.class);
verify(producer).send(any(ProducerRecord.class), argument.capture());
Callback callback = argument.getValue();
assertThat(callback, CoreMatchers.instanceOf(ProducerListenerInvokingCallback.class));
DirectFieldAccessor fieldAccessor = new DirectFieldAccessor(callback);
assertEquals(fieldAccessor.getPropertyValue("topic"), "default");
assertEquals(fieldAccessor.getPropertyValue("partition"), 2);
assertEquals(fieldAccessor.getPropertyValue("key"), "someKey");
assertEquals(fieldAccessor.getPropertyValue("payload"), "somePayload");
assertSame(fieldAccessor.getPropertyValue("producerListener"), producerListener);
verifyNoMoreInteractions(producer);
}
