mirror of
https://github.com/python/cpython.git
synced 2024-12-12 11:23:56 +08:00
a21e4ca3c6
........ r70734 | r.david.murray | 2009-03-30 15:04:00 -0400 (Mon, 30 Mar 2009) | 7 lines Add import_function method to test.test_support, and modify a number of tests that expect to be skipped if imports fail or functions don't exist to use import_function and import_module. The ultimate goal is to change regrtest to not skip automatically on ImportError. Checking in now to make sure the buldbots don't show any errors on platforms I can't direct test on. ........ r70775 | r.david.murray | 2009-03-30 19:05:48 -0400 (Mon, 30 Mar 2009) | 4 lines Change more tests to use import_module for the modules that should cause tests to be skipped. Also rename import_function to the more descriptive get_attribute and add a docstring. ........ r70856 | r.david.murray | 2009-03-31 14:32:17 -0400 (Tue, 31 Mar 2009) | 7 lines A few more test skips via import_module, and change import_module to return the error message produced by importlib, so that if an import in the package whose import is being wrapped is what failed the skip message will contain the name of that module instead of the name of the wrapped module. Also fixed formatting of some previous comments. ........ r70874 | r.david.murray | 2009-03-31 15:33:15 -0400 (Tue, 31 Mar 2009) | 5 lines Improve test_support.import_module docstring, remove deprecated flag from get_attribute since it isn't likely to do anything useful. ........ r70876 | r.david.murray | 2009-03-31 15:49:15 -0400 (Tue, 31 Mar 2009) | 4 lines Remove the regrtest check that turns any ImportError into a skipped test. Hopefully all modules whose imports legitimately result in a skipped test have been properly wrapped by the previous commits. ........ r70877 | r.david.murray | 2009-03-31 15:57:24 -0400 (Tue, 31 Mar 2009) | 2 lines Add NEWS entry for regrtest change. ........
98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
import unittest
|
|
from test import support
|
|
|
|
pwd = 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.assert_(isinstance(e.pw_name, str))
|
|
self.assertEqual(e[1], e.pw_passwd)
|
|
self.assert_(isinstance(e.pw_passwd, str))
|
|
self.assertEqual(e[2], e.pw_uid)
|
|
self.assert_(isinstance(e.pw_uid, int))
|
|
self.assertEqual(e[3], e.pw_gid)
|
|
self.assert_(isinstance(e.pw_gid, int))
|
|
self.assertEqual(e[4], e.pw_gecos)
|
|
self.assert_(isinstance(e.pw_gecos, str))
|
|
self.assertEqual(e[5], e.pw_dir)
|
|
self.assert_(isinstance(e.pw_dir, str))
|
|
self.assertEqual(e[6], e.pw_shell)
|
|
self.assert_(isinstance(e.pw_shell, str))
|
|
|
|
# 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.assert_(pwd.getpwnam(e.pw_name) in entriesbyname[e.pw_name])
|
|
self.assert_(pwd.getpwuid(e.pw_uid) in entriesbyuid[e.pw_uid])
|
|
|
|
def test_errors(self):
|
|
self.assertRaises(TypeError, pwd.getpwuid)
|
|
self.assertRaises(TypeError, pwd.getpwnam)
|
|
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 = list(bynames.keys())
|
|
namei = 0
|
|
fakename = allnames[namei]
|
|
while fakename in bynames:
|
|
chars = list(fakename)
|
|
for i in range(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)
|
|
|
|
# Choose a non-existent uid.
|
|
fakeuid = 4127
|
|
while fakeuid in byuids:
|
|
fakeuid = (fakeuid * 3) % 0x10000
|
|
|
|
self.assertRaises(KeyError, pwd.getpwuid, fakeuid)
|
|
|
|
def test_main():
|
|
support.run_unittest(PwdTest)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|