def test_data(self): # Zero-length payload self.assertRaises(WireProtocolError, DATADatagram.from_wire, '') # One byte payload self.assertRaises(WireProtocolError, DATADatagram.from_wire, '\x00') # Zero-length data self.assertEqual(DATADatagram.from_wire('\x00\x01').to_wire(), '\x00\x03\x00\x01') # Full-length data self.assertEqual(DATADatagram.from_wire('\x00\x01foobar').to_wire(), '\x00\x03\x00\x01foobar')
def test_data(self): # Zero-length payload self.assertRaises(WireProtocolError, DATADatagram.from_wire, b'') # One byte payload self.assertRaises(WireProtocolError, DATADatagram.from_wire, b'\x00') # Zero-length data self.assertEqual( DATADatagram.from_wire(b'\x00\x01').to_wire(), b'\x00\x03\x00\x01') # Full-length data self.assertEqual( DATADatagram.from_wire(b'\x00\x01foobar').to_wire(), b'\x00\x03\x00\x01foobar')