moving basic rest calls to requestshelper

Issue-ID: TEST-173
Change-Id: I76736b52a48510ce0a6f44607401b8e127a87476
Signed-off-by: DR695H <dr695h@att.com>

1 files changed, 46 insertions, 1 deletions

diff --git a/robotframework-onap/ONAPLibrary/RequestsHelper.py b/robotframework-onap/ONAPLibrary/RequestsHelper.py
index 3db6100..f818c54 100644
--- a/robotframework-onap/ONAPLibrary/RequestsHelper.py
+++ b/robotframework-onap/ONAPLibrary/RequestsHelper.py
@@ -13,15 +13,20 @@
 # limitations under the License.
 
 from ONAPLibrary.UUIDKeywords import UUIDKeywords
+from RequestsLibrary import RequestsLibrary
+from robot.api import logger
+import hashlib
+from ONAPLibrary.Base64Keywords import Base64Keywords
 
 
 class RequestsHelper(object):
-    """ non keywords methods related to Requests library """
+    """ non keyword methods related to Requests library """
 
     def __init__(self):
         super(RequestsHelper, self).__init__()
         self.uuid = UUIDKeywords()
         self.application_id = "robot-ete"
+        self.requests = RequestsLibrary()
 
     def create_headers(self, sdc_user_id=None, accept="application/json", content_type="application/json", md5=None):
         """Create the headers that are used by onap"""
@@ -37,3 +42,43 @@ class RequestsHelper(object):
         if not md5:
             headers["Content-MD5"] = md5
         return headers
+
+    def get_request(self, alias, endpoint, data_path, sdc_user=None, accept="application/json", auth=None):
+        """Runs a get request"""
+        logger.info("Creating session" + endpoint)
+        self.requests.create_session(alias, endpoint, auth=auth)
+        headers = self.create_headers(sdc_user_id=sdc_user, accept=accept)
+        resp = self.requests.get_request(alias, data_path, headers=headers)
+        logger.info("Received response from [" + alias + "]: " + resp.text)
+        return resp
+
+    def post_request(self, alias, endpoint, data_path, data, sdc_user=None, files=None, accept="application/json",
+                     content_type="application/json", auth=None):
+        """Runs a post request"""
+        logger.info("Creating session" + endpoint)
+        md5 = hashlib.md5()
+        md5.update(data)
+        md5checksum = Base64Keywords().base64_encode(md5.hexdigest())
+        self.requests.create_session(alias,  endpoint, auth=auth)
+        headers = self.create_headers(sdc_user_id=sdc_user, accept=accept, content_type=content_type, md5=md5checksum)
+        resp = self.requests.post_request(alias,  data_path, files=files, data=data, headers=headers)
+        logger.info("Received response from [" + alias + "]: " + resp.text)
+        return resp
+
+    def put_request(self, alias, endpoint, data_path, data, sdc_user=None, accept="application/json", auth=None):
+        """Runs a put request"""
+        logger.info("Creating session" + endpoint)
+        self.requests.create_session(alias,  endpoint, auth=auth)
+        headers = self.create_headers(sdc_user_id=sdc_user, accept=accept)
+        resp = self.requests.put_request(alias,  data_path, data=data, headers=headers)
+        logger.info("Received response from [" + alias + "]: " + resp.text)
+        return resp
+
+    def delete_request(self, alias, endpoint, data_path, data=None, sdc_user=None, accept="application/json", auth=None):
+        """Runs a delete request"""
+        logger.info("Creating session" + endpoint)
+        self.requests.create_session(alias, endpoint, auth=auth)
+        headers = self.create_headers(sdc_user_id=sdc_user, accept=accept)
+        resp = self.requests.delete_request(alias, data_path, data=data, headers=headers)
+        logger.info("Received response from [" + alias + "]: " + resp.text)
+        return resp