@Test public void testStopReplicaRequestNormalization() { Set<TopicPartition> tps = generateRandomTopicPartitions(10, 10); Map<TopicPartition, UpdateMetadataRequest.PartitionState> partitionStates = new HashMap<>(); for (TopicPartition tp: tps) { partitionStates.put(tp, new UpdateMetadataRequest.PartitionState(0, 0, 0, Collections.emptyList(), 0, Collections.emptyList(), Collections.emptyList())); } StopReplicaRequest.Builder builder = new StopReplicaRequest.Builder((short) 5, 0, 0, 0, false, tps); Assert.assertTrue(builder.build((short) 1).size() < builder.build((short) 0).size()); }