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)