use of com.google.longrunning.GetOperationRequest in project gax-java by googleapis.
the class GrpcLongRunningClientTest method get.
@Test
public void get() {
OperationsStub operationsStub = mock(OperationsStub.class);
when(operationsStub.getOperationCallable()).thenReturn(new UnaryCallable<GetOperationRequest, Operation>() {
@Override
public ApiFuture<Operation> futureCall(GetOperationRequest request, ApiCallContext context) {
if (!request.getName().equals("poseidon")) {
return ApiFutures.immediateFailedFuture(new IllegalStateException("Unexpected request: " + request));
} else {
return ApiFutures.immediateFuture(Operation.newBuilder().setName("greece").build());
}
}
});
LongRunningClient longRunningClient = new GrpcLongRunningClient(operationsStub);
Truth.assertThat(longRunningClient.getOperationCallable().call("poseidon").getName()).isEqualTo("greece");
}
use of com.google.longrunning.GetOperationRequest in project java-spanner by googleapis.
the class GapicSpannerRpc method getOperation.
@Override
public Operation getOperation(String name) throws SpannerException {
acquireAdministrativeRequestsRateLimiter();
final GetOperationRequest request = GetOperationRequest.newBuilder().setName(name).build();
final GrpcCallContext context = newCallContext(null, name, request, OperationsGrpc.getGetOperationMethod());
return runWithRetryOnAdministrativeRequestsExceeded(() -> get(databaseAdminStub.getOperationsStub().getOperationCallable().futureCall(request, context)));
}
use of com.google.longrunning.GetOperationRequest in project java-spanner by googleapis.
the class AbstractMockServerTest method startStaticServer.
@BeforeClass
public static void startStaticServer() throws IOException {
mockSpanner = new MockSpannerServiceImpl();
// We don't want any unpredictable aborted transactions.
mockSpanner.setAbortProbability(0.0D);
mockInstanceAdmin = new MockInstanceAdminImpl();
mockDatabaseAdmin = new MockDatabaseAdminImpl();
mockOperations = new OperationsImplBase() {
@Override
public void getOperation(GetOperationRequest request, StreamObserver<Operation> responseObserver) {
responseObserver.onNext(Operation.newBuilder().setDone(false).setName(request.getName()).setMetadata(Any.pack(Empty.getDefaultInstance())).build());
responseObserver.onCompleted();
}
};
address = new InetSocketAddress("localhost", 0);
server = NettyServerBuilder.forAddress(address).addService(mockSpanner).addService(mockInstanceAdmin).addService(mockDatabaseAdmin).addService(mockOperations).build().start();
mockSpanner.putStatementResult(StatementResult.query(SELECT_COUNT_STATEMENT, SELECT_COUNT_RESULTSET_BEFORE_INSERT));
mockSpanner.putStatementResult(StatementResult.update(INSERT_STATEMENT, UPDATE_COUNT));
mockSpanner.putStatementResult(StatementResult.query(SELECT_RANDOM_STATEMENT, RANDOM_RESULT_SET));
futureParentHandlers = Logger.getLogger(AbstractFuture.class.getName()).getUseParentHandlers();
exceptionRunnableParentHandlers = Logger.getLogger(LogExceptionRunnable.class.getName()).getUseParentHandlers();
nettyServerParentHandlers = Logger.getLogger("io.grpc.netty.shaded.io.grpc.netty.NettyServerHandler").getUseParentHandlers();
clientStreamParentHandlers = Logger.getLogger("io.grpc.netty.shaded.io.grpc.netty.NettyServerHandler").getUseParentHandlers();
Logger.getLogger(AbstractFuture.class.getName()).setUseParentHandlers(false);
Logger.getLogger(LogExceptionRunnable.class.getName()).setUseParentHandlers(false);
Logger.getLogger("io.grpc.netty.shaded.io.grpc.netty.NettyServerHandler").setUseParentHandlers(false);
Logger.getLogger("io.grpc.internal.AbstractClientStream").setUseParentHandlers(false);
}
use of com.google.longrunning.GetOperationRequest in project bazel-buildfarm by bazelbuild.
the class BuildFarmServerTest method getOperation.
private Operation getOperation(String name) {
GetOperationRequest getRequest = GetOperationRequest.newBuilder().setName(name).build();
OperationsGrpc.OperationsBlockingStub operationsStub = OperationsGrpc.newBlockingStub(inProcessChannel);
return operationsStub.getOperation(getRequest);
}
Aggregations