Initial commit

Author: tkachoff

.gitignore

@@ -0,0 +1,2 @@
+.idea/
+.DS_Store

LICENSE.txt

@@ -0,0 +1,36 @@
+class ErrorRS():
+    pass
+
+
+class ReportPortalException(Exception):
+    def __init__(self, status_code, status_message, error_content, **kwargs):
+        super(ReportPortalException, self).__init__(**kwargs)
+        self.status_code = status_code
+        self.status_message = status_message
+        self.error_content = error_content
+
+    @property
+    def message(self):
+        return "Report Portal returned error\n" \
+               "Status code: {0}\n" \
+               "Status message: {1}\n" \
+               "Error Message: {2}\n" \
+               "Error Type: {3}\n" \
+               "Stack Trace: ".format(self.status_code,
+                                      self.status_message,
+                                      self.error_content["message"],
+                                      self.error_content["type"])
+
+
+class ReportPortalClientException(ReportPortalException):
+    serialVersionUID = -3747137063782963453L
+
+    def __init__(self, *args, **kwargs):
+        super(ReportPortalClientException, self).__init__(*args, **kwargs)
+
+
+class ReportPortalServerException(ReportPortalException):
+    serialVersionUID = -3767279627075261149L
+
+    def __init__(self, *args, **kwargs):
+        super(ReportPortalServerException, self).__init__(*args, **kwargs)

README.md

@@ -0,0 +1,86 @@
+import json
+
+
+class RQ(object):
+    def __init__(self):
+        super(RQ, self).__init__()
+
+    def as_dict(self):
+        res = {}
+        for k, v in self.__dict__.items():
+            if v:
+                res[k] = v
+        return res
+
+    @property
+    def data(self):
+        return json.dumps(self.as_dict())
+
+
+class StartRQ(RQ):
+    def __init__(self, name=None, description=None, tags=None,
+                 start_time=None):
+        super(StartRQ, self).__init__()
+        self.name = name
+        self.description = description
+        self.tags = tags
+        self.start_time = start_time
+
+
+class StartLaunchRQ(StartRQ):
+    def __init__(self, name=None, description=None, tags=None, start_time=None,
+                 mode=None):
+        super(StartLaunchRQ, self).__init__(name=name, description=description,
+                                            tags=tags, start_time=start_time)
+        self.mode = mode
+
+
+class FinishExecutionRQ(RQ):
+    def __init__(self, end_time=None, status=None):
+        super(FinishExecutionRQ, self).__init__()
+        self.end_time = end_time
+        self.status = status
+
+
+class StartTestItemRQ(StartRQ):
+    def __init__(self, name=None, description=None, tags=None,
+                 start_time=None, launch_id=None, type=None):
+        """
+        type can be (SUITE, STORY, TEST, SCENARIO, STEP, BEFORE_CLASS,
+        BEFORE_GROUPS, BEFORE_METHOD, BEFORE_SUITE, BEFORE_TEST, AFTER_CLASS,
+        AFTER_GROUPS, AFTER_METHOD, AFTER_SUITE, AFTER_TEST)
+        :param launch_id:
+        :param type:
+        """
+        super(StartTestItemRQ, self).__init__(name=name,
+                                              description=description,
+                                              tags=tags, start_time=start_time)
+        self.launch_id = launch_id
+        self.type = type
+
+
+class FinishTestItemRQ(FinishExecutionRQ):
+    def __init__(self, end_time=None, status=None, issue=None):
+        super(FinishTestItemRQ, self).__init__(end_time=end_time,
+                                               status=status)
+        self.issue = issue
+
+
+class File(RQ):
+    def __init__(self, name, content):
+        super(File, self).__init__()
+        self.name = name
+        self.content = content
+
+
+class SaveLogRQ(RQ):
+    def __init__(self, item_id=None, time=None, message=None, level=None,
+                 file_obj=None):
+        super(SaveLogRQ, self).__init__()
+        self.item_id = item_id
+        self.time = time
+        self.message = message
+        self.level = level
+        self.file = None
+        if file_obj is not None:
+            self.file = file_obj.data

reportportal-client/__init__.py

@@ -0,0 +1,24 @@
+import json
+
+
+class EntryCreatedRS(object):
+    def __init__(self, id=None, raw=None):
+        super(EntryCreatedRS, self).__init__()
+        self.id = id
+        self.raw = raw
+        if raw is not None:
+            self.id = json.loads(raw)["id"]
+
+    def as_dict(self):
+        return {"id": self.id}
+
+
+class OperationCompletionRS(object):
+    def __init__(self, msg=None, raw=None):
+        super(OperationCompletionRS, self).__init__()
+        self.msg = msg
+        if raw is not None:
+            self.msg = json.loads(raw)["msg"]
+
+    def as_dict(self):
+        return {"msg": self.msg}

reportportal-client/service/__init__.py

@@ -0,0 +1,64 @@
+import os
+
+import requests
+
+from .model.response import EntryCreatedRS, OperationCompletionRS
+
+
+def print_request(r):
+    print("\n{0}: {1}\nRequest Body: {2}\nResponse Content: {3}\n".
+          format(r.request.method, r.request.url, r.request.body, r.content))
+
+
+class ReportPortalService(object):
+    def __init__(self, endpoint, project, token, api_base=None):
+        super(ReportPortalService, self).__init__()
+        self.endpoint = endpoint
+        if api_base is None:
+            self.api_base = "api/v1"
+        self.project = project
+        self.token = token
+        self.base_url = os.path.join(self.endpoint,
+                                     self.api_base,
+                                     self.project)
+        self.headers = {"Content-Type": "application/json",
+                        "Authorization": "{0} {1}".format("bearer",
+                                                          self.token)}
+
+    def start_launch(self, start_launch_rq):
+        url = os.path.join(self.base_url, "launch")
+        r = requests.post(url=url, headers=self.headers,
+                          data=start_launch_rq.data)
+        # print_request(r)
+        return EntryCreatedRS(raw=r.text)
+
+    def finish_launch(self, launch_id, finish_execution_rq):
+        url = os.path.join(self.base_url, "launch", launch_id, "finish")
+        r = requests.put(url=url, headers=self.headers,
+                         data=finish_execution_rq.data)
+        # print_request(r)
+        return OperationCompletionRS(raw=r.text)
+
+    def start_test_item(self, parent_item_id, start_test_item_rq):
+        if parent_item_id is not None:
+            url = os.path.join(self.base_url, "item", parent_item_id)
+        else:
+            url = os.path.join(self.base_url, "item")
+        r = requests.post(url=url, headers=self.headers,
+                          data=start_test_item_rq.data)
+        # print_request(r)
+        return EntryCreatedRS(raw=r.text)
+
+    def finish_test_item(self, item_id, finish_test_item_rq):
+        url = os.path.join(self.base_url, "item", item_id)
+        r = requests.put(url=url, headers=self.headers,
+                         data=finish_test_item_rq.data)
+        # print_request(r)
+        return OperationCompletionRS(raw=r.text)
+
+    def log(self, save_log_rq):
+        url = os.path.join(self.base_url, "log")
+        r = requests.post(url=url, headers=self.headers,
+                          data=save_log_rq.data)
+        # print_request(r)
+        return EntryCreatedRS(raw=r.text)