def test_repr(self):
p = BoltResponse(SofaHeader(a='1', b='2'), b"cdefgab", ptype=PTYPE.ONEWAY, request_id=0,
cmdcode=CMDCODE.HEARTBEAT,
respstatus=1)
print(p)
s = p.to_stream()
pr = BoltResponse.from_stream(s)
self.assertNotEqual(id(p), id(pr))
self.assertEqual(p.header, pr.header)
self.assertEqual(p.content, pr.content)
self.assertEqual(p.cmdcode, pr.cmdcode)
self.assertEqual(p.request_id, pr.request_id)
print(pr)
p = BoltRequest(SofaHeader(a='1', b='2'), b"jklmnhi", ptype=PTYPE.ONEWAY, request_id=0,
cmdcode=CMDCODE.HEARTBEAT,
timeout=-1)
print(p)
s = p.to_stream()
pr = BoltRequest.from_stream(s)
self.assertNotEqual(id(p), id(pr))
self.assertEqual(p.header, pr.header)
self.assertEqual(p.content, pr.content)
self.assertEqual(p.cmdcode, pr.cmdcode)
self.assertEqual(p.request_id, pr.request_id)
print(pr)
def test_header(self):
h = SofaHeader(keya='key1', keyabcxcs='key2')
b = h.to_bytes()
self.assertEqual(len(h), len(b))
print(b)
h_recover = SofaHeader.from_bytes(b)
print(h_recover)
self.assertEqual(h, h_recover)