78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
import xdrlib
|
||
|
|
|
||
|
|
class XDRTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_xdr(self):
|
||
|
|
p = xdrlib.Packer()
|
||
|
|
|
||
|
|
s = b'hello world'
|
||
|
|
a = [b'what', b'is', b'hapnin', b'doctor']
|
||
|
|
|
||
|
|
p.pack_int(42)
|
||
|
|
p.pack_int(-17)
|
||
|
|
p.pack_uint(9)
|
||
|
|
p.pack_bool(True)
|
||
|
|
p.pack_bool(False)
|
||
|
|
p.pack_uhyper(45)
|
||
|
|
p.pack_float(1.9)
|
||
|
|
p.pack_double(1.9)
|
||
|
|
p.pack_string(s)
|
||
|
|
p.pack_list(range(5), p.pack_uint)
|
||
|
|
p.pack_array(a, p.pack_string)
|
||
|
|
|
||
|
|
# now verify
|
||
|
|
data = p.get_buffer()
|
||
|
|
up = xdrlib.Unpacker(data)
|
||
|
|
|
||
|
|
self.assertEqual(up.get_position(), 0)
|
||
|
|
|
||
|
|
self.assertEqual(up.unpack_int(), 42)
|
||
|
|
self.assertEqual(up.unpack_int(), -17)
|
||
|
|
self.assertEqual(up.unpack_uint(), 9)
|
||
|
|
self.assertTrue(up.unpack_bool() is True)
|
||
|
|
|
||
|
|
# remember position
|
||
|
|
pos = up.get_position()
|
||
|
|
self.assertTrue(up.unpack_bool() is False)
|
||
|
|
|
||
|
|
# rewind and unpack again
|
||
|
|
up.set_position(pos)
|
||
|
|
self.assertTrue(up.unpack_bool() is False)
|
||
|
|
|
||
|
|
self.assertEqual(up.unpack_uhyper(), 45)
|
||
|
|
self.assertAlmostEqual(up.unpack_float(), 1.9)
|
||
|
|
self.assertAlmostEqual(up.unpack_double(), 1.9)
|
||
|
|
self.assertEqual(up.unpack_string(), s)
|
||
|
|
self.assertEqual(up.unpack_list(up.unpack_uint), list(range(5)))
|
||
|
|
self.assertEqual(up.unpack_array(up.unpack_string), a)
|
||
|
|
up.done()
|
||
|
|
self.assertRaises(EOFError, up.unpack_uint)
|
||
|
|
|
||
|
|
class ConversionErrorTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.packer = xdrlib.Packer()
|
||
|
|
|
||
|
|
def assertRaisesConversion(self, *args):
|
||
|
|
self.assertRaises(xdrlib.ConversionError, *args)
|
||
|
|
|
||
|
|
def test_pack_int(self):
|
||
|
|
self.assertRaisesConversion(self.packer.pack_int, 'string')
|
||
|
|
|
||
|
|
def test_pack_uint(self):
|
||
|
|
self.assertRaisesConversion(self.packer.pack_uint, 'string')
|
||
|
|
|
||
|
|
def test_float(self):
|
||
|
|
self.assertRaisesConversion(self.packer.pack_float, 'string')
|
||
|
|
|
||
|
|
def test_double(self):
|
||
|
|
self.assertRaisesConversion(self.packer.pack_double, 'string')
|
||
|
|
|
||
|
|
def test_uhyper(self):
|
||
|
|
self.assertRaisesConversion(self.packer.pack_uhyper, 'string')
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|