|  | 
| 33 | 33 | import java.util.Arrays; | 
| 34 | 34 | import java.util.BitSet; | 
| 35 | 35 | import java.util.Collection; | 
|  | 36 | +import java.util.Collections; | 
| 36 | 37 | import java.util.HashMap; | 
| 37 | 38 | import java.util.List; | 
| 38 | 39 | import java.util.Map; | 
| @@ -1612,7 +1613,7 @@ public void onPartitionsAssigned(Collection<TopicPartition> partitions) { | 
| 1612 | 1613 | 	public void testAckModeCount() throws Exception { | 
| 1613 | 1614 | 		ConsumerFactory<Integer, String> cf = mock(ConsumerFactory.class); | 
| 1614 | 1615 | 		Consumer<Integer, String> consumer = mock(Consumer.class); | 
| 1615 |  | -		given(cf.createConsumer(isNull(), eq("clientId"), isNull())).willReturn(consumer); | 
|  | 1616 | +		given(cf.createConsumer(isNull(), eq("clientId"))).willReturn(consumer); | 
| 1616 | 1617 | 		TopicPartition topicPartition = new TopicPartition("foo", 0); | 
| 1617 | 1618 | 		final Map<TopicPartition, List<ConsumerRecord<Integer, String>>> records1 = new HashMap<>(); | 
| 1618 | 1619 | 		records1.put(topicPartition, Arrays.asList( | 
| @@ -1657,13 +1658,13 @@ public void testAckModeCount() throws Exception { | 
| 1657 | 1658 | 		ContainerProperties containerProps = new ContainerProperties(topicPartitionOffset); | 
| 1658 | 1659 | 		containerProps.setAckMode(AckMode.COUNT); | 
| 1659 | 1660 | 		containerProps.setAckCount(3); | 
| 1660 |  | -		containerProps.setClientId("clientId"); | 
| 1661 | 1661 | 		AtomicInteger recordCount = new AtomicInteger(); | 
| 1662 | 1662 | 		containerProps.setMessageListener((MessageListener) r -> { | 
| 1663 | 1663 | 			recordCount.incrementAndGet(); | 
| 1664 | 1664 | 		}); | 
| 1665 | 1665 | 		KafkaMessageListenerContainer<Integer, String> container = | 
| 1666 | 1666 | 				new KafkaMessageListenerContainer<>(cf, containerProps); | 
|  | 1667 | +		container.setClientIdSuffix("clientId"); | 
| 1667 | 1668 | 		container.start(); | 
| 1668 | 1669 | 		assertThat(commitLatch.await(10, TimeUnit.SECONDS)).isTrue(); | 
| 1669 | 1670 | 		assertThat(recordCount.get()).isEqualTo(7); | 
|  | 
0 commit comments