/Lib/test/test_pwd.py

http://unladen-swallow.googlecode.com/ · Python · 100 lines · 72 code · 14 blank · 14 comment · 15 complexity · eb041dabd6fe401da62971ef795686be MD5 · raw file

  1. import sys
  2. import unittest
  3. from test import test_support
  4. import pwd
  5. class PwdTest(unittest.TestCase):
  6. def test_values(self):
  7. entries = pwd.getpwall()
  8. entriesbyname = {}
  9. entriesbyuid = {}
  10. for e in entries:
  11. self.assertEqual(len(e), 7)
  12. self.assertEqual(e[0], e.pw_name)
  13. self.assert_(isinstance(e.pw_name, basestring))
  14. self.assertEqual(e[1], e.pw_passwd)
  15. self.assert_(isinstance(e.pw_passwd, basestring))
  16. self.assertEqual(e[2], e.pw_uid)
  17. self.assert_(isinstance(e.pw_uid, int))
  18. self.assertEqual(e[3], e.pw_gid)
  19. self.assert_(isinstance(e.pw_gid, int))
  20. self.assertEqual(e[4], e.pw_gecos)
  21. self.assert_(isinstance(e.pw_gecos, basestring))
  22. self.assertEqual(e[5], e.pw_dir)
  23. self.assert_(isinstance(e.pw_dir, basestring))
  24. self.assertEqual(e[6], e.pw_shell)
  25. self.assert_(isinstance(e.pw_shell, basestring))
  26. # The following won't work, because of duplicate entries
  27. # for one uid
  28. # self.assertEqual(pwd.getpwuid(e.pw_uid), e)
  29. # instead of this collect all entries for one uid
  30. # and check afterwards
  31. entriesbyname.setdefault(e.pw_name, []).append(e)
  32. entriesbyuid.setdefault(e.pw_uid, []).append(e)
  33. if len(entries) > 1000: # Huge passwd file (NIS?) -- skip the rest
  34. return
  35. # check whether the entry returned by getpwuid()
  36. # for each uid is among those from getpwall() for this uid
  37. for e in entries:
  38. if not e[0] or e[0] == '+':
  39. continue # skip NIS entries etc.
  40. self.assert_(pwd.getpwnam(e.pw_name) in entriesbyname[e.pw_name])
  41. self.assert_(pwd.getpwuid(e.pw_uid) in entriesbyuid[e.pw_uid])
  42. def test_errors(self):
  43. self.assertRaises(TypeError, pwd.getpwuid)
  44. self.assertRaises(TypeError, pwd.getpwnam)
  45. self.assertRaises(TypeError, pwd.getpwall, 42)
  46. # try to get some errors
  47. bynames = {}
  48. byuids = {}
  49. for (n, p, u, g, gecos, d, s) in pwd.getpwall():
  50. bynames[n] = u
  51. byuids[u] = n
  52. allnames = bynames.keys()
  53. namei = 0
  54. fakename = allnames[namei]
  55. while fakename in bynames:
  56. chars = list(fakename)
  57. for i in xrange(len(chars)):
  58. if chars[i] == 'z':
  59. chars[i] = 'A'
  60. break
  61. elif chars[i] == 'Z':
  62. continue
  63. else:
  64. chars[i] = chr(ord(chars[i]) + 1)
  65. break
  66. else:
  67. namei = namei + 1
  68. try:
  69. fakename = allnames[namei]
  70. except IndexError:
  71. # should never happen... if so, just forget it
  72. break
  73. fakename = ''.join(chars)
  74. self.assertRaises(KeyError, pwd.getpwnam, fakename)
  75. # In some cases, byuids isn't a complete list of all users in the
  76. # system, so if we try to pick a value not in byuids (via a perturbing
  77. # loop, say), pwd.getpwuid() might still be able to find data for that
  78. # uid. Using sys.maxint may provoke the same problems, but hopefully
  79. # it will be a more repeatable failure.
  80. fakeuid = sys.maxint
  81. self.failIf(fakeuid in byuids)
  82. self.assertRaises(KeyError, pwd.getpwuid, fakeuid)
  83. def test_main():
  84. test_support.run_unittest(PwdTest)
  85. if __name__ == "__main__":
  86. test_main()