diff --git a/clients/src/test/java/org/apache/kafka/clients/admin/MockAdminClient.java b/clients/src/test/java/org/apache/kafka/clients/admin/MockAdminClient.java index 192eb6649a2b..239e8b728d56 100644 --- a/clients/src/test/java/org/apache/kafka/clients/admin/MockAdminClient.java +++ b/clients/src/test/java/org/apache/kafka/clients/admin/MockAdminClient.java @@ -1017,7 +1017,13 @@ synchronized public DescribeLogDirsResult describeLogDirs(Collection br for (Node node : nodes) { Map logDirDescriptionMap = unwrappedResults.get(node.id()); LogDirDescription logDirDescription = logDirDescriptionMap.getOrDefault(partitionLogDirs.get(0), new LogDirDescription(null, new HashMap<>())); - logDirDescription.replicaInfos().put(new TopicPartition(topicName, topicPartitionInfo.partition()), new ReplicaInfo(0, 0, false)); + Map topicPartitionReplicaInfoMap = new HashMap<>(logDirDescription.replicaInfos()); + topicPartitionReplicaInfoMap.put(new TopicPartition(topicName, topicPartitionInfo.partition()), new ReplicaInfo(0, 0, false)); + logDirDescriptionMap.put(partitionLogDirs.get(0), new LogDirDescription( + logDirDescription.error(), + topicPartitionReplicaInfoMap, + logDirDescription.totalBytes().orElse(DescribeLogDirsResponse.UNKNOWN_VOLUME_BYTES), + logDirDescription.usableBytes().orElse(DescribeLogDirsResponse.UNKNOWN_VOLUME_BYTES))); } } }