Skip to content

Commit b9413ea

Browse files
authored
MINOR: Cleanup Tools Module (2/n) (#20096)
Now that Kafka support Java 17, this PR makes some changes in tools module. The changes in this PR are limited to only some files. A future PR(s) shall follow. The changes mostly include: - Collections.emptyList(), Collections.singletonList() and Arrays.asList() are replaced with List.of() - Collections.emptyMap() and Collections.singletonMap() are replaced with Map.of() - Collections.singleton() is replaced with Set.of() Some minor changes to use the enhanced switch. Sub modules targeted: tools/src/test Reviewers: Chia-Ping Tsai <[email protected]>
1 parent a12d38f commit b9413ea

File tree

54 files changed

+958
-1051
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

54 files changed

+958
-1051
lines changed

tools/src/test/java/org/apache/kafka/tools/AbstractResetIntegrationTest.java

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -51,12 +51,10 @@
5151
import java.io.File;
5252
import java.io.FileWriter;
5353
import java.util.ArrayList;
54-
import java.util.Arrays;
55-
import java.util.Collections;
5654
import java.util.List;
5755
import java.util.Map;
5856
import java.util.Properties;
59-
import java.util.stream.Collectors;
57+
import java.util.Set;
6058

6159
import static java.time.Duration.ofMillis;
6260
import static org.apache.kafka.streams.integration.utils.IntegrationTestUtils.waitForEmptyConsumerGroup;
@@ -176,7 +174,7 @@ void cleanupTest() throws Exception {
176174
}
177175

178176
private void add10InputElements() {
179-
final List<KeyValue<Long, String>> records = Arrays.asList(KeyValue.pair(0L, "aaa"),
177+
final List<KeyValue<Long, String>> records = List.of(KeyValue.pair(0L, "aaa"),
180178
KeyValue.pair(1L, "bbb"),
181179
KeyValue.pair(0L, "ccc"),
182180
KeyValue.pair(1L, "ddd"),
@@ -189,7 +187,7 @@ private void add10InputElements() {
189187

190188
for (final KeyValue<Long, String> record : records) {
191189
mockTime.sleep(10);
192-
IntegrationTestUtils.produceKeyValuesSynchronouslyWithTimestamp(INPUT_TOPIC, Collections.singleton(record), producerConfig, mockTime.milliseconds());
190+
IntegrationTestUtils.produceKeyValuesSynchronouslyWithTimestamp(INPUT_TOPIC, Set.of(record), producerConfig, mockTime.milliseconds());
193191
}
194192
}
195193

@@ -211,7 +209,7 @@ public void testResetWhenInternalTopicsAreSpecified(final TestInfo testInfo) thr
211209

212210
final List<String> internalTopics = cluster.getAllTopicsInCluster().stream()
213211
.filter(StreamsResetter::matchesInternalTopicFormat)
214-
.collect(Collectors.toList());
212+
.toList();
215213
cleanGlobal(false,
216214
"--internal-topics",
217215
String.join(",", internalTopics.subList(1, internalTopics.size())),
@@ -288,7 +286,7 @@ private void testReprocessingFromScratchAfterResetWithIntermediateUserTopic(fina
288286
if (!useRepartitioned) {
289287
IntegrationTestUtils.produceKeyValuesSynchronouslyWithTimestamp(
290288
INTERMEDIATE_USER_TOPIC,
291-
Collections.singleton(badMessage),
289+
Set.of(badMessage),
292290
producerConfig,
293291
mockTime.milliseconds());
294292
}
@@ -375,7 +373,7 @@ protected boolean tryCleanGlobal(final boolean withIntermediateTopics,
375373
final String resetScenarioArg,
376374
final String appID) throws Exception {
377375
final List<String> parameterList = new ArrayList<>(
378-
Arrays.asList("--application-id", appID,
376+
List.of("--application-id", appID,
379377
"--bootstrap-server", cluster.bootstrapServers(),
380378
"--input-topics", INPUT_TOPIC
381379
));

tools/src/test/java/org/apache/kafka/tools/BrokerApiVersionsCommandTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,8 @@
3131
import org.apache.kafka.server.config.ServerConfigs;
3232

3333
import java.util.Arrays;
34-
import java.util.Collections;
3534
import java.util.Iterator;
35+
import java.util.List;
3636
import java.util.Properties;
3737

3838
import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -56,7 +56,7 @@ public void testBrokerApiVersionsCommandOutput(ClusterInstance clusterInstance)
5656

5757
NodeApiVersions nodeApiVersions = new NodeApiVersions(
5858
ApiVersionsResponse.filterApis(listenerType, true, true),
59-
Collections.emptyList());
59+
List.of());
6060
Iterator<ApiKeys> apiKeysIter = ApiKeys.clientApis().iterator();
6161
while (apiKeysIter.hasNext()) {
6262
ApiKeys apiKey = apiKeysIter.next();

tools/src/test/java/org/apache/kafka/tools/ClientMetricsCommandTest.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@
3333
import org.mockito.ArgumentCaptor;
3434

3535
import java.util.Collection;
36-
import java.util.Collections;
3736
import java.util.Map;
3837
import java.util.Set;
3938
import java.util.concurrent.ExecutionException;
@@ -209,7 +208,7 @@ public void testDelete() {
209208
ClientMetricsCommand.ClientMetricsService service = new ClientMetricsCommand.ClientMetricsService(adminClient);
210209

211210
ConfigResource cr = new ConfigResource(ConfigResource.Type.CLIENT_METRICS, clientMetricsName);
212-
Config cfg = new Config(Collections.singleton(new ConfigEntry("metrics", "org.apache.kafka.producer.")));
211+
Config cfg = new Config(Set.of(new ConfigEntry("metrics", "org.apache.kafka.producer.")));
213212
DescribeConfigsResult describeResult = AdminClientTestUtils.describeConfigsResult(cr, cfg);
214213
when(adminClient.describeConfigs(any())).thenReturn(describeResult);
215214
AlterConfigsResult alterResult = AdminClientTestUtils.alterConfigsResult(cr);
@@ -237,7 +236,7 @@ public void testDescribe() {
237236
ConfigResource.Type.CLIENT_METRICS, Set.of(clientMetricsName)
238237
));
239238
when(adminClient.listConfigResources(any(), any())).thenReturn(listConfigResourcesResult);
240-
Config cfg = new Config(Collections.singleton(new ConfigEntry("metrics", "org.apache.kafka.producer.")));
239+
Config cfg = new Config(Set.of(new ConfigEntry("metrics", "org.apache.kafka.producer.")));
241240
DescribeConfigsResult describeResult = AdminClientTestUtils.describeConfigsResult(cr, cfg);
242241
when(adminClient.describeConfigs(any())).thenReturn(describeResult);
243242

@@ -284,7 +283,7 @@ public void testDescribeAll() {
284283
ListConfigResourcesResult result = AdminClientTestUtils.listConfigResourcesResult(clientMetricsName);
285284
when(adminClient.listConfigResources(any(), any())).thenReturn(result);
286285
ConfigResource cr = new ConfigResource(ConfigResource.Type.CLIENT_METRICS, clientMetricsName);
287-
Config cfg = new Config(Collections.singleton(new ConfigEntry("metrics", "org.apache.kafka.producer.")));
286+
Config cfg = new Config(Set.of(new ConfigEntry("metrics", "org.apache.kafka.producer.")));
288287
DescribeConfigsResult describeResult = AdminClientTestUtils.describeConfigsResult(cr, cfg);
289288
when(adminClient.describeConfigs(any())).thenReturn(describeResult);
290289

@@ -326,7 +325,7 @@ public void testListFailsWithUnsupportedVersionException() {
326325
ListConfigResourcesResult result = AdminClientTestUtils.listConfigResourcesResult(Errors.UNSUPPORTED_VERSION.exception());
327326
when(adminClient.listConfigResources(any(), any())).thenReturn(result);
328327

329-
assertThrows(ExecutionException.class, () -> service.listClientMetrics());
328+
assertThrows(ExecutionException.class, service::listClientMetrics);
330329
}
331330

332331
private void assertInitializeInvalidOptionsExitCode(int expected, String[] options) {

tools/src/test/java/org/apache/kafka/tools/ClusterToolTest.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@
2929
import java.util.Arrays;
3030
import java.util.List;
3131
import java.util.Set;
32-
import java.util.stream.Collectors;
3332

3433
import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
3534
import static org.junit.jupiter.api.Assertions.assertEquals;
@@ -71,10 +70,10 @@ public void testListEndpointsWithBootstrapServer(ClusterInstance clusterInstance
7170

7271
@ClusterTest(brokers = 2, types = {Type.KRAFT, Type.CO_KRAFT})
7372
public void testListEndpointsArgumentWithBootstrapServer(ClusterInstance clusterInstance) {
74-
List<Integer> brokerIds = clusterInstance.brokerIds().stream().collect(Collectors.toList());
73+
List<Integer> brokerIds = clusterInstance.brokerIds().stream().toList();
7574
clusterInstance.shutdownBroker(brokerIds.get(0));
7675

77-
List<String> ports = Arrays.stream(clusterInstance.bootstrapServers().split(",")).map(b -> b.split(":")[1]).collect(Collectors.toList());
76+
List<String> ports = Arrays.stream(clusterInstance.bootstrapServers().split(",")).map(b -> b.split(":")[1]).toList();
7877
String format = "%-10s %-9s %-10s %-10s %-10s %-15s%n%-10s %-9s %-10s %-10s %-10s %-15s%n%-10s %-9s %-10s %-10s %-10s %-6s";
7978
String expected = String.format(format,
8079
"ID", "HOST", "PORT", "RACK", "STATE", "ENDPOINT_TYPE",
@@ -110,7 +109,7 @@ public void testListEndpointsWithBootstrapController(ClusterInstance clusterInst
110109
int id = clusterInstance.controllerIds().iterator().next();
111110
String format = "%-10s %-9s %-10s %-10s %-15s%n%-10s %-9s %-10s %-10s %-10s";
112111
String expected = String.format(format, "ID", "HOST", "PORT", "RACK", "ENDPOINT_TYPE", id, "localhost", port, "null", "controller");
113-
assertTrue(output.equals(expected));
112+
assertEquals(expected, output);
114113
}
115114

116115
@ClusterTest(brokers = 3, types = {Type.KRAFT, Type.CO_KRAFT})

0 commit comments

Comments
 (0)