Add test for RDMA Linode instance creation

Author: mawilk90

test/integration/conftest.py

@@ -42,6 +42,8 @@
 RUN_LONG_TESTS = "RUN_LONG_TESTS"
 SKIP_E2E_FIREWALL = "SKIP_E2E_FIREWALL"
 
+TEST_VPC_REGION = None
+
 ALL_ACCOUNT_AVAILABILITIES = {
     "Linodes",
     "NodeBalancers",
@@ -468,14 +470,16 @@ def test_oauth_client(test_linode_client):
 @pytest.fixture(scope="session")
 def create_vpc(test_linode_client):
     client = test_linode_client
-
     label = get_test_label(length=10)
 
+    global TEST_VPC_REGION
+    TEST_VPC_REGION = get_region(
+        test_linode_client, {"VPCs", "VPC IPv6 Stack", "Linode Interfaces"}
+    )
+
     vpc = client.vpcs.create(
         label=label,
-        region=get_region(
-            test_linode_client, {"VPCs", "VPC IPv6 Stack", "Linode Interfaces"}
-        ),
+        region=TEST_VPC_REGION,
         description="test description",
         ipv6=[{"range": "auto"}],
     )
@@ -489,14 +493,19 @@ def create_vpc_with_rdma_type(test_linode_client):
     client = test_linode_client
     label = get_test_label(length=10)
 
-    vpc = client.vpcs.create(
-        label=label,
-        region=get_region(
+    if TEST_VPC_REGION:
+        region = TEST_VPC_REGION
+    else:
+        region = get_region(
             # GPUDirect RDMA capability not available for now
             # test_linode_client, {Capability.vpcs, Capability.gpudirect_rdma}
             test_linode_client,
             {Capability.vpcs},
-        ),
+        )
+
+    vpc = client.vpcs.create(
+        label=label,
+        region=region,
         description="test description",
         vpc_type="rdma",
     )

test/integration/models/linode/interfaces/test_interfaces.py

@@ -1,6 +1,9 @@
 import copy
 import ipaddress
-from test.integration.helpers import get_test_label
+from test.integration.helpers import (
+    get_test_label,
+    wait_for_condition,
+)
 
 import pytest
 
@@ -16,6 +19,9 @@
     LinodeInterfacePublicIPv6Options,
     LinodeInterfacePublicIPv6RangeOptions,
     LinodeInterfacePublicOptions,
+    LinodeInterfaceRDMAVPCIPv4AddressOptions,
+    LinodeInterfaceRDMAVPCIPv4Options,
+    LinodeInterfaceRDMAVPCOptions,
     LinodeInterfaceVLANOptions,
     LinodeInterfaceVPCIPv4AddressOptions,
     LinodeInterfaceVPCIPv4Options,
@@ -43,6 +49,22 @@ def build_interface_public_ipv4(firewall, ip_address):
     )
 
 
+def build_interface_rdma_vpc_ipv4(subnet_id: int):
+    return LinodeInterfaceOptions(
+        firewall_id=-1,
+        rdma_vpc=LinodeInterfaceRDMAVPCOptions(
+            subnet_id=subnet_id,
+            ipv4=LinodeInterfaceRDMAVPCIPv4Options(
+                addresses=[
+                    LinodeInterfaceRDMAVPCIPv4AddressOptions(
+                        address="auto", primary=True
+                    )
+                ]
+            ),
+        ),
+    )
+
+
 def create_linode_with_legacy_config(
     client, ip_address, label, firewall, authorized_key
 ):
@@ -75,6 +97,22 @@ def create_linode_with_standard_interfaces(
     return linode
 
 
+def create_multiple_rdma_interfaces(amount: int, subnet_id: int):
+    """
+    Creates multiple VPC RDMA interfaces that may be needed for RDMA Linode instance
+
+    Note:
+    At least 8 separate VPC RDMA interfaces need to be created for a single RDMA linode instance
+    """
+    interfaces = list()
+
+    for _ in range(amount):
+        rdma_iface = build_interface_rdma_vpc_ipv4(subnet_id)
+        interfaces.append(rdma_iface)
+
+    return interfaces
+
+
 def test_linode_create_with_linode_interfaces(
     create_vpc_with_subnet,
     linode_with_linode_interfaces,
@@ -458,3 +496,61 @@ def test_linode_interfaces_with_reserved_ips(
     assert reserved_ips_list[0].reserved == True
     assert reserved_ips_list[0].linode_id is None
     assert reserved_ips_list[0].assigned_entity is None
+
+
+@pytest.mark.skip(
+    reason="Linode with RDMA interfaces requires manual infra changes at the moment"
+)
+def test_linode_interfaces_with_rdma_vpc_type(
+    request,
+    test_linode_client,
+    create_vpc_with_subnet,
+    create_vpc_with_subnet_and_rdma_type,
+):
+    client = test_linode_client
+    vpc, subnet = create_vpc_with_subnet
+    vpc_rdma, subnet_rdma = create_vpc_with_subnet_and_rdma_type
+
+    # Include RDMA VPC interfaces
+    multi_ifaces = create_multiple_rdma_interfaces(8, subnet_rdma.id)
+
+    # Include (at least one) regular interface
+    multi_ifaces.append(
+        LinodeInterfaceOptions(
+            firewall_id=-1,
+            default_route=LinodeInterfaceDefaultRouteOptions(
+                ipv4=True,
+            ),
+            vpc=LinodeInterfaceVPCOptions(
+                subnet_id=subnet.id,
+                ipv4=LinodeInterfaceVPCIPv4Options(
+                    addresses=[
+                        LinodeInterfaceVPCIPv4AddressOptions(
+                            address="auto",
+                            primary=True,
+                        )
+                    ],
+                ),
+            ),
+        ),
+    )
+
+    instance = client.linode.instance_create(
+        label="go-test-rdma-" + get_test_label(),
+        root_pass="aComplex@Password123",
+        image="linode/ubuntu24.04",
+        region=vpc.region,
+        # ltype="TBD",
+        # host_id="TBD",
+        interface_generation=InterfaceGeneration.LINODE,
+        interfaces=multi_ifaces,
+        booted=False,
+    )
+    request.addfinalizer(instance.delete)
+
+    def get_linode_status():
+        return instance.status == "offline"
+
+    wait_for_condition(5, 180, get_linode_status)
+
+    assert len(instance.linode_interfaces) == len(multi_ifaces)