public static List<TableName> parseList(List<String> formattedStrings) { List<TableName> list = new ArrayList<>(formattedStrings.size()); for (String formattedString : formattedStrings) { list.add(parse(formattedString)); } return list; }
@InternalApi public static Table fromProto(@Nonnull com.google.bigtable.admin.v2.Table proto) { ImmutableMap.Builder<String, ReplicationState> replicationStates = ImmutableMap.builder(); for (Entry<String, com.google.bigtable.admin.v2.Table.ClusterState> entry : proto.getClusterStatesMap().entrySet()) { replicationStates.put( entry.getKey(), ReplicationState.fromProto(entry.getValue().getReplicationState())); } ImmutableList.Builder<ColumnFamily> columnFamilies = ImmutableList.builder(); for (Entry<String, com.google.bigtable.admin.v2.ColumnFamily> entry : proto.getColumnFamiliesMap().entrySet()) { columnFamilies.add(ColumnFamily.fromProto(entry.getKey(), entry.getValue())); } return new Table( TableName.parse(proto.getName()), replicationStates.build(), columnFamilies.build()); }
@Test public void testAwaitReplication() { // Setup @SuppressWarnings("UnnecessaryLocalVariable") TableName expectedRequest = TableName.parse(TABLE_NAME); final AtomicBoolean wasCalled = new AtomicBoolean(false); Mockito.when(mockAwaitReplicationCallable.futureCall(expectedRequest)) .thenAnswer( new Answer<ApiFuture<Void>>() { @Override public ApiFuture<Void> answer(InvocationOnMock invocationOnMock) { wasCalled.set(true); return ApiFutures.immediateFuture(null); } }); // Execute adminClient.awaitReplication(TABLE_ID); // Verify assertThat(wasCalled.get()).isTrue(); }
expectedResults.add(TableName.parse(expectedProto.getName()).getTable());
@Test @SuppressWarnings("all") public void deleteTableTest() { Empty expectedResponse = Empty.newBuilder().build(); mockBigtableTableAdmin.addResponse(expectedResponse); TableName name = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); client.deleteTable(name); List<GeneratedMessageV3> actualRequests = mockBigtableTableAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteTableRequest actualRequest = (DeleteTableRequest) actualRequests.get(0); Assert.assertEquals(name, TableName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void modifyColumnFamiliesTest() { TableName name2 = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); Table expectedResponse = Table.newBuilder().setName(name2.toString()).build(); mockBigtableTableAdmin.addResponse(expectedResponse); TableName name = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); List<ModifyColumnFamiliesRequest.Modification> modifications = new ArrayList<>(); Table actualResponse = client.modifyColumnFamilies(name, modifications); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockBigtableTableAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); ModifyColumnFamiliesRequest actualRequest = (ModifyColumnFamiliesRequest) actualRequests.get(0); Assert.assertEquals(name, TableName.parse(actualRequest.getName())); Assert.assertEquals(modifications, actualRequest.getModificationsList()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void generateConsistencyTokenTest() { String consistencyToken = "consistencyToken-1090516718"; GenerateConsistencyTokenResponse expectedResponse = GenerateConsistencyTokenResponse.newBuilder().setConsistencyToken(consistencyToken).build(); mockBigtableTableAdmin.addResponse(expectedResponse); TableName name = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); GenerateConsistencyTokenResponse actualResponse = client.generateConsistencyToken(name); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockBigtableTableAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); GenerateConsistencyTokenRequest actualRequest = (GenerateConsistencyTokenRequest) actualRequests.get(0); Assert.assertEquals(name, TableName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void getTableTest() { TableName name2 = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); Table expectedResponse = Table.newBuilder().setName(name2.toString()).build(); mockBigtableTableAdmin.addResponse(expectedResponse); TableName name = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); Table actualResponse = client.getTable(name); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockBigtableTableAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); GetTableRequest actualRequest = (GetTableRequest) actualRequests.get(0); Assert.assertEquals(name, TableName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void checkConsistencyTest() { boolean consistent = true; CheckConsistencyResponse expectedResponse = CheckConsistencyResponse.newBuilder().setConsistent(consistent).build(); mockBigtableTableAdmin.addResponse(expectedResponse); TableName name = TableName.of("[PROJECT]", "[INSTANCE]", "[TABLE]"); String consistencyToken = "consistencyToken-1090516718"; CheckConsistencyResponse actualResponse = client.checkConsistency(name, consistencyToken); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockBigtableTableAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); CheckConsistencyRequest actualRequest = (CheckConsistencyRequest) actualRequests.get(0); Assert.assertEquals(name, TableName.parse(actualRequest.getName())); Assert.assertEquals(consistencyToken, actualRequest.getConsistencyToken()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
SnapshotTableRequest actualRequest = (SnapshotTableRequest) actualRequests.get(0); Assert.assertEquals(name, TableName.parse(actualRequest.getName())); Assert.assertEquals(cluster, ClusterName.parse(actualRequest.getCluster())); Assert.assertEquals(snapshotId, SnapshotName.parse(actualRequest.getSnapshotId()));
public static List<TableName> parseList(List<String> formattedStrings) { List<TableName> list = new ArrayList<>(formattedStrings.size()); for (String formattedString : formattedStrings) { list.add(parse(formattedString)); } return list; }
@InternalApi public static Table fromProto(@Nonnull com.google.bigtable.admin.v2.Table proto) { ImmutableMap.Builder<String, ReplicationState> replicationStates = ImmutableMap.builder(); for (Entry<String, com.google.bigtable.admin.v2.Table.ClusterState> entry : proto.getClusterStatesMap().entrySet()) { replicationStates.put( entry.getKey(), ReplicationState.fromProto(entry.getValue().getReplicationState())); } ImmutableList.Builder<ColumnFamily> columnFamilies = ImmutableList.builder(); for (Entry<String, com.google.bigtable.admin.v2.ColumnFamily> entry : proto.getColumnFamiliesMap().entrySet()) { columnFamilies.add(ColumnFamily.fromProto(entry.getKey(), entry.getValue())); } return new Table( TableName.parse(proto.getName()), replicationStates.build(), columnFamilies.build()); }