def test_read_rows(self): row = self._table.row(ROW_KEY) row_alt = self._table.row(ROW_KEY_ALT) self.rows_to_delete.extend([row, row_alt]) cell1, cell2, cell3, cell4 = self._write_to_row(row, row_alt, row, row_alt) row.commit() row_alt.commit() rows_data = self._table.read_rows() self.assertEqual(rows_data.rows, {}) rows_data.consume_all() # NOTE: We should refrain from editing protected data on instances. # Instead we should make the values public or provide factories # for constructing objects with them. row_data = PartialRowData(ROW_KEY) row_data._chunks_encountered = True row_data._committed = True row_data._cells = {COLUMN_FAMILY_ID1: {COL_NAME1: [cell1], COL_NAME2: [cell3]}} row_alt_data = PartialRowData(ROW_KEY_ALT) row_alt_data._chunks_encountered = True row_alt_data._committed = True row_alt_data._cells = { COLUMN_FAMILY_ID1: {COL_NAME1: [cell2]}, COLUMN_FAMILY_ID2: {COL_NAME3: [cell4]}, } expected_rows = {ROW_KEY: row_data, ROW_KEY_ALT: row_alt_data} self.assertEqual(rows_data.rows, expected_rows)
def test_table_read_rows(data_table, rows_to_delete): from google.cloud.bigtable.row_data import PartialRowData row = data_table.direct_row(ROW_KEY) rows_to_delete.append(row) row_alt = data_table.direct_row(ROW_KEY_ALT) rows_to_delete.append(row_alt) cell1, cell2, cell3, cell4 = _write_to_row(row, row_alt, row, row_alt) row.commit() row_alt.commit() rows_data = data_table.read_rows() assert rows_data.rows == {} rows_data.consume_all() # NOTE: We should refrain from editing protected data on instances. # Instead we should make the values public or provide factories # for constructing objects with them. row_data = PartialRowData(ROW_KEY) row_data._chunks_encountered = True row_data._committed = True row_data._cells = { COLUMN_FAMILY_ID1: { COL_NAME1: [cell1], COL_NAME2: [cell3] } } row_alt_data = PartialRowData(ROW_KEY_ALT) row_alt_data._chunks_encountered = True row_alt_data._committed = True row_alt_data._cells = { COLUMN_FAMILY_ID1: { COL_NAME1: [cell2] }, COLUMN_FAMILY_ID2: { COL_NAME3: [cell4] }, } expected_rows = {ROW_KEY: row_data, ROW_KEY_ALT: row_alt_data} assert rows_data.rows == expected_rows