11/*
2- * Copyright 2022 the original author or authors.
2+ * Copyright 2022-2023 the original author or authors.
33 *
44 * Licensed under the Apache License, Version 2.0 (the "License");
55 * you may not use this file except in compliance with the License.
@@ -103,12 +103,14 @@ void endToEnd(@Autowired Listener listener, @Autowired KafkaTemplate<Integer, St
103103 SimpleSpan span = spans .poll ();
104104 assertThat (span .getTags ()).containsEntry ("spring.kafka.template.name" , "template" );
105105 assertThat (span .getName ()).isEqualTo ("observation.testT1 send" );
106+ assertThat (span .getRemoteServiceName ()).startsWith ("Apache Kafka: " );
106107 await ().until (() -> spans .peekFirst ().getTags ().size () == 3 );
107108 span = spans .poll ();
108109 assertThat (span .getTags ())
109110 .containsAllEntriesOf (
110111 Map .of ("spring.kafka.listener.id" , "obs1-0" , "foo" , "some foo value" , "bar" , "some bar value" ));
111112 assertThat (span .getName ()).isEqualTo ("observation.testT1 receive" );
113+ assertThat (span .getRemoteServiceName ()).startsWith ("Apache Kafka: " );
112114 await ().until (() -> spans .peekFirst ().getTags ().size () == 1 );
113115 span = spans .poll ();
114116 assertThat (span .getTags ()).containsEntry ("spring.kafka.template.name" , "template" );
@@ -179,24 +181,27 @@ public KeyValues getLowCardinalityKeyValues(KafkaRecordReceiverContext context)
179181 assertThat (admin .getConfigurationProperties ())
180182 .containsEntry (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG , broker .getBrokersAsString ());
181183 // producer factory broker different to admin
182- assertThat (
183- KafkaTestUtils .getPropertyValue (template , "kafkaAdmin" , KafkaAdmin .class ).getConfigurationProperties ())
184- .containsEntry (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG ,
185- broker .getBrokersAsString () + "," + broker .getBrokersAsString ());
184+ KafkaAdmin pAdmin = KafkaTestUtils .getPropertyValue (template , "kafkaAdmin" , KafkaAdmin .class );
185+ assertThat (pAdmin .getOperationTimeout ()).isEqualTo (admin .getOperationTimeout ());
186+ assertThat (pAdmin .getConfigurationProperties ())
187+ .containsEntry (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG ,
188+ broker .getBrokersAsString () + "," + broker .getBrokersAsString ());
186189 Object container = KafkaTestUtils
187190 .getPropertyValue (endpointRegistry .getListenerContainer ("obs1" ), "containers" , List .class ).get (0 );
188191 // consumer factory broker different to admin
189- assertThat (KafkaTestUtils .getPropertyValue (container , "listenerConsumer.kafkaAdmin" , KafkaAdmin .class )
190- .getConfigurationProperties ())
191- .containsEntry (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG ,
192- broker .getBrokersAsString () + "," + broker .getBrokersAsString () + ","
193- + broker .getBrokersAsString ());
192+ KafkaAdmin cAdmin = KafkaTestUtils .getPropertyValue (container , "listenerConsumer.kafkaAdmin" , KafkaAdmin .class );
193+ assertThat (cAdmin .getOperationTimeout ()).isEqualTo (admin .getOperationTimeout ());
194+ assertThat (cAdmin .getConfigurationProperties ())
195+ .containsEntry (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG ,
196+ broker .getBrokersAsString () + "," + broker .getBrokersAsString () + ","
197+ + broker .getBrokersAsString ());
194198 // broker override in annotation
195199 container = KafkaTestUtils
196200 .getPropertyValue (endpointRegistry .getListenerContainer ("obs2" ), "containers" , List .class ).get (0 );
197- assertThat (KafkaTestUtils .getPropertyValue (container , "listenerConsumer.kafkaAdmin" , KafkaAdmin .class )
198- .getConfigurationProperties ())
199- .containsEntry (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG , broker .getBrokersAsString ());
201+ cAdmin = KafkaTestUtils .getPropertyValue (container , "listenerConsumer.kafkaAdmin" , KafkaAdmin .class );
202+ assertThat (cAdmin .getOperationTimeout ()).isEqualTo (admin .getOperationTimeout ());
203+ assertThat (cAdmin .getConfigurationProperties ())
204+ .containsEntry (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG , broker .getBrokersAsString ());
200205 }
201206
202207 @ Configuration
@@ -205,7 +210,10 @@ public static class Config {
205210
206211 @ Bean
207212 KafkaAdmin admin (EmbeddedKafkaBroker broker ) {
208- return new KafkaAdmin (Map .of (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG , broker .getBrokersAsString ()));
213+ KafkaAdmin admin = new KafkaAdmin (
214+ Map .of (AdminClientConfig .BOOTSTRAP_SERVERS_CONFIG , broker .getBrokersAsString ()));
215+ admin .setOperationTimeout (42 );
216+ return admin ;
209217 }
210218
211219 @ Bean
0 commit comments