[rhmessaging-commits] rhmessaging commits: r4046 - mgmt/newdata/cumin/bin.

[rhmessaging-commits at lists.jboss.org]

Author: justi9
Date: 2010-06-21 17:51:13 -0400 (Mon, 21 Jun 2010)
New Revision: 4046

Modified:
   mgmt/newdata/cumin/bin/cumin-smoke-test
Log:
Make cumin-smoke-test an entry point for multiple, individually selectable smoke tests; add a submit smoke test

Modified: mgmt/newdata/cumin/bin/cumin-smoke-test
===================================================================
--- mgmt/newdata/cumin/bin/cumin-smoke-test	2010-06-21 19:15:25 UTC (rev 4045)
+++ mgmt/newdata/cumin/bin/cumin-smoke-test	2010-06-21 21:51:13 UTC (rev 4046)
@@ -44,28 +44,70 @@
         conn = cumin.database.get_connection()
         cursor = conn.cursor()
 
-        # cls = cumin.model.org_apache_qpid_broker.Broker XXX fails
-        cls = cumin.model.com_redhat_grid.Submission
-        obj = cls.get_object(cursor)
+        def invoke_test(test):
+            print "[%s]" % name[6:]
 
-        print "Calling method on", obj
+            try:
+                test(cumin, cursor)
+            except:
+                print "FAILES!"
 
-        summs = cumin.model.get_submission_job_summaries(obj)
+        try:
+            chosen = sys.argv[1]
+        except IndexError:
+            chosen = None
 
-        pprint(summs)
+        attrs = globals()
 
-        #completed = Event()
-        
-        #def completion(x, y):
-        #    print x, y
-        #    completed.set()
+        if chosen:
+            name = "_test_%s" % chosen
 
-        #cumin.session.call_method(completion, obj, "GetJobSummaries", ())
+            assert name in attrs
 
-        #completed.wait()
+            invoke_test(attrs[name])
+        else:
+            for name in [x for x in attrs if x.startswith("_test")]:
+                invoke_test(attrs[name])
     finally:
         cumin.stop()
 
+def _test_submit(cumin, cursor):
+    cls = cumin.model.com_redhat_grid.Scheduler
+    obj = cls.get_object(cursor)
+
+    print "Submitting to", obj, 
+
+    ad = {"Cmd": "/bin/sleep",
+          "Args": "5m",
+          "JobUniverse": 5,
+          "Requirements": "TRUE",
+          "Iwd": "/tmp",
+          "Owner": "gridmonkey@%s" % obj.Machine,
+          "!!descriptors": {"Requirements": "com.redhat.grid.Expression"}}
+
+    completed = Event()
+
+    def completion(x, y):
+        print "-> [%s, %s]" % (x, y)
+
+        completed.set()
+
+    cumin.session.call_method(completion, obj, "SubmitJob", (ad,))
+
+    completed.wait(30)
+
+def _test_job_summaries(cumin, cursor):
+    cls = cumin.model.com_redhat_grid.Submission
+    obj = cls.get_object(cursor)
+
+    print "Getting job summaries for", obj,
+
+    summs = cumin.model.get_submission_job_summaries(obj)
+
+    print "-> [%i job summaries]" % len(summs)
+
+    #pprint(summs)
+
 if __name__ == "__main__":
     try:
         main()