Merge pull request #102 from akx/floating-ip

Add basic floating IP support

Author: ajmyyra

test/json_data/ip_address_post.json

@@ -4,6 +4,8 @@
         "address": "10.1.0.101",
         "family": "IPv4",
         "ptr_record": "a.ptr.record",
-        "server": "00798b85-efdc-41ca-8021-f6ef457b8531"
+        "server": "00798b85-efdc-41ca-8021-f6ef457b8531",
+        "floating": "yes",
+        "zone": "fi-hel2"
     }
-}
+}

test/test_ip_manager.py

@@ -44,3 +44,11 @@ def test_ip_delete(self, manager):
         Mock.mock_delete('ip_address/10.1.0.101')
         res = manager.release_ip('10.1.0.101')
         assert res == {}
+
+    @responses.activate
+    def test_create_floating_ip(self, manager):
+        Mock.mock_post('ip_address')
+        floating_ip = manager.create_floating_ip('fi-hel2')
+        assert type(floating_ip).__name__ == 'IPAddress'
+        assert floating_ip.floating == 'yes'
+        assert floating_ip.zone == 'fi-hel2'

upcloud_api/cloud_manager/ip_address_mixin.py

@@ -55,3 +55,15 @@ def release_ip(self, ip_addr):
         Accepts an IPAddress instance (object) or its address (string).
         """
         return self.api.delete_request('/ip_address/' + str(ip_addr))
+
+    def create_floating_ip(self, zone: str, mac: str = '', family: str = 'IPv4') -> IPAddress:
+        """
+        Create a floating IP and returns an IPAddress object.
+        Specify MAC address of network interface to attach the floating IP when it is created
+        """
+        body = {'ip_address': {'family': family, 'floating': 'yes', 'zone': zone}}
+        if mac:
+            body['ip_address']['mac'] = mac
+
+        res = self.api.post_request('/ip_address', body)
+        return IPAddress(cloud_manager=self, **res['ip_address'])