Skip to content

Commit cc45f8e

Browse files
committed
chore: expect family column name
1 parent 4da9d67 commit cc45f8e

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

google/cloud/bigtable/tests/data_integration_test.cc

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -650,7 +650,7 @@ TEST_P(DataIntegrationTest, ClientQueryColumnFamily) {
650650

651651
auto prepared_query = client.PrepareQuery(
652652
instance_resource,
653-
SqlStatement("SELECT family4 AS c0 FROM " + quoted_table_name +
653+
SqlStatement("SELECT family4 AS family4 FROM " + quoted_table_name +
654654
" WHERE _key = '" + row_key + "'"));
655655

656656
ASSERT_STATUS_OK(prepared_query);
@@ -666,10 +666,11 @@ TEST_P(DataIntegrationTest, ClientQueryColumnFamily) {
666666
ASSERT_EQ(rows.size(), 1);
667667
ASSERT_STATUS_OK(rows[0]);
668668
auto const& row1 = *rows[0];
669-
ASSERT_EQ(row1.columns().at(0), column1);
670-
ASSERT_EQ(row1.columns().at(1), column2);
671-
ASSERT_EQ(row1.values().at(0), Value(value1));
672-
ASSERT_EQ(row1.values().at(1), Value(value2));
669+
ASSERT_EQ(row1.columns().size(), 1);
670+
ASSERT_EQ(row1.columns().at(0), family);
671+
ASSERT_EQ(row1.values().at(0),
672+
Value(std::unordered_map<std::string, std::string>{
673+
{column1, value1}, {column2, value2}}));
673674
}
674675

675676
TEST_P(DataIntegrationTest, ClientQueryColumnFamilyWithHistory) {
@@ -754,7 +755,7 @@ TEST_P(DataIntegrationTest, ClientQueryColumnFamilyWithHistory) {
754755
ASSERT_EQ(rows.size(), 1);
755756
ASSERT_TRUE(rows[0].ok()) << rows[0].status().message();
756757
auto const& row = *rows[0];
757-
ASSERT_EQ(row.columns().size(), 2);
758+
ASSERT_EQ(row.columns().size(), 1);
758759
EXPECT_EQ(row.columns().at(0), "family4_history");
759760

760761
auto value_hist = row.get("family4_history");

0 commit comments

Comments
 (0)