Allows providing a custom KafkaTemplate bean; fixes gh-1704

spring-cloud-contract-verifier/src/main/java/org/springframework/cloud/contract/verifier/messaging/kafka/ContractVerifierKafkaConfiguration.java

@@ -19,6 +19,7 @@ package org.springframework.cloud.contract.verifier.messaging.kafka;
 import java.nio.charset.StandardCharsets;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.function.Supplier;
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -55,9 +56,15 @@ public class ContractVerifierKafkaConfiguration {
 
 	@Bean
 	@ConditionalOnMissingBean
-	MessageVerifier<Message<?>> contractVerifierKafkaMessageExchange(KafkaTemplate kafkaTemplate,
+	MessageVerifier<Message<?>> contractVerifierKafkaMessageExchange(Supplier<KafkaTemplate> kafkaTemplate,
 			EmbeddedKafkaBroker broker, KafkaProperties kafkaProperties, KafkaStubMessagesInitializer initializer) {
-		return new KafkaStubMessages(kafkaTemplate, broker, kafkaProperties, initializer);
+		return new KafkaStubMessages(kafkaTemplate.get(), broker, kafkaProperties, initializer);
+	}
+
+	@Bean
+	@ConditionalOnMissingBean
+	Supplier<KafkaTemplate> contractVerifierKafkaTemplateSupplier(KafkaTemplate kafkaTemplate) {
+		return () -> kafkaTemplate;
 	}
 
 	@Bean

spring-cloud-contract-verifier/src/main/java/org/springframework/cloud/contract/verifier/messaging/kafka/KafkaStubMessages.java

@@ -46,7 +46,7 @@ class KafkaStubMessages implements MessageVerifier<Message<?>> {
 
 	private static final Log log = LogFactory.getLog(KafkaStubMessages.class);
 
-	private final KafkaTemplate kafkaTemplate;
+	final KafkaTemplate kafkaTemplate;
 
 	private final Receiver receiver;
 

spring-cloud-contract-verifier/src/test/java/org/springframework/cloud/contract/verifier/messaging/kafka/ContractVerifierKafkaConfigurationTests.java

@@ -16,12 +16,16 @@
 
 package org.springframework.cloud.contract.verifier.messaging.kafka;
 
+import java.util.function.Supplier;
+
 import org.junit.Test;
+import org.mockito.BDDMockito;
 
 import org.springframework.boot.autoconfigure.AutoConfigurations;
 import org.springframework.boot.autoconfigure.kafka.KafkaAutoConfiguration;
 import org.springframework.boot.test.context.runner.ApplicationContextRunner;
 import org.springframework.context.annotation.Bean;
+import org.springframework.kafka.core.KafkaTemplate;
 import org.springframework.kafka.test.EmbeddedKafkaBroker;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -44,6 +48,14 @@ public class ContractVerifierKafkaConfigurationTests {
 		});
 	}
 
+	@Test
+	public void shouldPickCustomKafkaTemplate() {
+		this.contextRunner.withUserConfiguration(CustomKafkaTemplateConfiguration.class).run((context) -> {
+			assertThat(context.getBeansOfType(KafkaStubMessages.class)).hasSize(1);
+			assertThat(context.getBean(KafkaStubMessages.class).kafkaTemplate).isSameAs(context.getBean(CustomKafkaTemplateConfiguration.class).myKafkaTemplate);
+		});
+	}
+
 	@Test
 	public void shouldNotCreateBeansWhenDisabled() {
 		this.contextRunner.withPropertyValues("stubrunner.kafka.enabled=false").run((context) -> {
@@ -69,4 +81,15 @@ public class ContractVerifierKafkaConfigurationTests {
 
 	}
 
+	static class CustomKafkaTemplateConfiguration {
+
+		KafkaTemplate myKafkaTemplate = BDDMockito.mock(KafkaTemplate.class);
+
+		@Bean
+		Supplier<KafkaTemplate> myKafkaTemplate() {
+			return () -> myKafkaTemplate;
+		}
+
+	}
+
 }