Home | History | Annotate | Download | only in test 
      1 import unittest
      2 
      3 import xdrlib
      4 
      5 class XDRTest(unittest.TestCase):
      6 
      7     def test_xdr(self):
      8         p = xdrlib.Packer()
      9 
     10         s = b'hello world'
     11         a = [b'what', b'is', b'hapnin', b'doctor']
     12 
     13         p.pack_int(42)
     14         p.pack_int(-17)
     15         p.pack_uint(9)
     16         p.pack_bool(True)
     17         p.pack_bool(False)
     18         p.pack_uhyper(45)
     19         p.pack_float(1.9)
     20         p.pack_double(1.9)
     21         p.pack_string(s)
     22         p.pack_list(range(5), p.pack_uint)
     23         p.pack_array(a, p.pack_string)
     24 
     25         # now verify
     26         data = p.get_buffer()
     27         up = xdrlib.Unpacker(data)
     28 
     29         self.assertEqual(up.get_position(), 0)
     30 
     31         self.assertEqual(up.unpack_int(), 42)
     32         self.assertEqual(up.unpack_int(), -17)
     33         self.assertEqual(up.unpack_uint(), 9)
     34         self.assertTrue(up.unpack_bool() is True)
     35 
     36         # remember position
     37         pos = up.get_position()
     38         self.assertTrue(up.unpack_bool() is False)
     39 
     40         # rewind and unpack again
     41         up.set_position(pos)
     42         self.assertTrue(up.unpack_bool() is False)
     43 
     44         self.assertEqual(up.unpack_uhyper(), 45)
     45         self.assertAlmostEqual(up.unpack_float(), 1.9)
     46         self.assertAlmostEqual(up.unpack_double(), 1.9)
     47         self.assertEqual(up.unpack_string(), s)
     48         self.assertEqual(up.unpack_list(up.unpack_uint), list(range(5)))
     49         self.assertEqual(up.unpack_array(up.unpack_string), a)
     50         up.done()
     51         self.assertRaises(EOFError, up.unpack_uint)
     52 
     53 class ConversionErrorTest(unittest.TestCase):
     54 
     55     def setUp(self):
     56         self.packer = xdrlib.Packer()
     57 
     58     def assertRaisesConversion(self, *args):
     59         self.assertRaises(xdrlib.ConversionError, *args)
     60 
     61     def test_pack_int(self):
     62         self.assertRaisesConversion(self.packer.pack_int, 'string')
     63 
     64     def test_pack_uint(self):
     65         self.assertRaisesConversion(self.packer.pack_uint, 'string')
     66 
     67     def test_float(self):
     68         self.assertRaisesConversion(self.packer.pack_float, 'string')
     69 
     70     def test_double(self):
     71         self.assertRaisesConversion(self.packer.pack_double, 'string')
     72 
     73     def test_uhyper(self):
     74         self.assertRaisesConversion(self.packer.pack_uhyper, 'string')
     75 
     76 if __name__ == "__main__":
     77     unittest.main()
     78