1 files changed, 62 insertions, 0 deletions
diff --git a/vnftest/common/rest_client.py b/vnftest/common/rest_client.py
new file mode 100644
index 0000000..23a108c
--- /dev/null
+++ b/vnftest/common/rest_client.py
@@ -0,0 +1,62 @@
+#!/usr/bin/env python
+##############################################################################
+# Copyright 2018 EuropeanSoftwareMarketingLtd.
+# ===================================================================
+#  Licensed under the ApacheLicense, Version2.0 (the"License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+# software distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and limitations under
+# the License
+##############################################################################
+
+import json
+import urllib2
+import requests
+
+
+def post(url, headers, data, logger):
+    return call(url, 'POST', headers, data, logger)
+
+
+def call(url, method, headers, data, logger):
+    data_json = json.dumps(data)
+    f = None
+    try:
+        req = urllib2.Request(url, data=data_json, headers=headers)
+        req.get_method = lambda: method
+        f = urllib2.urlopen(req)
+        return_code = f.code
+        response_body = f.read()
+        f.close()
+        if len(str(response_body)) == 0:
+            response_body = "{}"
+        response_body = json.loads(response_body)
+        result = {'return_code': return_code, 'body': response_body}
+        return result
+
+    except Exception as e:
+        message = "Cannot read content from {}, exception: {}".format(url, e)
+        logger.exception(message)
+        raise RuntimeError(message)
+    finally:
+        if f is not None:
+            f.close()
+
+
+def upload_file(url, headers, file, logger):
+    logger.debug("Upload file. URL: {}".format(url))
+    response = None
+    try:
+        response = requests.post(url, headers=headers, files=file)
+        return {'return_code': response.status_code, 'body': response.json()}
+    except Exception as e:
+        message = "Error while uploading file to {}, exception: {}".format(url, e)
+        logger.exception(message)
+        raise RuntimeError(message)
+    finally:
+        if response is not None:
+            response.close()