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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#!/usr/bin/env python
import sys, getopt, os, smtplib
class TestSite:
def __init__(self, line):
(self.host, self.userid, self.password, \
self.proto, self.options, self.version, self.comment) = \
line.strip().split(":")
def allattrs(self):
return (self.host, self.userid, self.password, \
self.proto, self.options, self.version, self.comment)
def __repr__(self):
return ":".join(self.allattrs())
def prettyprint(self):
"Prettyprint a site entry in human-readable form."
return "Host: %s\n" \
"Userid: %s\n" \
"Password: %s\n" \
"Protocol: %s\n" \
"Optiond: %s\n" \
"Version: %s\n" \
"Comment: %s\n" \
% self.allattrs()
def entryprint(self):
"Print a .fetchmailrc entry corresponding to a site entry."
return "poll %s-%s via %s with proto %s\n" \
" user %s there with password %s is esr here\n\n" \
% (self.host,self.proto,self.host,self.proto,self.userid,self.password)
def tableprint(self):
"Print an HTML server-type table entry."
return "<tr><td>%s %s</td><td>%s</td><td>%s</td></tr>\n" \
% (self.proto, self.version, self.options, self.comment)
if __name__ == "__main__":
# Start by reading in the sitelist
ifp = open("testsites")
sitelist = []
while 1:
line = ifp.readline()
if not line:
break
elif line[0] in ("#", "\n"):
continue
else:
sitelist.append(TestSite(line))
(options, arguments) = getopt.getopt(sys.argv[1:], "dft")
for (switch, value) in options:
if switch == "-d":
# Prettprint the sitelist
map(lambda x: sys.stdout.write(x.prettyprint() + "%%\n"), sitelist)
sys.exit(0)
elif switch == "-f":
# Dump the sitelist as a .fetchmailrc file
map(lambda x: sys.stdout.write(x.entryprint()), sitelist)
sys.exit(0)
elif switch == "-t":
# Dump the sitelist in HTML table form
map(lambda x: sys.stdout.write(x.tableprint()), sitelist)
sys.exit(0)
elif switch == "-g":
# Send test mail to each site
server = smtplib.SMTP("localhost")
fromaddr = "esr@thyrsus.com"
for site in sitelist:
toaddr = "%s@%s" % (site.userid, site.host)
msg = ("From: %s\r\nTo: %s\r\n\r\n" % (fromaddr, toaddr))
msg += "Test mail collected from %s.\n" % (toaddr,)
server.sendmail(fromaddr, toaddr, msg)
server.quit()
sys.stdout.write("Delaying to give the test mail time to land...")
time.sleep(5)
sys.stdout.write("here we go:\n")
# Fall through
# If no options, run the torture test
try:
failures = successes = 0
returns = []
for site in sitelist:
print "#\n#Testing %s %s at %s\n#" \
% (site.proto,site.version,site.host)
# Generate the control file for this run
temp = "/usr/tmp/torturestest-%d" % os.getpid()
ofp = open(temp, "w")
ofp.write(site.entryprint())
ofp.close()
# Run the test
status = os.system("fetchmail -d0 -v -f - <%s" % temp)
print "Status: %d" % status
returns.append((site, status))
if not os.WIFEXITED(status) or os.WEXITSTATUS(status) > 1:
failures += 1
else:
successes += 1
finally:
os.remove(temp)
# OK, summarize results
print "\n%d successes and %d failures out of %d tests" \
% (successes, failures, len(sitelist))
if failures:
print "Bad status was returned on the following sites:"
for (site, status) in returns:
if not os.WIFEXITED(status):
sys.stdout.write("%s %s at %s: " \
% (site.proto,site.version,site.host))
sys.stdout.write("abnormal termination\n")
elif os.WEXITSTATUS(status) > 1:
sys.stdout.write("%s %s at %s: " \
% (site.proto,site.version,site.host))
sys.stdout.write("%d\n" % os.WEXITSTATUS(status))
# end
|
