Added option to send email when experiments complete.

3 files changed, 37 insertions, 5 deletions

diff --git a/run/emailer.py b/run/emailer.py
new file mode 100644
index 0000000..6d66da2
--- /dev/null
+++ b/run/emailer.py
@@ -0,0 +1,28 @@
+import os
+import pwd
+import smtplib
+import socket
+
+class Emailer(object):
+    def __init__(self, target):
+        user = pwd.getpwuid(os.getuid())[0]
+        host = socket.gethostname()
+
+        self.sender = "%s@%s" % (user, host)
+        self.target = target
+
+        self.body = "\r\n".join(["From: %s" % self.sender,
+                                 "To: %s" % target,
+                                 "Subject: Test Completed!", "", "{}"])
+
+        self.mail = smtplib.SMTP("localhost")
+
+        # Hopefully crash if the server is not running
+        self.mail.ehlo()
+
+    def close(self):
+        self.mail.quit()
+
+    def send(self, text):
+        self.mail.sendmail(self.sender, [self.target],
+                           self.body.format(text))
diff --git a/run/experiment.py b/run/experiment.py
index 3dd4866..e5811f8 100644
--- a/run/experiment.py
+++ b/run/experiment.py
@@ -87,6 +87,10 @@ class Experiment(object):
     def __run_tasks(self):
         already_waiting = lu.waiting_tasks()
 
+        if already_waiting:
+            self.log("Already %d tasks waiting for release!")
+            self.log("Experiment will fail if any of these tasks are released.")
+
         self.log("Starting the programs")
         for e in self.executables:
             try:
@@ -115,7 +119,6 @@ class Experiment(object):
             # Need to re-release non-released tasks before we can kill them though
             self.log("Failed to release {} tasks! Re-releasing and killing".format(
                 len(self.executables) - released, len(self.executables)))
-
             time.sleep(5)
 
             released = lu.release_tasks()
@@ -160,7 +163,10 @@ class Experiment(object):
                 self.teardown()
         finally:
             self.log("Switching to Linux scheduler")
-            lu.switch_scheduler("Linux")
+            try:
+                lu.switch_scheduler("Linux")
+            except:
+                self.log("Failed to switch back to Linux.")
 
         if succ:
             self.__save_results()
diff --git a/run/litmus_util.py b/run/litmus_util.py
index b9080c1..4709b66 100644
--- a/run/litmus_util.py
+++ b/run/litmus_util.py
@@ -8,7 +8,6 @@ def switch_scheduler(switch_to_in):
 
     This methods sleeps for two seconds to give Linux the chance to execute
     schedule switching code. Raises an exception if the switch does not work.
-
     '''
 
     switch_to = str(switch_to_in).strip()
@@ -23,7 +22,7 @@ def switch_scheduler(switch_to_in):
         cur_plugin = active_plugin.read().strip()
 
     if switch_to != cur_plugin:
-        raise Exception("Could not switch to plugin: %s" % switch_to)
+        raise Exception("Could not switch to '%s' (check dmesg)"  % switch_to)
 
 def waiting_tasks():
     reg = re.compile(r'^ready.*?(?P<READY>\d+)$', re.M)
@@ -37,7 +36,6 @@ def waiting_tasks():
     return 0 if not ready else int(ready)
 
 def release_tasks():
-
     try:
         data = subprocess.check_output([conf.BINS['release']])
     except subprocess.CalledProcessError: