/** * * * <pre> * The list of requested instances. * </pre> * * <code>repeated .google.bigtable.admin.v2.Instance instances = 1;</code> */ public Builder setInstances( int index, com.google.bigtable.admin.v2.Instance.Builder builderForValue) { if (instancesBuilder_ == null) { ensureInstancesIsMutable(); instances_.set(index, builderForValue.build()); onChanged(); } else { instancesBuilder_.setMessage(index, builderForValue.build()); } return this; } /**
/** * * * <pre> * The list of requested instances. * </pre> * * <code>repeated .google.bigtable.admin.v2.Instance instances = 1;</code> */ public Builder addInstances(com.google.bigtable.admin.v2.Instance.Builder builderForValue) { if (instancesBuilder_ == null) { ensureInstancesIsMutable(); instances_.add(builderForValue.build()); onChanged(); } else { instancesBuilder_.addMessage(builderForValue.build()); } return this; } /**
/** * * * <pre> * The list of requested instances. * </pre> * * <code>repeated .google.bigtable.admin.v2.Instance instances = 1;</code> */ public Builder addInstances( int index, com.google.bigtable.admin.v2.Instance.Builder builderForValue) { if (instancesBuilder_ == null) { ensureInstancesIsMutable(); instances_.add(index, builderForValue.build()); onChanged(); } else { instancesBuilder_.addMessage(index, builderForValue.build()); } return this; } /**
/** * <pre> * The list of requested instances. * </pre> * * <code>repeated .google.bigtable.admin.v2.Instance instances = 1;</code> */ public Builder addInstances( int index, com.google.bigtable.admin.v2.Instance.Builder builderForValue) { if (instancesBuilder_ == null) { ensureInstancesIsMutable(); instances_.add(index, builderForValue.build()); onChanged(); } else { instancesBuilder_.addMessage(index, builderForValue.build()); } return this; } /**
/** * <pre> * The list of requested instances. * </pre> * * <code>repeated .google.bigtable.admin.v2.Instance instances = 1;</code> */ public Builder addInstances( com.google.bigtable.admin.v2.Instance.Builder builderForValue) { if (instancesBuilder_ == null) { ensureInstancesIsMutable(); instances_.add(builderForValue.build()); onChanged(); } else { instancesBuilder_.addMessage(builderForValue.build()); } return this; } /**
com.google.bigtable.admin.v2.Instance.newBuilder().setName(INSTANCE_NAME).build();
String displayName = "displayName1615086568"; Instance expectedResponse = Instance.newBuilder().setName(name.toString()).setDisplayName(displayName).build(); Operation resultOperation = Operation.newBuilder() Instance instance = Instance.newBuilder().build(); Map<String, Cluster> clusters = new HashMap<>();
com.google.bigtable.admin.v2.Instance.newBuilder() .setName(INSTANCE_NAME + "1") .build()) .addFailedLocations(NameUtil.formatLocationName(PROJECT_ID, "us-east1-d")) .build();
@Test @SuppressWarnings("all") public void partialUpdateInstanceTest() throws Exception { InstanceName name = InstanceName.of("[PROJECT]", "[INSTANCE]"); String displayName = "displayName1615086568"; Instance expectedResponse = Instance.newBuilder().setName(name.toString()).setDisplayName(displayName).build(); Operation resultOperation = Operation.newBuilder() .setName("partialUpdateInstanceTest") .setDone(true) .setResponse(Any.pack(expectedResponse)) .build(); mockBigtableInstanceAdmin.addResponse(resultOperation); Instance instance = Instance.newBuilder().build(); FieldMask updateMask = FieldMask.newBuilder().build(); Instance actualResponse = client.partialUpdateInstanceAsync(instance, updateMask).get(); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockBigtableInstanceAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); PartialUpdateInstanceRequest actualRequest = (PartialUpdateInstanceRequest) actualRequests.get(0); Assert.assertEquals(instance, actualRequest.getInstance()); Assert.assertEquals(updateMask, actualRequest.getUpdateMask()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test public void testListInstances() { // Setup com.google.bigtable.admin.v2.ListInstancesRequest expectedRequest = com.google.bigtable.admin.v2.ListInstancesRequest.newBuilder() .setParent(PROJECT_NAME) .build(); com.google.bigtable.admin.v2.ListInstancesResponse expectedResponse = com.google.bigtable.admin.v2.ListInstancesResponse.newBuilder() .addInstances( com.google.bigtable.admin.v2.Instance.newBuilder() .setName(INSTANCE_NAME + "1") .build()) .addInstances( com.google.bigtable.admin.v2.Instance.newBuilder() .setName(INSTANCE_NAME + "2") .build()) .build(); Mockito.when(mockListInstancesCallable.futureCall(expectedRequest)) .thenReturn(ApiFutures.immediateFuture(expectedResponse)); // Execute List<Instance> actualResult = adminClient.listInstances(); // Verify assertThat(actualResult) .containsExactly( Instance.fromProto(expectedResponse.getInstances(0)), Instance.fromProto(expectedResponse.getInstances(1))); }
@Test public void testUpdateInstance() { // Setup com.google.bigtable.admin.v2.PartialUpdateInstanceRequest expectedRequest = com.google.bigtable.admin.v2.PartialUpdateInstanceRequest.newBuilder() .setUpdateMask(FieldMask.newBuilder().addPaths("display_name")) .setInstance( com.google.bigtable.admin.v2.Instance.newBuilder() .setName(INSTANCE_NAME) .setDisplayName("new display name")) .build(); com.google.bigtable.admin.v2.Instance expectedResponse = com.google.bigtable.admin.v2.Instance.newBuilder().setName(INSTANCE_NAME).build(); mockOperationResult(mockUpdateInstanceCallable, expectedRequest, expectedResponse); // Execute Instance actualResult = adminClient.updateInstance( UpdateInstanceRequest.of(INSTANCE_ID).setDisplayName("new display name")); // Verify assertThat(actualResult).isEqualTo(Instance.fromProto(expectedResponse)); }
/** * * * <pre> * The instance to create. * Fields marked `OutputOnly` must be left blank. * </pre> * * <code>.google.bigtable.admin.v2.Instance instance = 3;</code> */ public Builder setInstance(com.google.bigtable.admin.v2.Instance.Builder builderForValue) { if (instanceBuilder_ == null) { instance_ = builderForValue.build(); onChanged(); } else { instanceBuilder_.setMessage(builderForValue.build()); } return this; } /**
/** * * * <pre> * The Instance which will (partially) replace the current value. * </pre> * * <code>.google.bigtable.admin.v2.Instance instance = 1;</code> */ public Builder setInstance(com.google.bigtable.admin.v2.Instance.Builder builderForValue) { if (instanceBuilder_ == null) { instance_ = builderForValue.build(); onChanged(); } else { instanceBuilder_.setMessage(builderForValue.build()); } return this; } /**
@Test @SuppressWarnings("all") public void getInstanceTest() { InstanceName name2 = InstanceName.of("[PROJECT]", "[INSTANCE]"); String displayName = "displayName1615086568"; Instance expectedResponse = Instance.newBuilder().setName(name2.toString()).setDisplayName(displayName).build(); mockBigtableInstanceAdmin.addResponse(expectedResponse); InstanceName name = InstanceName.of("[PROJECT]", "[INSTANCE]"); Instance actualResponse = client.getInstance(name); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockBigtableInstanceAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); GetInstanceRequest actualRequest = (GetInstanceRequest) actualRequests.get(0); Assert.assertEquals(name, InstanceName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test public void testFromProto() { com.google.bigtable.admin.v2.Instance proto = com.google.bigtable.admin.v2.Instance.newBuilder() .setName("projects/my-project/instances/my-instance") .setDisplayName("my display name") .setType(com.google.bigtable.admin.v2.Instance.Type.PRODUCTION) .setState(com.google.bigtable.admin.v2.Instance.State.READY) .putLabels("label1", "value1") .putLabels("label2", "value2") .build(); Instance result = Instance.fromProto(proto); assertThat(result.getId()).isEqualTo("my-instance"); assertThat(result.getDisplayName()).isEqualTo("my display name"); assertThat(result.getType()).isEqualTo(Instance.Type.PRODUCTION); assertThat(result.getState()).isEqualTo(Instance.State.READY); assertThat(result.getLabels()) .containsExactly( "label1", "value1", "label2", "value2"); }
@Test @SuppressWarnings("all") public void createInstanceExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockBigtableInstanceAdmin.addException(exception); try { ProjectName parent = ProjectName.of("[PROJECT]"); String instanceId = "instanceId-2101995259"; Instance instance = Instance.newBuilder().build(); Map<String, Cluster> clusters = new HashMap<>(); client.createInstanceAsync(parent, instanceId, instance, clusters).get(); Assert.fail("No exception raised"); } catch (ExecutionException e) { Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass()); InvalidArgumentException apiException = (InvalidArgumentException) e.getCause(); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } }
@Test public void testRequiresName() { com.google.bigtable.admin.v2.Instance proto = com.google.bigtable.admin.v2.Instance.newBuilder() .setDisplayName("my display name") .setType(com.google.bigtable.admin.v2.Instance.Type.PRODUCTION) .setState(com.google.bigtable.admin.v2.Instance.State.READY) .putLabels("label1", "value1") .putLabels("label2", "value2") .build(); Exception actualException = null; try { Instance.fromProto(proto); } catch (Exception e) { actualException = e; } assertThat(actualException).isInstanceOf(IllegalArgumentException.class); }
@Test @SuppressWarnings("all") public void partialUpdateInstanceExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockBigtableInstanceAdmin.addException(exception); try { Instance instance = Instance.newBuilder().build(); FieldMask updateMask = FieldMask.newBuilder().build(); client.partialUpdateInstanceAsync(instance, updateMask).get(); Assert.fail("No exception raised"); } catch (ExecutionException e) { Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass()); InvalidArgumentException apiException = (InvalidArgumentException) e.getCause(); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } }
@Test public void testExistsTrue() { // Setup com.google.bigtable.admin.v2.Instance expectedResponse = com.google.bigtable.admin.v2.Instance.newBuilder() .setName(NameUtil.formatInstanceName(PROJECT_ID, INSTANCE_ID)) .build(); Mockito.when(mockGetInstanceCallable.futureCall(Matchers.any(GetInstanceRequest.class))) .thenReturn(ApiFutures.immediateFuture(expectedResponse)); // Execute boolean found = adminClient.exists(INSTANCE_ID); // Verify assertThat(found).isTrue(); }
@Test public void testGetInstance() { // Setup com.google.bigtable.admin.v2.GetInstanceRequest expectedRequest = com.google.bigtable.admin.v2.GetInstanceRequest.newBuilder().setName(INSTANCE_NAME).build(); com.google.bigtable.admin.v2.Instance expectedResponse = com.google.bigtable.admin.v2.Instance.newBuilder().setName(INSTANCE_NAME).build(); Mockito.when(mockGetInstanceCallable.futureCall(expectedRequest)) .thenReturn(ApiFutures.immediateFuture(expectedResponse)); // Execute Instance actualResult = adminClient.getInstance(INSTANCE_ID); // Verify assertThat(actualResult).isEqualTo(Instance.fromProto(expectedResponse)); }