def _write_packet_header(self, length, seq, fde): """ Write out packet header with given length and sequence id to file-like fde """ length_field = FixedLengthInteger(3, length) seq_field = FixedLengthInteger(1, seq) length_field.write_out(fde, label=None) seq_field.write_out(fde, label=None)
def runTest(self): """ Test fixed-length integer read-in """ from StringIO import StringIO from mysqlproxy.types import FixedLengthInteger # wire read tests proto_buf = StringIO(b'\x01\x00\x00\x25') fli = FixedLengthInteger(3) fli.read_in(proto_buf) self.assertEqual(fli.val, 1) # wire write tests proto_buf = StringIO() fli = FixedLengthInteger(3, 1) fli.write_out(proto_buf) self.assertEqual(proto_buf.getvalue(), b'\x01\x00\x00')