@java.lang.Override public Builder newBuilderForType() { return newBuilder(); }
/** * * * <pre> * The Instance which will (partially) replace the current value. * </pre> * * <code>.google.bigtable.admin.v2.Instance instance = 1;</code> */ public Builder mergeInstance(com.google.bigtable.admin.v2.Instance value) { if (instanceBuilder_ == null) { if (instance_ != null) { instance_ = com.google.bigtable.admin.v2.Instance.newBuilder(instance_) .mergeFrom(value) .buildPartial(); } else { instance_ = value; } onChanged(); } else { instanceBuilder_.mergeFrom(value); } return this; } /**
/** * * * <pre> * The instance to create. * Fields marked `OutputOnly` must be left blank. * </pre> * * <code>.google.bigtable.admin.v2.Instance instance = 3;</code> */ public Builder mergeInstance(com.google.bigtable.admin.v2.Instance value) { if (instanceBuilder_ == null) { if (instance_ != null) { instance_ = com.google.bigtable.admin.v2.Instance.newBuilder(instance_) .mergeFrom(value) .buildPartial(); } else { instance_ = value; } onChanged(); } else { instanceBuilder_.mergeFrom(value); } return this; } /**
@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)); }
@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 @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 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 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 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)); }
@Test public void testType() { UpdateInstanceRequest input = UpdateInstanceRequest.of("my-instance").setProductionType(); PartialUpdateInstanceRequest actual = input.toProto("my-project"); PartialUpdateInstanceRequest expected = PartialUpdateInstanceRequest.newBuilder() .setUpdateMask(FieldMask.newBuilder().addPaths("type")) .setInstance( Instance.newBuilder() .setName("projects/my-project/instances/my-instance") .setType(Type.PRODUCTION)) .build(); assertThat(actual).isEqualTo(expected); } }
@Test public void testLabels() { UpdateInstanceRequest input = UpdateInstanceRequest.of("my-instance") .setAllLabels( ImmutableMap.of( "label1", "value1", "label2", "value2")); PartialUpdateInstanceRequest actual = input.toProto("my-project"); PartialUpdateInstanceRequest expected = PartialUpdateInstanceRequest.newBuilder() .setUpdateMask(FieldMask.newBuilder().addPaths("labels")) .setInstance( Instance.newBuilder() .setName("projects/my-project/instances/my-instance") .putLabels("label1", "value1") .putLabels("label2", "value2")) .build(); assertThat(actual).isEqualTo(expected); }
com.google.bigtable.admin.v2.ListInstancesResponse.newBuilder() .addInstances( com.google.bigtable.admin.v2.Instance.newBuilder() .setName(INSTANCE_NAME + "1") .build())
@Test public void testDisplayName() { UpdateInstanceRequest input = UpdateInstanceRequest.of("my-instance").setDisplayName("my display name"); PartialUpdateInstanceRequest actual = input.toProto("my-project"); PartialUpdateInstanceRequest expected = PartialUpdateInstanceRequest.newBuilder() .setUpdateMask(FieldMask.newBuilder().addPaths("display_name")) .setInstance( Instance.newBuilder() .setName("projects/my-project/instances/my-instance") .setDisplayName("my display name")) .build(); assertThat(actual).isEqualTo(expected); }
@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())); }
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<>();
@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 testProductionSingle() { CreateInstanceRequest input = CreateInstanceRequest.of("my-instance") .setType(Instance.Type.PRODUCTION) .addCluster("cluster1", "us-east1-c", 3, StorageType.SSD); com.google.bigtable.admin.v2.CreateInstanceRequest actual = input.toProto("my-project"); com.google.bigtable.admin.v2.CreateInstanceRequest expected = com.google.bigtable.admin.v2.CreateInstanceRequest.newBuilder() .setParent(NameUtil.formatProjectName("my-project")) .setInstanceId("my-instance") .setInstance( com.google.bigtable.admin.v2.Instance.newBuilder() .setDisplayName("my-instance") .setType(com.google.bigtable.admin.v2.Instance.Type.PRODUCTION)) .putClusters( "cluster1", com.google.bigtable.admin.v2.Cluster.newBuilder() .setLocation("projects/my-project/locations/us-east1-c") .setServeNodes(3) .setDefaultStorageType(com.google.bigtable.admin.v2.StorageType.SSD) .build()) .build(); assertThat(actual).isEqualTo(expected); }
@Test public void testDevelopment() { CreateInstanceRequest input = CreateInstanceRequest.of("my-instance") .setType(Instance.Type.DEVELOPMENT) .addCluster("cluster1", "us-east1-c", 1, StorageType.SSD); com.google.bigtable.admin.v2.CreateInstanceRequest actual = input.toProto("my-project"); com.google.bigtable.admin.v2.CreateInstanceRequest expected = com.google.bigtable.admin.v2.CreateInstanceRequest.newBuilder() .setParent(NameUtil.formatProjectName("my-project")) .setInstanceId("my-instance") .setInstance( com.google.bigtable.admin.v2.Instance.newBuilder() .setDisplayName("my-instance") .setType(com.google.bigtable.admin.v2.Instance.Type.DEVELOPMENT)) .putClusters( "cluster1", com.google.bigtable.admin.v2.Cluster.newBuilder() .setLocation("projects/my-project/locations/us-east1-c") .setServeNodes(1) .setDefaultStorageType(com.google.bigtable.admin.v2.StorageType.SSD) .build()) .build(); assertThat(actual).isEqualTo(expected); }