def _decode_update_row(self, body):
proto = pb.UpdateRowResponse()
proto.ParseFromString(body)
consumed = self._parse_capacity_unit(proto.consumed.capacity_unit)
return_row = None
if len(proto.row) != 0:
inputStream = PlainBufferInputStream(proto.row)
codedInputStream = PlainBufferCodedInputStream(inputStream)
primary_key, attribute_columns = codedInputStream.read_row()
return_row = Row(primary_key, attribute_columns)
return (consumed, return_row), proto
def _mock_update_row(self, body):
proto = pb2.UpdateRowResponse()
proto.consumed.capacity_unit.write = 10
return proto