diff options
| -rwxr-xr-x | torturetest.py | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/torturetest.py b/torturetest.py new file mode 100755 index 00000000..a112d279 --- /dev/null +++ b/torturetest.py @@ -0,0 +1,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 + + + |