Search in sources :

Example 6 with MockDatabaseAdminImpl

use of com.google.cloud.spanner.admin.database.v1.MockDatabaseAdminImpl 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);
}
Also used : LogExceptionRunnable(io.grpc.internal.LogExceptionRunnable) GetOperationRequest(com.google.longrunning.GetOperationRequest) InetSocketAddress(java.net.InetSocketAddress) MockDatabaseAdminImpl(com.google.cloud.spanner.admin.database.v1.MockDatabaseAdminImpl) MockInstanceAdminImpl(com.google.cloud.spanner.admin.instance.v1.MockInstanceAdminImpl) Operation(com.google.longrunning.Operation) OperationsImplBase(com.google.longrunning.OperationsGrpc.OperationsImplBase) MockSpannerServiceImpl(com.google.cloud.spanner.MockSpannerServiceImpl) AbstractFuture(com.google.common.util.concurrent.AbstractFuture) BeforeClass(org.junit.BeforeClass)

Example 7 with MockDatabaseAdminImpl

use of com.google.cloud.spanner.admin.database.v1.MockDatabaseAdminImpl in project liquibase-spanner by cloudspannerecosystem.

the class AbstractMockServerTest method startStaticServer.

@BeforeAll
static void startStaticServer() throws IOException {
    mockSpanner = new MockSpannerServiceImpl();
    mockSpanner.setAbortProbability(0.0D);
    mockAdmin = new MockDatabaseAdminImpl();
    address = new InetSocketAddress("localhost", 0);
    server = NettyServerBuilder.forAddress(address).addService(mockSpanner).addService(mockAdmin).intercept(new ServerInterceptor() {

        @Override
        public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(ServerCall<ReqT, RespT> call, Metadata headers, ServerCallHandler<ReqT, RespT> next) {
            // backend is the 'right' database, instead of at least checking the driver first...
            if (!(call.getMethodDescriptor().getFullMethodName().equals("google.spanner.v1.Spanner/BatchCreateSessions") || call.getMethodDescriptor().getFullMethodName().equals("google.spanner.v1.Spanner/ExecuteStreamingSql") || call.getMethodDescriptor().getFullMethodName().equals("google.spanner.v1.Spanner/DeleteSession"))) {
                String clientLibToken = headers.get(Key.of("x-goog-api-client", Metadata.ASCII_STRING_MARSHALLER));
                if (clientLibToken == null || !clientLibToken.contains("sp-liq")) {
                    if (!receivedRequestWithNonLiquibaseToken.getAndSet(true)) {
                        invalidClientLibToken = clientLibToken;
                    }
                }
            }
            return Contexts.interceptCall(Context.current(), call, headers, next);
        }
    }).build().start();
    registerDefaultResults();
}
Also used : ServerCallHandler(io.grpc.ServerCallHandler) ServerCall(io.grpc.ServerCall) InetSocketAddress(java.net.InetSocketAddress) MockDatabaseAdminImpl(com.google.cloud.spanner.admin.database.v1.MockDatabaseAdminImpl) ServerInterceptor(io.grpc.ServerInterceptor) UpdateDatabaseDdlMetadata(com.google.spanner.admin.database.v1.UpdateDatabaseDdlMetadata) Metadata(io.grpc.Metadata) ResultSetMetadata(com.google.spanner.v1.ResultSetMetadata) MockSpannerServiceImpl(com.google.cloud.spanner.MockSpannerServiceImpl) BeforeAll(org.junit.jupiter.api.BeforeAll)

Aggregations

MockDatabaseAdminImpl (com.google.cloud.spanner.admin.database.v1.MockDatabaseAdminImpl)7 InetSocketAddress (java.net.InetSocketAddress)6 MockSpannerServiceImpl (com.google.cloud.spanner.MockSpannerServiceImpl)5 BeforeClass (org.junit.BeforeClass)5 MockInstanceAdminImpl (com.google.cloud.spanner.admin.instance.v1.MockInstanceAdminImpl)4 UpdateDatabaseDdlMetadata (com.google.spanner.admin.database.v1.UpdateDatabaseDdlMetadata)2 ResultSetMetadata (com.google.spanner.v1.ResultSetMetadata)2 Metadata (io.grpc.Metadata)2 ServerCall (io.grpc.ServerCall)2 ServerCallHandler (io.grpc.ServerCallHandler)2 ServerInterceptor (io.grpc.ServerInterceptor)2 OptionsMetadata (com.google.cloud.spanner.pgadapter.metadata.OptionsMetadata)1 ImmutableList (com.google.common.collect.ImmutableList)1 AbstractFuture (com.google.common.util.concurrent.AbstractFuture)1 GetOperationRequest (com.google.longrunning.GetOperationRequest)1 Operation (com.google.longrunning.Operation)1 OperationsImplBase (com.google.longrunning.OperationsGrpc.OperationsImplBase)1 LogExceptionRunnable (io.grpc.internal.LogExceptionRunnable)1 ScheduledThreadPoolExecutor (java.util.concurrent.ScheduledThreadPoolExecutor)1 BeforeAll (org.junit.jupiter.api.BeforeAll)1