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_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
