@Override public void updateInstance(Instance request, StreamObserver<Instance> responseObserver) { Object response = responses.remove(); if (response instanceof Instance) { requests.add(request); responseObserver.onNext((Instance) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
public void onCompleted() { long seconds = NANOSECONDS.toSeconds(System.nanoTime() - startTime); responseObserver.onNext(RouteSummary.newBuilder().setPointCount(pointCount) .setFeatureCount(featureCount).setDistance(distance) .setElapsedTime((int) seconds).build()); responseObserver.onCompleted(); } };
@Override public void setIamPolicy(SetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { Object response = responses.remove(); if (response instanceof Policy) { requests.add(request); responseObserver.onNext((Policy) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
public Object answer(InvocationOnMock invocation) { StreamObserver<OpenLocalBlockResponse> observer = invocation.getArgumentAt(0, StreamObserver.class); observer.onNext(response); observer.onCompleted(); return mStreamObserver; } });
@Override public void getIamPolicy(GetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { Object response = responses.remove(); if (response instanceof Policy) { requests.add(request); responseObserver.onNext((Policy) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
public Object answer(InvocationOnMock invocation) { StreamObserver<OpenLocalBlockResponse> observer = invocation.getArgumentAt(0, StreamObserver.class); observer.onNext(OpenLocalBlockResponse.newBuilder().setPath("/tmp").build()); observer.onCompleted(); return requestObserver; } });
@Override public void importData(ImportDataRequest request, StreamObserver<Operation> responseObserver) { Object response = responses.remove(); if (response instanceof Operation) { requests.add(request); responseObserver.onNext((Operation) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void createModel(CreateModelRequest request, StreamObserver<Operation> responseObserver) { Object response = responses.remove(); if (response instanceof Operation) { requests.add(request); responseObserver.onNext((Operation) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void setIamPolicy(SetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { Object response = responses.remove(); if (response instanceof Policy) { requests.add(request); responseObserver.onNext((Policy) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void getIamPolicy(GetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { Object response = responses.remove(); if (response instanceof Policy) { requests.add(request); responseObserver.onNext((Policy) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void submitJob(SubmitJobRequest request, StreamObserver<Job> responseObserver) { Object response = responses.remove(); if (response instanceof Job) { requests.add(request); responseObserver.onNext((Job) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void updateJob(UpdateJobRequest request, StreamObserver<Job> responseObserver) { Object response = responses.remove(); if (response instanceof Job) { requests.add(request); responseObserver.onNext((Job) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void getDlpJob(GetDlpJobRequest request, StreamObserver<DlpJob> responseObserver) { Object response = responses.remove(); if (response instanceof DlpJob) { requests.add(request); responseObserver.onNext((DlpJob) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void exportAgent(ExportAgentRequest request, StreamObserver<Operation> responseObserver) { Object response = responses.remove(); if (response instanceof Operation) { requests.add(request); responseObserver.onNext((Operation) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void importAgent(ImportAgentRequest request, StreamObserver<Operation> responseObserver) { Object response = responses.remove(); if (response instanceof Operation) { requests.add(request); responseObserver.onNext((Operation) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void getInstance(GetInstanceRequest request, StreamObserver<Instance> responseObserver) { Object response = responses.remove(); if (response instanceof Instance) { requests.add(request); responseObserver.onNext((Instance) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void createIntent(CreateIntentRequest request, StreamObserver<Intent> responseObserver) { Object response = responses.remove(); if (response instanceof Intent) { requests.add(request); responseObserver.onNext((Intent) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void deleteContext(DeleteContextRequest request, StreamObserver<Empty> responseObserver) { Object response = responses.remove(); if (response instanceof Empty) { requests.add(request); responseObserver.onNext((Empty) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void getIamPolicy(GetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { Object response = responses.remove(); if (response instanceof Policy) { requests.add(request); responseObserver.onNext((Policy) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }
@Override public void importAgent(ImportAgentRequest request, StreamObserver<Operation> responseObserver) { Object response = responses.remove(); if (response instanceof Operation) { requests.add(request); responseObserver.onNext((Operation) response); responseObserver.onCompleted(); } else if (response instanceof Exception) { responseObserver.onError((Exception) response); } else { responseObserver.onError(new IllegalArgumentException("Unrecognized response type")); } }