test_pwd.py 1.6 KB
Newer Older
1
from test_support import verbose
Barry Warsaw's avatar
Barry Warsaw committed
2 3
import pwd

4
print 'pwd.getpwall()'
Barry Warsaw's avatar
Barry Warsaw committed
5 6 7 8 9 10
entries = pwd.getpwall()

for e in entries:
    name = e[0]
    uid = e[2]
    if verbose:
11
        print name, uid
12
    print 'pwd.getpwuid()'
Barry Warsaw's avatar
Barry Warsaw committed
13
    dbuid = pwd.getpwuid(uid)
14
    if dbuid[0] != name:
15
        print 'Mismatch in pwd.getpwuid()'
16
    print 'pwd.getpwnam()'
Barry Warsaw's avatar
Barry Warsaw committed
17
    dbname = pwd.getpwnam(name)
18
    if dbname[2] != uid:
19
        print 'Mismatch in pwd.getpwnam()'
20
    else:
21
        print 'name matches uid'
Barry Warsaw's avatar
Barry Warsaw committed
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
    break

# try to get some errors
bynames = {}
byuids = {}
for n, p, u, g, gecos, d, s in entries:
    bynames[n] = u
    byuids[u] = n

allnames = bynames.keys()
namei = 0
fakename = allnames[namei]
while bynames.has_key(fakename):
    chars = map(None, fakename)
    for i in range(len(chars)):
37 38 39 40 41 42 43 44
        if chars[i] == 'z':
            chars[i] = 'A'
            break
        elif chars[i] == 'Z':
            continue
        else:
            chars[i] = chr(ord(chars[i]) + 1)
            break
Barry Warsaw's avatar
Barry Warsaw committed
45
    else:
46 47 48 49 50 51
        namei = namei + 1
        try:
            fakename = allnames[namei]
        except IndexError:
            # should never happen... if so, just forget it
            break
52
    fakename = ''.join(map(None, chars))
53

Barry Warsaw's avatar
Barry Warsaw committed
54 55 56
try:
    pwd.getpwnam(fakename)
except KeyError:
57
    print 'caught expected exception'
Barry Warsaw's avatar
Barry Warsaw committed
58 59 60
else:
    print 'fakename', fakename, 'did not except pwd.getpwnam()'

Jeremy Hylton's avatar
Jeremy Hylton committed
61
# Choose a non-existent uid.
62 63
fakeuid = 4127
while byuids.has_key(fakeuid):
64
    fakeuid = (fakeuid * 3) % 0x10000
Barry Warsaw's avatar
Barry Warsaw committed
65 66 67 68

try:
    pwd.getpwuid(fakeuid)
except KeyError:
69
    print 'caught expected exception'
Barry Warsaw's avatar
Barry Warsaw committed
70 71
else:
    print 'fakeuid', fakeuid, 'did not except pwd.getpwuid()'