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')
