def test__parse_rmw_row_response():
from google.cloud._helpers import _datetime_from_microseconds
from google.cloud.bigtable.row import _parse_rmw_row_response
col_fam1 = u"col-fam-id"
col_fam2 = u"col-fam-id2"
col_name1 = b"col-name1"
col_name2 = b"col-name2"
col_name3 = b"col-name3-but-other-fam"
cell_val1 = b"cell-val"
cell_val2 = b"cell-val-newer"
cell_val3 = b"altcol-cell-val"
cell_val4 = b"foo"
microseconds = 1000871
timestamp = _datetime_from_microseconds(microseconds)
expected_output = {
col_fam1: {
col_name1: [(cell_val1, timestamp), (cell_val2, timestamp)],
col_name2: [(cell_val3, timestamp)],
},
col_fam2: {
col_name3: [(cell_val4, timestamp)]
},
}
response_row = _RowPB(families=[
_FamilyPB(
name=col_fam1,
columns=[
_ColumnPB(
qualifier=col_name1,
cells=[
_CellPB(value=cell_val1,
timestamp_micros=microseconds),
_CellPB(value=cell_val2,
timestamp_micros=microseconds),
],
),
_ColumnPB(
qualifier=col_name2,
cells=[
_CellPB(value=cell_val3, timestamp_micros=microseconds)
],
),
],
),
_FamilyPB(
name=col_fam2,
columns=[
_ColumnPB(
qualifier=col_name3,
cells=[
_CellPB(value=cell_val4, timestamp_micros=microseconds)
],
)
],
),
])
sample_input = _ReadModifyWriteRowResponsePB(row=response_row)
assert expected_output == _parse_rmw_row_response(sample_input)
def _call_fut(self, row_response):
from google.cloud.bigtable.row import _parse_rmw_row_response
return _parse_rmw_row_response(row_response)
def _callFUT(self, row_response):
from google.cloud.bigtable.row import _parse_rmw_row_response
return _parse_rmw_row_response(row_response)
