use of org.apache.bookkeeper.stream.proto.storage.StorageContainerRequest in project bookkeeper by apache.
the class TestKvUtils method testNewKvRangeRequest.
@Test
public void testNewKvRangeRequest() {
try (RangeOption<ByteBuf> rangeOption = optionFactory.newRangeOption().endKey(key.retainedDuplicate()).countOnly(true).keysOnly(true).limit(10).maxCreateRev(1234L).minCreateRev(234L).maxModRev(2345L).minModRev(1235L).build()) {
RangeRequest.Builder rrBuilder = newRangeRequest(key, rangeOption);
StorageContainerRequest request = newKvRangeRequest(scId, rrBuilder);
assertEquals(scId, request.getScId());
assertEquals(KV_RANGE_REQ, request.getRequestCase());
assertEquals(rrBuilder.build(), request.getKvRangeReq());
}
}
use of org.apache.bookkeeper.stream.proto.storage.StorageContainerRequest in project bookkeeper by apache.
the class TestGrpcTableService method testRangeActiveRangesException.
@Test
public void testRangeActiveRangesException() throws Exception {
RangeStore rangeService = mock(RangeStore.class);
GrpcTableService grpcService = new GrpcTableService(rangeService);
StorageContainerRequest request = StorageContainerRequest.newBuilder().setKvRangeReq(RangeRequest.newBuilder().setKey(TEST_KEY).setHeader(ROUTING_HEADER)).build();
when(rangeService.range(request)).thenReturn(FutureUtils.exception(new StatusRuntimeException(Status.NOT_FOUND)));
TestResponseObserver<StorageContainerResponse> responseObserver = new TestResponseObserver<>();
grpcService.range(request, responseObserver);
responseObserver.verifyException(Status.NOT_FOUND);
verify(rangeService, times(1)).range(request);
}
use of org.apache.bookkeeper.stream.proto.storage.StorageContainerRequest in project bookkeeper by apache.
the class TestGrpcTableService method testPutFailure.
@Test
public void testPutFailure() throws Exception {
RangeStore rangeService = mock(RangeStore.class);
GrpcTableService grpcService = new GrpcTableService(rangeService);
StorageContainerRequest request = StorageContainerRequest.newBuilder().setKvPutReq(PutRequest.newBuilder().setKey(TEST_KEY).setValue(TEST_VAL).setHeader(ROUTING_HEADER)).build();
StorageContainerResponse response = StorageContainerResponse.newBuilder().setCode(StatusCode.INTERNAL_SERVER_ERROR).build();
when(rangeService.put(request)).thenReturn(FutureUtils.exception(CAUSE));
TestResponseObserver<StorageContainerResponse> responseObserver = new TestResponseObserver<>();
grpcService.put(request, responseObserver);
responseObserver.verifySuccess(response);
verify(rangeService, times(1)).put(request);
}
use of org.apache.bookkeeper.stream.proto.storage.StorageContainerRequest in project bookkeeper by apache.
the class TestGrpcTableService method testPutSuccess.
//
// Meta KeyRange Server Requests tests
//
@Test
public void testPutSuccess() throws Exception {
RangeStore rangeService = mock(RangeStore.class);
GrpcTableService grpcService = new GrpcTableService(rangeService);
StorageContainerRequest request = StorageContainerRequest.newBuilder().setKvPutReq(PutRequest.newBuilder().setKey(TEST_KEY).setValue(TEST_VAL).setHeader(ROUTING_HEADER)).build();
StorageContainerResponse response = StorageContainerResponse.newBuilder().setCode(StatusCode.SUCCESS).setKvPutResp(PutResponse.newBuilder()).build();
when(rangeService.put(request)).thenReturn(CompletableFuture.completedFuture(response));
TestResponseObserver<StorageContainerResponse> responseObserver = new TestResponseObserver<>();
grpcService.put(request, responseObserver);
responseObserver.verifySuccess(response);
verify(rangeService, times(1)).put(request);
}
use of org.apache.bookkeeper.stream.proto.storage.StorageContainerRequest in project bookkeeper by apache.
the class TestGrpcTableService method testRangeSuccess.
@Test
public void testRangeSuccess() throws Exception {
RangeStore rangeService = mock(RangeStore.class);
GrpcTableService grpcService = new GrpcTableService(rangeService);
StorageContainerRequest request = StorageContainerRequest.newBuilder().setKvRangeReq(RangeRequest.newBuilder().setKey(TEST_KEY).setHeader(ROUTING_HEADER)).build();
StorageContainerResponse response = StorageContainerResponse.newBuilder().setCode(StatusCode.SUCCESS).setKvRangeResp(RangeResponse.newBuilder()).build();
when(rangeService.range(request)).thenReturn(CompletableFuture.completedFuture(response));
TestResponseObserver<StorageContainerResponse> responseObserver = new TestResponseObserver<>();
grpcService.range(request, responseObserver);
responseObserver.verifySuccess(response);
verify(rangeService, times(1)).range(request);
}
Aggregations