@Test public void testScaleStream() throws ExecutionException, InterruptedException { when(this.mockControllerService.checkScale(anyString(), anyString(), anyInt())).thenReturn( CompletableFuture.completedFuture(Controller.ScaleStatusResponse.newBuilder() .setStatus(Controller.ScaleStatusResponse.ScaleStatus.SUCCESS).build())); when(this.mockControllerService.scale(any(), any(), any(), any(), anyLong())).thenReturn( CompletableFuture.completedFuture(Controller.ScaleResponse.newBuilder() .setStatus(Controller.ScaleResponse.ScaleStreamStatus.STARTED).build())); Assert.assertTrue(this.testController.scaleStream(new StreamImpl("scope", "stream"), new ArrayList<>(), new HashMap<>(), executorService()).getFuture().join()); when(this.mockControllerService.scale(any(), any(), any(), any(), anyLong())).thenReturn( CompletableFuture.completedFuture(Controller.ScaleResponse.newBuilder() .setStatus(Controller.ScaleResponse.ScaleStreamStatus.PRECONDITION_FAILED).build())); Assert.assertFalse(this.testController.scaleStream(new StreamImpl("scope", "stream"), new ArrayList<>(), new HashMap<>(), executorService()).getFuture().join()); when(this.mockControllerService.scale(any(), any(), any(), any(), anyLong())).thenReturn( CompletableFuture.completedFuture(Controller.ScaleResponse.newBuilder() .setStatus(Controller.ScaleResponse.ScaleStreamStatus.FAILURE).build())); assertThrows("Expected ControllerFailureException", () -> this.testController.startScale(new StreamImpl("scope", "stream"), new ArrayList<>(), new HashMap<>()).join(), ex -> ex instanceof ControllerFailureException); when(this.mockControllerService.scale(any(), any(), any(), any(), anyLong())).thenReturn( CompletableFuture.completedFuture(Controller.ScaleResponse.newBuilder() .setStatusValue(-1).build())); assertThrows("Expected ControllerFailureException", () -> this.testController.startScale(new StreamImpl("scope", "stream"), new ArrayList<>(), new HashMap<>()).join(), ex -> ex instanceof ControllerFailureException); }
public Builder mergeFrom( com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException { io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse parsedMessage = null; try { parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); } catch (com.google.protobuf.InvalidProtocolBufferException e) { parsedMessage = (io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse) e.getUnfinishedMessage(); throw e.unwrapIOException(); } finally { if (parsedMessage != null) { mergeFrom(parsedMessage); } } return this; }
Throwable e = Exceptions.unwrap(ex); if (e instanceof StoreException.DataNotFoundException) { response.setStatus(ScaleStatusResponse.ScaleStatus.INVALID_INPUT); } else { response.setStatus(ScaleStatusResponse.ScaleStatus.INTERNAL_ERROR); response.setStatus(ScaleStatusResponse.ScaleStatus.INVALID_INPUT); } else if (activeEpoch.getEpoch() == epoch || activeEpoch.getReferenceEpoch() == epoch) { response.setStatus(ScaleStatusResponse.ScaleStatus.IN_PROGRESS); } else { response.setStatus(ScaleStatusResponse.ScaleStatus.SUCCESS); return response.build(); });
private Builder() { maybeForceBuilderInitialization(); }
public static Builder newBuilder(io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse prototype) { return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); } public Builder toBuilder() {
public Builder mergeFrom(io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse other) { if (other == io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse.getDefaultInstance()) return this; if (other.status_ != 0) { setStatusValue(other.getStatusValue()); } this.mergeUnknownFields(other.unknownFields); onChanged(); return this; }
private Builder( com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { super(parent); maybeForceBuilderInitialization(); } private void maybeForceBuilderInitialization() {
public io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse build() { io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse result = buildPartial(); if (!result.isInitialized()) { throw newUninitializedMessageException(result); } return result; }
@Override public void checkScale(ScaleStatusRequest request, StreamObserver<ScaleStatusResponse> responseObserver) { if (request.getStreamInfo().getStream().equals("stream1")) { responseObserver.onNext(ScaleStatusResponse.newBuilder() .setStatus(ScaleStatusResponse.ScaleStatus.SUCCESS).build()); responseObserver.onCompleted(); } else { responseObserver.onError(Status.INTERNAL.withDescription("Server error").asRuntimeException()); } }
public Builder mergeFrom(com.google.protobuf.Message other) { if (other instanceof io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse) { return mergeFrom((io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse)other); } else { super.mergeFrom(other); return this; } }
@java.lang.Override protected Builder newBuilderForType( com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { Builder builder = new Builder(parent); return builder; } /**
public io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse buildPartial() { io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse result = new io.pravega.controller.stream.api.grpc.v1.Controller.ScaleStatusResponse(this); result.status_ = status_; onBuilt(); return result; }