@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); }