Examples with AlterReplicaLogDirCollection org.apache.kafka.common.message.AlterReplicaLogDirsRequestData.AlterReplicaLogDirCollection used on opensource projects

Example 1 with AlterReplicaLogDirCollection

use of org.apache.kafka.common.message.AlterReplicaLogDirsRequestData.AlterReplicaLogDirCollection in project kafka by apache.

the class AlterReplicaLogDirsRequestTest method testErrorResponse.

@Test
public void testErrorResponse() {
    AlterReplicaLogDirsRequestData data = new AlterReplicaLogDirsRequestData().setDirs(new AlterReplicaLogDirCollection(singletonList(new AlterReplicaLogDir().setPath("/data0").setTopics(new AlterReplicaLogDirTopicCollection(singletonList(new AlterReplicaLogDirTopic().setName("topic").setPartitions(asList(0, 1, 2))).iterator()))).iterator()));
    AlterReplicaLogDirsResponse errorResponse = new AlterReplicaLogDirsRequest.Builder(data).build().getErrorResponse(123, new LogDirNotFoundException("/data0"));
    assertEquals(1, errorResponse.data().results().size());
    AlterReplicaLogDirTopicResult topicResponse = errorResponse.data().results().get(0);
    assertEquals("topic", topicResponse.topicName());
    assertEquals(3, topicResponse.partitions().size());
    for (int i = 0; i < 3; i++) {
        assertEquals(i, topicResponse.partitions().get(i).partitionIndex());
        assertEquals(Errors.LOG_DIR_NOT_FOUND.code(), topicResponse.partitions().get(i).errorCode());
    }
}

Example 2 with AlterReplicaLogDirCollection

use of org.apache.kafka.common.message.AlterReplicaLogDirsRequestData.AlterReplicaLogDirCollection in project kafka by apache.

the class AlterReplicaLogDirsRequestTest method testPartitionDir.

@Test
public void testPartitionDir() {
    AlterReplicaLogDirsRequestData data = new AlterReplicaLogDirsRequestData().setDirs(new AlterReplicaLogDirCollection(asList(new AlterReplicaLogDir().setPath("/data0").setTopics(new AlterReplicaLogDirTopicCollection(asList(new AlterReplicaLogDirTopic().setName("topic").setPartitions(asList(0, 1)), new AlterReplicaLogDirTopic().setName("topic2").setPartitions(asList(7))).iterator())), new AlterReplicaLogDir().setPath("/data1").setTopics(new AlterReplicaLogDirTopicCollection(asList(new AlterReplicaLogDirTopic().setName("topic3").setPartitions(asList(12))).iterator()))).iterator()));
    AlterReplicaLogDirsRequest request = new AlterReplicaLogDirsRequest.Builder(data).build();
    Map<TopicPartition, String> expect = new HashMap<>();
    expect.put(new TopicPartition("topic", 0), "/data0");
    expect.put(new TopicPartition("topic", 1), "/data0");
    expect.put(new TopicPartition("topic2", 7), "/data0");
    expect.put(new TopicPartition("topic3", 12), "/data1");
    assertEquals(expect, request.partitionDirs());
}