add jobs support

shakedown/__init__.py

@@ -13,5 +13,6 @@
 from shakedown.dcos.zookeeper import *
 from shakedown.dcos.agent import *
 from shakedown.dcos.master import *
+from shakedown.dcos.jobs import *
 
 VERSION='1.1.10'

shakedown/dcos/jobs.py

@@ -0,0 +1,54 @@
+import spinner
+import time
+import urllib
+from dcos import config, errors, http
+
+
+def post_job(job):
+    return http.post(_metronome_url('v1/jobs'), json=job)
+
+
+def get_job(job_id, history=True):
+    url = 'v1/jobs/{}'.format(job_id)
+    if history:
+        url = url + '?embed=history'
+    return http.get(_metronome_url(url))
+
+
+def post_run(job_id):
+    return http.post(_metronome_url('v1/jobs/{}/runs').format(job_id))
+
+
+def get_run(job_id, run_id):
+    return http.get(_metronome_url('v1/jobs/{}/runs/{}').format(job_id, run_id))
+
+
+def run_job(job_spec, wait=True, assertSuccess=True):
+    job_id = post_job(job_spec).json()['id']
+    run_id = post_run(job_id).json()['id']
+    if wait:
+        wait_for_run(job_id, run_id)
+    if assertSuccess:
+        job = get_job(job_id).json()
+        assert(len(job['history']['successfulFinishedRuns']) > 0)
+    return job_id
+
+
+def wait_for_run(job_id, run_id, duration=5):
+    def run_finished():
+        try:
+            response = get_run(job_id, run_id)
+        except errors.DCOSHTTPException as ex:
+            if ex.response.status_code == 404:
+                return True
+            else:
+                raise ex
+        return False
+    spinner.wait_for(run_finished, ignore_exceptions=False)
+
+
+def _metronome_url(url):
+    dcos_url = config.get_config_val('core.dcos_url')
+    return urllib.parse.urljoin(
+        urllib.parse.urljoin(dcos_url, 'service/metronome/'),
+        url)

shakedown/dcos/spinner.py

@@ -2,10 +2,9 @@
 
 
 def wait_for(predicate, timeout_seconds=120, sleep_seconds=1, ignore_exceptions=True, inverse_predicate=False):
-    """ waits or spins for a predicate.  Predicate is in function that returns a True or False.
-        An exception in the function will be returned.
-        A timeout will throw a TimeoutExpired Exception.
-
+    """ Waits for predicate to return True.
+        If predicate raises an exception and ignore_exceptions is true, the exception will be ignored.
+        A timeout will raise a TimeoutExpired Exception.
     """
 
     timeout = Deadline.create_deadline(timeout_seconds)