/** * * * <pre> * Databases that matched the request. * </pre> * * <code>repeated .google.spanner.admin.database.v1.Database databases = 1;</code> */ public Builder setDatabases( int index, com.google.spanner.admin.database.v1.Database.Builder builderForValue) { if (databasesBuilder_ == null) { ensureDatabasesIsMutable(); databases_.set(index, builderForValue.build()); onChanged(); } else { databasesBuilder_.setMessage(index, builderForValue.build()); } return this; } /**
/** * * * <pre> * Databases that matched the request. * </pre> * * <code>repeated .google.spanner.admin.database.v1.Database databases = 1;</code> */ public Builder addDatabases( int index, com.google.spanner.admin.database.v1.Database.Builder builderForValue) { if (databasesBuilder_ == null) { ensureDatabasesIsMutable(); databases_.add(index, builderForValue.build()); onChanged(); } else { databasesBuilder_.addMessage(index, builderForValue.build()); } return this; } /**
/** * * * <pre> * Databases that matched the request. * </pre> * * <code>repeated .google.spanner.admin.database.v1.Database databases = 1;</code> */ public Builder addDatabases( com.google.spanner.admin.database.v1.Database.Builder builderForValue) { if (databasesBuilder_ == null) { ensureDatabasesIsMutable(); databases_.add(builderForValue.build()); onChanged(); } else { databasesBuilder_.addMessage(builderForValue.build()); } return this; } /**
/** * * * <pre> * Databases that matched the request. * </pre> * * <code>repeated .google.spanner.admin.database.v1.Database databases = 1;</code> */ public Builder addDatabases( int index, com.google.spanner.admin.database.v1.Database.Builder builderForValue) { if (databasesBuilder_ == null) { ensureDatabasesIsMutable(); databases_.add(index, builderForValue.build()); onChanged(); } else { databasesBuilder_.addMessage(index, builderForValue.build()); } return this; } /**
/** * * * <pre> * Databases that matched the request. * </pre> * * <code>repeated .google.spanner.admin.database.v1.Database databases = 1;</code> */ public Builder setDatabases( int index, com.google.spanner.admin.database.v1.Database.Builder builderForValue) { if (databasesBuilder_ == null) { ensureDatabasesIsMutable(); databases_.set(index, builderForValue.build()); onChanged(); } else { databasesBuilder_.setMessage(index, builderForValue.build()); } return this; } /**
/** * * * <pre> * Databases that matched the request. * </pre> * * <code>repeated .google.spanner.admin.database.v1.Database databases = 1;</code> */ public Builder addDatabases( com.google.spanner.admin.database.v1.Database.Builder builderForValue) { if (databasesBuilder_ == null) { ensureDatabasesIsMutable(); databases_.add(builderForValue.build()); onChanged(); } else { databasesBuilder_.addMessage(builderForValue.build()); } return this; } /**
@Test @SuppressWarnings("all") public void listDatabasesTest() { String nextPageToken = ""; Database databasesElement = Database.newBuilder().build(); List<Database> databases = Arrays.asList(databasesElement); ListDatabasesResponse expectedResponse = ListDatabasesResponse.newBuilder() .setNextPageToken(nextPageToken) .addAllDatabases(databases) .build(); mockDatabaseAdmin.addResponse(expectedResponse); InstanceName parent = InstanceName.of("[PROJECT]", "[INSTANCE]"); ListDatabasesPagedResponse pagedListResponse = client.listDatabases(parent); List<Database> resources = Lists.newArrayList(pagedListResponse.iterateAll()); Assert.assertEquals(1, resources.size()); Assert.assertEquals(expectedResponse.getDatabasesList().get(0), resources.get(0)); List<GeneratedMessageV3> actualRequests = mockDatabaseAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListDatabasesRequest actualRequest = (ListDatabasesRequest) actualRequests.get(0); Assert.assertEquals(parent, InstanceName.parse(actualRequest.getParent())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void createDatabaseTest() throws Exception { DatabaseName name = DatabaseName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"); Database expectedResponse = Database.newBuilder().setName(name.toString()).build(); Operation resultOperation = Operation.newBuilder() .setName("createDatabaseTest") .setDone(true) .setResponse(Any.pack(expectedResponse)) .build(); mockDatabaseAdmin.addResponse(resultOperation); InstanceName parent = InstanceName.of("[PROJECT]", "[INSTANCE]"); String createStatement = "createStatement552974828"; Database actualResponse = client.createDatabaseAsync(parent, createStatement).get(); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockDatabaseAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); CreateDatabaseRequest actualRequest = (CreateDatabaseRequest) actualRequests.get(0); Assert.assertEquals(parent, InstanceName.parse(actualRequest.getParent())); Assert.assertEquals(createStatement, actualRequest.getCreateStatement()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test public void waitForCompletes() throws Exception { com.google.longrunning.Operation proto = newBuilder().setName("op1").setDone(false).build(); Operation<Database, String> op = Operation.create(rpc, proto, new ParserImpl()); com.google.spanner.admin.database.v1.Database db = com.google.spanner.admin.database.v1.Database.newBuilder() .setName(NAME) .setState(com.google.spanner.admin.database.v1.Database.State.READY) .build(); com.google.longrunning.Operation proto2 = newBuilder().setName("op1").setDone(true).setResponse(Any.pack(db)).build(); when(rpc.getOperation("op1")).thenReturn(proto, proto2); op = op.waitFor( RetryOption.totalTimeout(Duration.ofSeconds(3)), RetryOption.initialRetryDelay(Duration.ZERO)); assertThat(op.getName()).isEqualTo("op1"); assertThat(op.isDone()).isTrue(); assertThat(op.isSuccessful()).isTrue(); assertThat(op.getMetadata()).isNull(); assertThat(op.getResult().getId().getName()).isEqualTo(NAME); }
@Test @SuppressWarnings("all") public void getDatabaseTest() { DatabaseName name2 = DatabaseName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"); Database expectedResponse = Database.newBuilder().setName(name2.toString()).build(); mockDatabaseAdmin.addResponse(expectedResponse); DatabaseName name = DatabaseName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"); Database actualResponse = client.getDatabase(name); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockDatabaseAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); GetDatabaseRequest actualRequest = (GetDatabaseRequest) actualRequests.get(0); Assert.assertEquals(name, DatabaseName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test public void successfulOperation() { com.google.spanner.admin.database.v1.Database db = com.google.spanner.admin.database.v1.Database.newBuilder() .setName(NAME) .setState(com.google.spanner.admin.database.v1.Database.State.READY) .build(); com.google.longrunning.Operation proto = newBuilder().setName("op1").setDone(true).setResponse(Any.pack(db)).build(); Operation<Database, String> op = Operation.create(rpc, proto, new ParserImpl()); assertThat(op.getName()).isEqualTo("op1"); assertThat(op.isDone()).isTrue(); assertThat(op.isSuccessful()).isTrue(); assertThat(op.getMetadata()).isNull(); assertThat(op.getResult().getId().getName()).isEqualTo(NAME); }
@Test public void reload() { com.google.longrunning.Operation proto = newBuilder().setName("op1").setDone(false).build(); Operation<Database, String> op = Operation.create(rpc, proto, new ParserImpl()); com.google.spanner.admin.database.v1.Database db = com.google.spanner.admin.database.v1.Database.newBuilder() .setName(NAME) .setState(com.google.spanner.admin.database.v1.Database.State.READY) .build(); proto = newBuilder().setName("op1").setDone(true).setResponse(Any.pack(db)).build(); when(rpc.getOperation("op1")).thenReturn(proto); op = op.reload(); assertThat(op.getName()).isEqualTo("op1"); assertThat(op.isDone()).isTrue(); assertThat(op.isSuccessful()).isTrue(); assertThat(op.getMetadata()).isNull(); assertThat(op.getResult().getId().getName()).isEqualTo(NAME); }
private Database getDatabaseProto() { return Database.newBuilder().setName(DB_NAME).setState(Database.State.READY).build(); }
private Database getAnotherDatabaseProto() { return Database.newBuilder().setName(DB_NAME2).setState(Database.State.READY).build(); }
private Database createDatabase() { com.google.spanner.admin.database.v1.Database proto = com.google.spanner.admin.database.v1.Database.newBuilder() .setName(NAME) .setState(com.google.spanner.admin.database.v1.Database.State.CREATING) .build(); return Database.fromProto(proto, dbClient); } }