diff options
Diffstat (limited to 'lib/python2.7/test/test_pwd.py')
-rw-r--r-- | lib/python2.7/test/test_pwd.py | 109 |
1 files changed, 109 insertions, 0 deletions
diff --git a/lib/python2.7/test/test_pwd.py b/lib/python2.7/test/test_pwd.py new file mode 100644 index 0000000..696161a --- /dev/null +++ b/lib/python2.7/test/test_pwd.py @@ -0,0 +1,109 @@ +import sys +import unittest +from test import test_support + +pwd = test_support.import_module('pwd') + +class PwdTest(unittest.TestCase): + + def test_values(self): + entries = pwd.getpwall() + entriesbyname = {} + entriesbyuid = {} + + for e in entries: + self.assertEqual(len(e), 7) + self.assertEqual(e[0], e.pw_name) + self.assertIsInstance(e.pw_name, basestring) + self.assertEqual(e[1], e.pw_passwd) + self.assertIsInstance(e.pw_passwd, basestring) + self.assertEqual(e[2], e.pw_uid) + self.assertIsInstance(e.pw_uid, (int, long)) + self.assertEqual(e[3], e.pw_gid) + self.assertIsInstance(e.pw_gid, (int, long)) + self.assertEqual(e[4], e.pw_gecos) + self.assertIsInstance(e.pw_gecos, basestring) + self.assertEqual(e[5], e.pw_dir) + self.assertIsInstance(e.pw_dir, basestring) + self.assertEqual(e[6], e.pw_shell) + self.assertIsInstance(e.pw_shell, basestring) + + # The following won't work, because of duplicate entries + # for one uid + # self.assertEqual(pwd.getpwuid(e.pw_uid), e) + # instead of this collect all entries for one uid + # and check afterwards + entriesbyname.setdefault(e.pw_name, []).append(e) + entriesbyuid.setdefault(e.pw_uid, []).append(e) + + if len(entries) > 1000: # Huge passwd file (NIS?) -- skip the rest + return + + # check whether the entry returned by getpwuid() + # for each uid is among those from getpwall() for this uid + for e in entries: + if not e[0] or e[0] == '+': + continue # skip NIS entries etc. + self.assertIn(pwd.getpwnam(e.pw_name), entriesbyname[e.pw_name]) + self.assertIn(pwd.getpwuid(e.pw_uid), entriesbyuid[e.pw_uid]) + + def test_errors(self): + self.assertRaises(TypeError, pwd.getpwuid) + self.assertRaises(TypeError, pwd.getpwuid, 3.14) + self.assertRaises(TypeError, pwd.getpwnam) + self.assertRaises(TypeError, pwd.getpwnam, 42) + self.assertRaises(TypeError, pwd.getpwall, 42) + + # try to get some errors + bynames = {} + byuids = {} + for (n, p, u, g, gecos, d, s) in pwd.getpwall(): + bynames[n] = u + byuids[u] = n + + allnames = bynames.keys() + namei = 0 + fakename = allnames[namei] + while fakename in bynames: + chars = list(fakename) + for i in xrange(len(chars)): + if chars[i] == 'z': + chars[i] = 'A' + break + elif chars[i] == 'Z': + continue + else: + chars[i] = chr(ord(chars[i]) + 1) + break + else: + namei = namei + 1 + try: + fakename = allnames[namei] + except IndexError: + # should never happen... if so, just forget it + break + fakename = ''.join(chars) + + self.assertRaises(KeyError, pwd.getpwnam, fakename) + + # In some cases, byuids isn't a complete list of all users in the + # system, so if we try to pick a value not in byuids (via a perturbing + # loop, say), pwd.getpwuid() might still be able to find data for that + # uid. Using sys.maxint may provoke the same problems, but hopefully + # it will be a more repeatable failure. + fakeuid = sys.maxint + self.assertNotIn(fakeuid, byuids) + self.assertRaises(KeyError, pwd.getpwuid, fakeuid) + + # -1 shouldn't be a valid uid because it has a special meaning in many + # uid-related functions + self.assertRaises(KeyError, pwd.getpwuid, -1) + # should be out of uid_t range + self.assertRaises(KeyError, pwd.getpwuid, 2**128) + self.assertRaises(KeyError, pwd.getpwuid, -2**128) + +def test_main(): + test_support.run_unittest(PwdTest) + +if __name__ == "__main__": + test_main() |