@@ -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
675676TEST_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