def test_read_row_complete(self):
from google.cloud.bigtable.row_data import Cell
from google.cloud.bigtable.row_data import PartialRowData
chunk = _ReadRowsResponseCellChunkPB(
row_key=self.ROW_KEY,
family_name=self.FAMILY_NAME,
qualifier=self.QUALIFIER,
timestamp_micros=self.TIMESTAMP_MICROS,
value=self.VALUE,
commit_row=True,
)
chunks = [chunk]
expected_result = PartialRowData(row_key=self.ROW_KEY)
family = expected_result._cells.setdefault(self.FAMILY_NAME, {})
column = family.setdefault(self.QUALIFIER, [])
column.append(Cell.from_pb(chunk))
self._read_row_helper(chunks, expected_result)
def test_read_row_complete(self):
from google.cloud.bigtable.row_data import Cell
from google.cloud.bigtable.row_data import PartialRowData
chunk = _ReadRowsResponseCellChunkPB(
row_key=self.ROW_KEY,
family_name=self.FAMILY_NAME,
qualifier=self.QUALIFIER,
timestamp_micros=self.TIMESTAMP_MICROS,
value=self.VALUE,
commit_row=True,
)
chunks = [chunk]
expected_result = PartialRowData(row_key=self.ROW_KEY)
family = expected_result._cells.setdefault(self.FAMILY_NAME, {})
column = family.setdefault(self.QUALIFIER, [])
column.append(Cell.from_pb(chunk))
self._read_row_helper(chunks, expected_result)
def _cell_from_pb_test_helper(labels=None):
import datetime
from google.cloud._helpers import _EPOCH
from google.cloud.bigtable_v2.types import data as data_v2_pb2
from google.cloud.bigtable.row_data import Cell
timestamp = _EPOCH + datetime.timedelta(microseconds=TIMESTAMP_MICROS)
value = b"value-bytes"
if labels is None:
cell_pb = data_v2_pb2.Cell(value=value,
timestamp_micros=TIMESTAMP_MICROS)
cell_expected = _make_cell(value, TIMESTAMP_MICROS)
else:
cell_pb = data_v2_pb2.Cell(value=value,
timestamp_micros=TIMESTAMP_MICROS,
labels=labels)
cell_expected = _make_cell(value, TIMESTAMP_MICROS, labels=labels)
result = Cell.from_pb(cell_pb)
assert result == cell_expected
assert result.timestamp == timestamp
