def test_basicUDP(self): u = UDP() u.dstport = 5500 u.srcport = 4400 u.payload = struct.pack('B', 0x5) mypacket = u.pack() self.assertEqual(mypacket, struct.pack('>HHHHB', 4400, 5500, 9, 0, 0x5))
def test_unpackUDPShort(self): u = UDP() dymmypayload = struct.pack('H', 0xa5) self.assertRaises(ValueError, lambda: u.unpack(dymmypayload))
def test_defaultUDP(self): u = UDP() self.assertRaises(ValueError, lambda: u.pack())