Merge pull request #87 from vineyyadav/master

Added support for providing host_uri argument when creating a Nebula object

Author: naorlivne

NebulaPythonSDK/sdk.py

@@ -12,14 +12,17 @@ class Nebula:
     # the nebula class init module serves as the login against the nebula API as it's the only shared thing among the
     # class functions
     def __init__(self, username=None, password=None, token=None, host="127.0.0.1", port=80, protocol="http",
-                 request_timeout=60):
+                 host_uri=None, request_timeout=60):
         self.request_timeout = request_timeout
         self.username = username
         self.password = password
         self.token = token
         self.protocol = protocol
         self.port = port
-        self.host = protocol + "://" + host + ":" + str(port)
+        if host_uri:
+            self.host = host_uri
+        else:
+            self.host = protocol + "://" + host + ":" + str(port)
         self.headers = {
             'content-type': "application/json",
             'cache-control': "no-cache"

test/test_nebula_python_sdk.py

@@ -454,3 +454,18 @@ def test_cron_job_flow(self, cron_job="unit_test_cron_job"):
         reply = nebula_connection_object.delete_cron_job(cron_job)
         self.assertEqual(reply["status_code"], 200)
         self.assertEqual(reply["reply"], {})
+
+    def test_nebula_connection_with_host_uri(self):
+        nebula_user = os.getenv("NEBULA_TEST_USERNAME", "nebula")
+        nebula_password = os.getenv("NEBULA_TEST_PASSWORD", "nebula")
+        nebula_host_uri = os.getenv("NEBULA_HOST_URI", "http://127.0.0.1:80")
+        connection = Nebula(username=nebula_user, password=nebula_password, host_uri=nebula_host_uri)
+
+        # check that the host param is set correctly
+        self.assertEqual(connection.host, nebula_host_uri)
+
+        # check that connects to manager successfully
+        reply = connection.check_api()
+        self.assertEqual(reply["status_code"], 200)
+        self.assertEqual(reply["reply"]["api_available"], True)
+