def test_encode_data_for_put(self):
c = PbufCodec()
tsobj = TsObject(None, self.table, self.rows, None)
msg = c.encode_timeseries_put(tsobj)
req = riak.pb.riak_ts_pb2.TsPutReq()
req.ParseFromString(msg.data)
# NB: expected, actual
self.assertEqual(self.table.name, bytes_to_str(req.table))
self.assertEqual(len(self.rows), len(req.rows))
r0 = req.rows[0]
self.assertEqual(bytes_to_str(r0.cells[0].varchar_value),
self.rows[0][0])
self.assertEqual(r0.cells[1].sint64_value, self.rows[0][1])
self.assertEqual(r0.cells[2].double_value, self.rows[0][2])
self.assertEqual(r0.cells[3].timestamp_value, self.ts0ms)
self.assertEqual(r0.cells[4].boolean_value, self.rows[0][4])
self.assertFalse(r0.cells[5].HasField('varchar_value'))
r1 = req.rows[1]
self.assertEqual(bytes_to_str(r1.cells[0].varchar_value),
self.rows[1][0])
self.assertEqual(r1.cells[1].sint64_value, self.rows[1][1])
self.assertEqual(r1.cells[2].double_value, self.rows[1][2])
self.assertEqual(r1.cells[3].timestamp_value, self.ts1ms)
self.assertEqual(r1.cells[4].boolean_value, self.rows[1][4])
self.assertEqual(r1.cells[5].varchar_value, self.rows[1][5])
def test_encode_data_for_put(self):
c = PbufCodec()
tsobj = TsObject(None, self.table, self.rows, None)
msg = c.encode_timeseries_put(tsobj)
req = riak.pb.riak_ts_pb2.TsPutReq()
req.ParseFromString(msg.data)
# NB: expected, actual
self.assertEqual(self.table.name, bytes_to_str(req.table))
self.assertEqual(len(self.rows), len(req.rows))
r0 = req.rows[0]
self.assertEqual(bytes_to_str(r0.cells[0].varchar_value),
self.rows[0][0])
self.assertEqual(r0.cells[1].sint64_value, self.rows[0][1])
self.assertEqual(r0.cells[2].double_value, self.rows[0][2])
self.assertEqual(r0.cells[3].timestamp_value, self.ts0ms)
self.assertEqual(r0.cells[4].boolean_value, self.rows[0][4])
self.assertFalse(r0.cells[5].HasField('varchar_value'))
r1 = req.rows[1]
self.assertEqual(bytes_to_str(r1.cells[0].varchar_value),
self.rows[1][0])
self.assertEqual(r1.cells[1].sint64_value, self.rows[1][1])
self.assertEqual(r1.cells[2].double_value, self.rows[1][2])
self.assertEqual(r1.cells[3].timestamp_value, self.ts1ms)
self.assertEqual(r1.cells[4].boolean_value, self.rows[1][4])
self.assertEqual(r1.cells[5].varchar_value, self.rows[1][5])
