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

Author: sreev

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
 

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