Initial revision

svn path=/trunk/; revision=3084

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
+
+
+