Search in sources :

Example 1 with ListPartitionsResponse

use of io.confluent.kafkarest.entities.v3.ListPartitionsResponse in project kafka-rest by confluentinc.

the class PartitionsResourceIntegrationTest method listPartitions_existingTopic_returnPartitions.

@Test
public void listPartitions_existingTopic_returnPartitions() {
    String baseUrl = restConnect;
    String clusterId = getClusterId();
    ListPartitionsResponse expected = ListPartitionsResponse.create(PartitionDataList.builder().setMetadata(ResourceCollection.Metadata.builder().setSelf(baseUrl + "/v3/clusters/" + clusterId + "/topics/" + TOPIC_NAME + "/partitions").build()).setData(singletonList(PartitionData.builder().setMetadata(Resource.Metadata.builder().setSelf(baseUrl + "/v3/clusters/" + clusterId + "/topics/" + TOPIC_NAME + "/partitions/0").setResourceName("crn://" + "/kafka=" + clusterId + "/topic=" + TOPIC_NAME + "/partition=0").build()).setClusterId(clusterId).setTopicName(TOPIC_NAME).setPartitionId(0).setLeader(Resource.Relationship.create(baseUrl + "/v3/clusters/" + clusterId + "/topics/" + TOPIC_NAME + "/partitions/0/replicas/0")).setReplicas(Resource.Relationship.create(baseUrl + "/v3/clusters/" + clusterId + "/topics/" + TOPIC_NAME + "/partitions/0/replicas")).setReassignment(Resource.Relationship.create(baseUrl + "/v3/clusters/" + clusterId + "/topics/" + TOPIC_NAME + "/partitions/0/reassignment")).build())).build());
    Response response = request("/v3/clusters/" + clusterId + "/topics/" + TOPIC_NAME + "/partitions").accept(MediaType.APPLICATION_JSON).get();
    assertEquals(Status.OK.getStatusCode(), response.getStatus());
    ListPartitionsResponse actual = response.readEntity(ListPartitionsResponse.class);
    assertEquals(expected, actual);
}
Also used : ListPartitionsResponse(io.confluent.kafkarest.entities.v3.ListPartitionsResponse) Response(javax.ws.rs.core.Response) GetPartitionResponse(io.confluent.kafkarest.entities.v3.GetPartitionResponse) ListPartitionsResponse(io.confluent.kafkarest.entities.v3.ListPartitionsResponse) Test(org.junit.jupiter.api.Test)

Example 2 with ListPartitionsResponse

use of io.confluent.kafkarest.entities.v3.ListPartitionsResponse in project kafka-rest by confluentinc.

the class PartitionsResourceTest method listPartitions_existingTopic_returnsPartitions.

@Test
public void listPartitions_existingTopic_returnsPartitions() {
    expect(partitionManager.listPartitions(CLUSTER_ID, TOPIC_NAME)).andReturn(CompletableFuture.completedFuture(Arrays.asList(PARTITION_1, PARTITION_2, PARTITION_3)));
    replay(partitionManager);
    FakeAsyncResponse response = new FakeAsyncResponse();
    partitionsResource.listPartitions(response, CLUSTER_ID, TOPIC_NAME);
    ListPartitionsResponse expected = ListPartitionsResponse.create(PartitionDataList.builder().setMetadata(ResourceCollection.Metadata.builder().setSelf("/v3/clusters/cluster-1/topics/topic-1/partitions").build()).setData(Arrays.asList(PartitionData.builder().setMetadata(Resource.Metadata.builder().setSelf("/v3/clusters/cluster-1/topics/topic-1/partitions/0").setResourceName("crn:///kafka=cluster-1/topic=topic-1/partition=0").build()).setClusterId(CLUSTER_ID).setTopicName(TOPIC_NAME).setPartitionId(PARTITION_1.getPartitionId()).setLeader(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/0" + "/replicas/1")).setReplicas(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/0/replicas")).setReassignment(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/0" + "/reassignment")).build(), PartitionData.builder().setMetadata(Resource.Metadata.builder().setSelf("/v3/clusters/cluster-1/topics/topic-1/partitions/1").setResourceName("crn:///kafka=cluster-1/topic=topic-1/partition=1").build()).setClusterId(CLUSTER_ID).setTopicName(TOPIC_NAME).setPartitionId(PARTITION_2.getPartitionId()).setLeader(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/1" + "/replicas/2")).setReplicas(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/1/replicas")).setReassignment(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/1" + "/reassignment")).build(), PartitionData.builder().setMetadata(Resource.Metadata.builder().setSelf("/v3/clusters/cluster-1/topics/topic-1/partitions/2").setResourceName("crn:///kafka=cluster-1/topic=topic-1/partition=2").build()).setClusterId(CLUSTER_ID).setTopicName(TOPIC_NAME).setPartitionId(PARTITION_3.getPartitionId()).setLeader(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/2" + "/replicas/3")).setReplicas(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/2/replicas")).setReassignment(Resource.Relationship.create("/v3/clusters/cluster-1/topics/topic-1/partitions/2/reassignment")).build())).build());
    assertEquals(expected, response.getValue());
}
Also used : FakeAsyncResponse(io.confluent.kafkarest.response.FakeAsyncResponse) ListPartitionsResponse(io.confluent.kafkarest.entities.v3.ListPartitionsResponse) Test(org.junit.jupiter.api.Test)

Aggregations

ListPartitionsResponse (io.confluent.kafkarest.entities.v3.ListPartitionsResponse)2 Test (org.junit.jupiter.api.Test)2 GetPartitionResponse (io.confluent.kafkarest.entities.v3.GetPartitionResponse)1 FakeAsyncResponse (io.confluent.kafkarest.response.FakeAsyncResponse)1 Response (javax.ws.rs.core.Response)1