Create int tests for Reserved IPs networking endpoints

Author: mawilk90

test/integration/models/networking/test_networking.py

@@ -1,3 +1,4 @@
+import ipaddress
 import time
 from test.integration.conftest import (
     get_api_ca_file,
@@ -13,8 +14,8 @@
 
 import pytest
 
-from linode_api4 import Instance, LinodeClient
-from linode_api4.objects import Config, ConfigInterfaceIPv4, Firewall, IPAddress
+from linode_api4 import Instance, LinodeClient, ApiError
+from linode_api4.objects import Config, ConfigInterfaceIPv4, Firewall, IPAddress, ReservedIPAddress
 from linode_api4.objects.networking import (
     FirewallCreateDevicesOptions,
     NetworkTransferPrice,
@@ -351,3 +352,120 @@ def test_ip_info(test_linode_client, create_linode):
     assert ip_info.subnet_mask is not None
     assert ip_info.type == "ipv4"
     assert ip_info.vpc_nat_1_1 is None
+
+
+@pytest.fixture
+def create_reserved_ip(test_linode_client):
+    client = test_linode_client
+    reserved_ip = client.networking.reserved_ip_create(
+        region=TEST_REGION,
+        tags=["test1"]
+    )
+
+    yield reserved_ip
+
+    reserved_ip.delete()
+
+
+@pytest.fixture
+def create_reserved_ip_assigned(test_linode_client, create_linode):
+    client = test_linode_client
+    linode = create_linode
+    reserved_ip = client.networking.reserved_ip_create(
+        region=linode.region,
+        tags=["test", "assigned"],
+    )
+
+    client.networking.ip_addresses_assign(
+        assignments=[{"address": reserved_ip.address, "linode_id": linode.id}],
+        region=linode.region,
+    )
+
+    reserved_ip = test_linode_client.load(ReservedIPAddress, reserved_ip.address)
+
+    yield linode, reserved_ip
+
+    # Delete only if IP exists (some tests delete it earlier)
+    if reserved_ip.address in [ip.address for ip in client.networking.reserved_ips()]:
+        reserved_ip.delete()
+
+
+@pytest.mark.smoke
+@pytest.mark.parametrize("region, tags", [
+    (TEST_REGION, ["test"]),
+    (TEST_REGION, None),
+])
+def test_create_reserved_ip(request, test_linode_client, region, tags):
+    client = test_linode_client
+    reserved_ip = client.networking.reserved_ip_create(
+        region=region,
+        tags=tags
+    )
+
+    request.addfinalizer(reserved_ip.delete)
+
+    assert isinstance(ipaddress.ip_address(reserved_ip.address), ipaddress.IPv4Address)
+    assert reserved_ip.type == "ipv4"
+    assert reserved_ip.public == True
+    assert reserved_ip.linode_id is None
+    assert reserved_ip.reserved == True
+    # assert reserved_ip.tags == tags  # NOTE: Skipped as tags not available in the API yet
+    assert reserved_ip.assigned_entity is None
+
+
+def test_create_reserved_ip_wo_region_fail(test_linode_client):
+    client = test_linode_client
+
+    with pytest.raises(ApiError) as exc_info:
+        client.networking.reserved_ip_create(
+            region=None,
+            tags=["test"]
+        )
+
+    error_msg = str(exc_info.value.json)
+    assert exc_info.value.status == 400
+    assert "region is required" in error_msg
+
+
+@pytest.mark.skip   # NOTE: Skipped as tags not available in the API yet
+def test_update_reserved_ip_tags(create_reserved_ip):
+    reserved_ip = create_reserved_ip
+    assert reserved_ip.tags == ["test"]
+
+    reserved_ip.save(tags=["updated"])
+    assert reserved_ip.tags == ["updated"]
+
+
+def test_create_reserved_ip_assigned(test_linode_client, create_reserved_ip_assigned):
+    client = test_linode_client
+    linode, reserved_ip = create_reserved_ip_assigned
+
+    assert reserved_ip.reserved == True
+    # assert reserved_ip.tags == tags  # NOTE: Skipped as tags not available in the API yet
+    assert reserved_ip.linode_id == linode.id
+    assert reserved_ip.assigned_entity.id == linode.id
+    assert reserved_ip.assigned_entity.type == "linode"
+    assert reserved_ip.assigned_entity.label == linode.label
+    assert reserved_ip.assigned_entity.url == f"/v4/linode/instances/{linode.id}"
+
+    ips_list = client.networking.ips()
+    assert reserved_ip.address in [ip.address for ip in ips_list]
+
+    reserved_ips_list = client.networking.reserved_ips()
+    assert reserved_ip.address in [ip.address for ip in reserved_ips_list]
+
+    # linode_ips = linode.ips.ipv4.public
+    # assert len(linode_ips) == 2
+    # assert any([ip.reserved for ip in linode_ips])
+
+    reserved_ip.delete()
+    reserved_ips_list = client.networking.reserved_ips()
+    assert reserved_ip.address not in [ip.address for ip in reserved_ips_list]
+
+    reserved_ips_list = client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)
+    assert len(reserved_ips_list) == 0
+
+    # delattr(linode, "_ips")
+    # linode_ips = linode.ips.ipv4.public
+    # assert len(linode_ips) == 2
+    # assert not any([ip.reserved for ip in linode_ips])