def _from_pb_test_helper(self, labels=None):
import datetime
from google.cloud._helpers import _EPOCH
from google.cloud.bigtable_v2.types import data as data_v2_pb2
timestamp_micros = TestCell.timestamp_micros
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 = self._make_one(value, timestamp_micros)
else:
cell_pb = data_v2_pb2.Cell(value=value,
timestamp_micros=timestamp_micros,
labels=labels)
cell_expected = self._make_one(value,
timestamp_micros,
labels=labels)
klass = self._get_target_class()
result = klass.from_pb(cell_pb)
self.assertEqual(result, cell_expected)
self.assertEqual(result.timestamp, timestamp)
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
def _CellPB(*args, **kw):
from google.cloud.bigtable_v2.types import data as data_v2_pb2
return data_v2_pb2.Cell(*args, **kw)