Lib/test/test_grp.py

   1 """Test script for the grp module."""
   2
   3 import grp
   4 import unittest
   5 from test import test_support
   6
   7 class GroupDatabaseTestCase(unittest.TestCase):
   8
   9     def check_value(self, value):
  10         # check that a grp tuple has the entries and
  11         # attributes promised by the docs
  12         self.assertEqual(len(value), 4)
  13         self.assertEqual(value[0], value.gr_name)
  14         self.assert_(isinstance(value.gr_name, basestring))
  15         self.assertEqual(value[1], value.gr_passwd)
  16         self.assert_(isinstance(value.gr_passwd, basestring))
  17         self.assertEqual(value[2], value.gr_gid)
  18         self.assert_(isinstance(value.gr_gid, int))
  19         self.assertEqual(value[3], value.gr_mem)
  20         self.assert_(isinstance(value.gr_mem, list))
  21
  22     def test_values(self):
  23         entries = grp.getgrall()
  24
  25         for e in entries:
  26             self.check_value(e)
  27
  28         for e in entries:
  29             e2 = grp.getgrgid(e.gr_gid)
  30             self.check_value(e2)
  31             self.assertEqual(e2.gr_gid, e.gr_gid)
  32             e2 = grp.getgrnam(e.gr_name)
  33             self.check_value(e2)
  34             self.assertEqual(e2.gr_name, e.gr_name)
  35
  36     def test_errors(self):
  37         self.assertRaises(TypeError, grp.getgrgid)
  38         self.assertRaises(TypeError, grp.getgrnam)
  39         self.assertRaises(TypeError, grp.getgrall, 42)
  40
  41         # try to get some errors
  42         bynames = {}
  43         bygids = {}
  44         for (n, p, g, mem) in grp.getgrall():
  45             if not n or n == '+':
  46                 continue # skip NIS entries etc.
  47             bynames[n] = g
  48             bygids[g] = n
  49
  50         allnames = bynames.keys()
  51         namei = 0
  52         fakename = allnames[namei]
  53         while fakename in bynames:
  54             chars = map(None, fakename)
  55             for i in xrange(len(chars)):
  56                 if chars[i] == 'z':
  57                     chars[i] = 'A'
  58                     break
  59                 elif chars[i] == 'Z':
  60                     continue
  61                 else:
  62                     chars[i] = chr(ord(chars[i]) + 1)
  63                     break
  64             else:
  65                 namei = namei + 1
  66                 try:
  67                     fakename = allnames[namei]
  68                 except IndexError:
  69                     # should never happen... if so, just forget it
  70                     break
  71             fakename = ''.join(map(None, chars))
  72
  73         self.assertRaises(KeyError, grp.getgrnam, fakename)
  74
  75         # Choose a non-existent gid.
  76         fakegid = 4127
  77         while fakegid in bygids:
  78             fakegid = (fakegid * 3) % 0x10000
  79
  80         self.assertRaises(KeyError, grp.getgrgid, fakegid)
  81
  82 def test_main():
  83     test_support.run_unittest(GroupDatabaseTestCase)
  84
  85 if __name__ == "__main__":
  86     test_main()