Merge pull request swayf#54 from pvanagtmaal/port-in-hostname

Add option to specify port in host

Author: swayf

proxmoxer/backends/https.py

@@ -113,7 +113,12 @@ def request(self, method, url, params=None, data=None, headers=None, cookies=Non
 class Backend(object):
     def __init__(self, host, user, password, port=8006, verify_ssl=True,
                  mode='json', timeout=5, auth_token=None, csrf_token=None):
+        if ':' in host:
+            host, host_port = host.split(':')
+            port = host_port if host_port.isdigit() else port
+
         self.base_url = "https://{0}:{1}/api2/{2}".format(host, port, mode)
+
         if auth_token is not None:
             self.auth = ProxmoxHTTPTokenAuth(auth_token, csrf_token)
         else:

tests/https_tests.py

@@ -20,6 +20,32 @@ def test_https_connection(req_session):
     eq_(call['verify'], False)
 
 
+@patch('requests.sessions.Session')
+def test_https_connection_wth_port_in_host(req_session):
+    response = {'ticket': 'ticket',
+                'CSRFPreventionToken': 'CSRFPreventionToken'}
+    req_session.request.return_value = response
+    ProxmoxAPI('proxmox:123', user='root@pam', password='secret', port=124, verify_ssl=False)
+    call = req_session.return_value.request.call_args[1]
+    eq_(call['url'], 'https://proxmox:123/api2/json/access/ticket')
+    eq_(call['data'], {'username': 'root@pam', 'password': 'secret'})
+    eq_(call['method'], 'post')
+    eq_(call['verify'], False)
+
+
+@patch('requests.sessions.Session')
+def test_https_connection_wth_bad_port_in_host(req_session):
+    response = {'ticket': 'ticket',
+                'CSRFPreventionToken': 'CSRFPreventionToken'}
+    req_session.request.return_value = response
+    ProxmoxAPI('proxmox:notaport', user='root@pam', password='secret', port=124, verify_ssl=False)
+    call = req_session.return_value.request.call_args[1]
+    eq_(call['url'], 'https://proxmox:124/api2/json/access/ticket')
+    eq_(call['data'], {'username': 'root@pam', 'password': 'secret'})
+    eq_(call['method'], 'post')
+    eq_(call['verify'], False)
+
+
 class TestSuite():
     proxmox = None
     serializer = None