Add verify_ssl option

Author: JMoravec

reportportal_client/service.py

@@ -88,7 +88,7 @@ class ReportPortalService(object):
     """Service class with report portal event callbacks."""
 
     def __init__(self, endpoint, project, token, api_base="api/v1",
-                 is_skipped_an_issue=True):
+                 is_skipped_an_issue=True, verify_ssl=True):
         """Init the service class.
 
         Args:
@@ -98,6 +98,7 @@ def __init__(self, endpoint, project, token, api_base="api/v1",
             api_base: defaults to api/v1, can be changed to other version.
             is_skipped_an_issue: option to mark skipped tests as not
                 'To Investigate' items on Server side.
+            verify_ssl: option to not verify ssl certificates
         """
         super(ReportPortalService, self).__init__()
         self.endpoint = endpoint
@@ -113,6 +114,8 @@ def __init__(self, endpoint, project, token, api_base="api/v1",
         self.session.headers["Authorization"] = "bearer {0}".format(self.token)
         self.stack = [None]
         self.launch_id = None
+        self.verify_ssl = verify_ssl
+
 
     def terminate(self):
         pass
@@ -127,7 +130,7 @@ def start_launch(self, name, start_time, description=None, tags=None,
             "mode": mode
         }
         url = uri_join(self.base_url, "launch")
-        r = self.session.post(url=url, json=data)
+        r = self.session.post(url=url, json=data, verify=self.verify_ssl)
         self.launch_id = _get_id(r)
         self.stack.append(None)
         logger.debug("start_launch - Stack: %s", self.stack)
@@ -139,7 +142,7 @@ def _finalize_launch(self, end_time, action, status):
             "status": status
         }
         url = uri_join(self.base_url, "launch", self.launch_id, action)
-        r = self.session.put(url=url, json=data)
+        r = self.session.put(url=url, json=data, verify=self.verify_ssl)
         self.stack.pop()
         logger.debug("%s_launch - Stack: %s", action, self.stack)
         return _get_msg(r)
@@ -184,7 +187,7 @@ def start_test_item(self, name, start_time, item_type, description=None,
             url = uri_join(self.base_url, "item", parent_item_id)
         else:
             url = uri_join(self.base_url, "item")
-        r = self.session.post(url=url, json=data)
+        r = self.session.post(url=url, json=data, verify=self.verify_ssl)
 
         item_id = _get_id(r)
         self.stack.append(item_id)
@@ -204,7 +207,7 @@ def finish_test_item(self, end_time, status, issue=None):
         }
         item_id = self.stack.pop()
         url = uri_join(self.base_url, "item", item_id)
-        r = self.session.put(url=url, json=data)
+        r = self.session.put(url=url, json=data, verify=self.verify_ssl)
         logger.debug("finish_test_item - Stack: %s", self.stack)
         return _get_msg(r)
 
@@ -220,7 +223,7 @@ def log(self, time, message, level=None, attachment=None):
             return self.log_batch([data])
         else:
             url = uri_join(self.base_url, "log")
-            r = self.session.post(url=url, json=data)
+            r = self.session.post(url=url, json=data, verify=self.verify_ssl)
             logger.debug("log - Stack: %s", self.stack)
             return _get_id(r)
 
@@ -268,7 +271,7 @@ def log_batch(self, log_data):
             )
         )]
         files.extend(attachments)
-        r = self.session.post(url=url, files=files)
+        r = self.session.post(url=url, files=files, verify=self.verify_ssl)
         logger.debug("log_batch - Stack: %s", self.stack)
         logger.debug("log_batch response: %s", r.text)
 

reportportal_client/service_async.py

@@ -118,7 +118,8 @@ class ReportPortalServiceAsync(object):
 
     def __init__(self, endpoint, project, token, api_base="api/v1",
                  error_handler=None, log_batch_size=20,
-                 is_skipped_an_issue=True):
+                 is_skipped_an_issue=True,
+                 verify_ssl=True):
         """Init the service class.
 
         Args:
@@ -130,12 +131,13 @@ def __init__(self, endpoint, project, token, api_base="api/v1",
                 during items processing (in thread)
             is_skipped_an_issue: option to mark skipped tests as not
                 'To Investigate' items on Server side.
+            verify_ssl: option to not verify ssl certificates
         """
         super(ReportPortalServiceAsync, self).__init__()
         self.error_handler = error_handler
         self.log_batch_size = log_batch_size
         self.rp_client = ReportPortalService(
-            endpoint, project, token, api_base, is_skipped_an_issue)
+            endpoint, project, token, api_base, is_skipped_an_issue, verify_ssl)
         self.log_batch = []
         self.supported_methods = ["start_launch", "finish_launch",
                                   "start_test_item", "finish_test_item", "log"]