summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_pwd.py
blob: 2cb122beeea6c458900cc03387452390f780fab9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
from test_support import verbose
import pwd

print 'pwd.getpwall()'
entries = pwd.getpwall()

for e in entries:
    name = e[0]
    uid = e[2]
    if verbose:
        print name, uid
    print 'pwd.getpwuid()'
    dbuid = pwd.getpwuid(uid)
    if dbuid[0] != name:
        print 'Mismatch in pwd.getpwuid()'
    print 'pwd.getpwnam()'
    dbname = pwd.getpwnam(name)
    if dbname[2] != uid:
        print 'Mismatch in pwd.getpwnam()'
    else:
        print 'name matches uid'
    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)):
        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(map(None, chars))

try:
    pwd.getpwnam(fakename)
except KeyError:
    print 'caught expected exception'
else:
    print 'fakename', fakename, 'did not except pwd.getpwnam()'

# Choose a non-existent uid.
fakeuid = 4127
while byuids.has_key(fakeuid):
    fakeuid = (fakeuid * 3) % 0x10000

try:
    pwd.getpwuid(fakeuid)
except KeyError:
    print 'caught expected exception'
else:
    print 'fakeuid', fakeuid, 'did not except pwd.getpwuid()'