add run_id; update client and client_wo; add test (#111)

* add run_id; update client and client_wo; add test Signed-off-by: SreeV <[email protected]> * revert test wo client Signed-off-by: SreeV <[email protected]>
MarquezProject · Oct 13, 2020 · c1fccd6 · c1fccd6
1 parent c60845f
commit c1fccd6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/marquez_client/client.py b/marquez_client/client.py
@@ -192,8 +192,8 @@ def tag_dataset_field(self, namespace_name, dataset_name, field_name,
 
     # Job API
     def create_job(self, namespace_name, job_name, job_type, location=None,
-                   input_dataset=None,
-                   output_dataset=None, description=None, context=None):
+                   input_dataset=None, output_dataset=None, description=None,
+                   context=None, run_id=None):
         Utils.check_name_length(namespace_name, 'namespace_name')
         Utils.check_name_length(job_name, 'job_name')
         Utils.is_instance_of(job_type, JobType)
@@ -204,6 +204,9 @@ def create_job(self, namespace_name, job_name, job_type, location=None,
             'type': job_type.name
         }
 
+        if run_id:
+            payload['runId'] = run_id
+
         if context:
             payload['context'] = context
 

diff --git a/marquez_client/client_wo.py b/marquez_client/client_wo.py
@@ -115,7 +115,8 @@ def create_dataset(self, namespace_name, dataset_name, dataset_type,
     # Job API
     def create_job(self, namespace_name, job_name, job_type,
                    location=None, input_dataset=None,
-                   output_dataset=None, description=None, context=None):
+                   output_dataset=None, description=None, context=None,
+                   run_id=None):
         Utils.check_name_length(namespace_name, 'namespace_name')
         Utils.check_name_length(job_name, 'job_name')
         Utils.is_instance_of(job_type, JobType)
@@ -126,6 +127,9 @@ def create_job(self, namespace_name, job_name, job_type,
             'type': job_type.name
         }
 
+        if run_id:
+            payload['runId'] = run_id
+
         if context:
             payload['context'] = context