Search in sources :

Example 66 with MockGrpcService

use of com.google.api.gax.grpc.testing.MockGrpcService in project java-asset by googleapis.

the class AssetServiceClientTest method startStaticServer.

@BeforeClass
public static void startStaticServer() {
    mockAssetService = new MockAssetService();
    mockServiceHelper = new MockServiceHelper(UUID.randomUUID().toString(), Arrays.<MockGrpcService>asList(mockAssetService));
    mockServiceHelper.start();
}
Also used : MockGrpcService(com.google.api.gax.grpc.testing.MockGrpcService) MockServiceHelper(com.google.api.gax.grpc.testing.MockServiceHelper) BeforeClass(org.junit.BeforeClass)

Example 67 with MockGrpcService

use of com.google.api.gax.grpc.testing.MockGrpcService in project java-bigquerystorage by googleapis.

the class JsonStreamWriterTest method setUp.

@Before
public void setUp() throws Exception {
    testBigQueryWrite = new FakeBigQueryWrite();
    serviceHelper = new MockServiceHelper(UUID.randomUUID().toString(), Arrays.<MockGrpcService>asList(testBigQueryWrite));
    serviceHelper.start();
    channelProvider = serviceHelper.createChannelProvider();
    fakeExecutor = new FakeScheduledExecutorService();
    testBigQueryWrite.setExecutor(fakeExecutor);
    Instant time = Instant.now();
    Timestamp timestamp = Timestamp.newBuilder().setSeconds(time.getEpochSecond()).setNanos(time.getNano()).build();
    // Add enough GetWriteStream response.
    for (int i = 0; i < 4; i++) {
        testBigQueryWrite.addResponse(WriteStream.newBuilder().setName(TEST_STREAM).setCreateTime(timestamp).build());
    }
}
Also used : MockGrpcService(com.google.api.gax.grpc.testing.MockGrpcService) Instant(org.threeten.bp.Instant) MockServiceHelper(com.google.api.gax.grpc.testing.MockServiceHelper) Timestamp(com.google.protobuf.Timestamp) Before(org.junit.Before)

Example 68 with MockGrpcService

use of com.google.api.gax.grpc.testing.MockGrpcService in project java-bigquerystorage by googleapis.

the class BaseBigQueryReadClientTest method startStaticServer.

@BeforeClass
public static void startStaticServer() {
    mockBigQueryRead = new MockBigQueryRead();
    mockServiceHelper = new MockServiceHelper(UUID.randomUUID().toString(), Arrays.<MockGrpcService>asList(mockBigQueryRead));
    mockServiceHelper.start();
}
Also used : MockGrpcService(com.google.api.gax.grpc.testing.MockGrpcService) MockServiceHelper(com.google.api.gax.grpc.testing.MockServiceHelper) BeforeClass(org.junit.BeforeClass)

Example 69 with MockGrpcService

use of com.google.api.gax.grpc.testing.MockGrpcService in project java-bigquerystorage by googleapis.

the class BigQueryReadClientTest method startStaticServer.

@BeforeClass
public static void startStaticServer() {
    mockBigQueryRead = new MockBigQueryRead();
    serviceHelper = new MockServiceHelper(UUID.randomUUID().toString(), Arrays.<MockGrpcService>asList(mockBigQueryRead));
    serviceHelper.start();
}
Also used : MockGrpcService(com.google.api.gax.grpc.testing.MockGrpcService) MockServiceHelper(com.google.api.gax.grpc.testing.MockServiceHelper) BeforeClass(org.junit.BeforeClass)

Example 70 with MockGrpcService

use of com.google.api.gax.grpc.testing.MockGrpcService in project java-bigquerystorage by googleapis.

the class JsonStreamWriterTest method setUp.

@Before
public void setUp() throws Exception {
    testBigQueryWrite = new FakeBigQueryWrite();
    serviceHelper = new MockServiceHelper(UUID.randomUUID().toString(), Arrays.<MockGrpcService>asList(testBigQueryWrite));
    serviceHelper.start();
    channelProvider = serviceHelper.createChannelProvider();
    fakeExecutor = new FakeScheduledExecutorService();
    testBigQueryWrite.setExecutor(fakeExecutor);
    Instant time = Instant.now();
    Timestamp timestamp = Timestamp.newBuilder().setSeconds(time.getEpochSecond()).setNanos(time.getNano()).build();
    // Add enough GetWriteStream response.
    for (int i = 0; i < 4; i++) {
        testBigQueryWrite.addResponse(WriteStream.newBuilder().setName(TEST_STREAM).setCreateTime(timestamp).build());
    }
}
Also used : MockGrpcService(com.google.api.gax.grpc.testing.MockGrpcService) Instant(org.threeten.bp.Instant) MockServiceHelper(com.google.api.gax.grpc.testing.MockServiceHelper) Timestamp(com.google.protobuf.Timestamp) Before(org.junit.Before)

Aggregations

MockGrpcService (com.google.api.gax.grpc.testing.MockGrpcService)332 MockServiceHelper (com.google.api.gax.grpc.testing.MockServiceHelper)332 BeforeClass (org.junit.BeforeClass)327 MockIAMPolicy (com.google.iam.v1.MockIAMPolicy)4 Before (org.junit.Before)4 MockLocations (com.google.cloud.location.MockLocations)3 Timestamp (com.google.protobuf.Timestamp)2 Instant (org.threeten.bp.Instant)2 GaxGrpcProperties (com.google.api.gax.grpc.GaxGrpcProperties)1 LocalChannelProvider (com.google.api.gax.grpc.testing.LocalChannelProvider)1 MockStreamObserver (com.google.api.gax.grpc.testing.MockStreamObserver)1 StatusCode (com.google.api.gax.rpc.StatusCode)1 AnnotationNode (com.google.api.generator.engine.ast.AnnotationNode)1 AssignmentExpr (com.google.api.generator.engine.ast.AssignmentExpr)1 CastExpr (com.google.api.generator.engine.ast.CastExpr)1 ConcreteReference (com.google.api.generator.engine.ast.ConcreteReference)1 EnumRefExpr (com.google.api.generator.engine.ast.EnumRefExpr)1 Expr (com.google.api.generator.engine.ast.Expr)1 ExprStatement (com.google.api.generator.engine.ast.ExprStatement)1 InstanceofExpr (com.google.api.generator.engine.ast.InstanceofExpr)1