Skip to content

Commit 6ee4d06

Browse files
committed
Move fixtures for reserved IPs into conftest
1 parent 7da867a commit 6ee4d06

1 file changed

Lines changed: 40 additions & 0 deletions

File tree

test/integration/conftest.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
PlacementGroupPolicy,
2626
PlacementGroupType,
2727
PostgreSQLDatabase,
28+
ReservedIPAddress,
2829
)
2930
from linode_api4.errors import ApiError
3031
from linode_api4.linode_client import LinodeClient, MonitorClient
@@ -727,3 +728,42 @@ def test_monitor_client(get_monitor_token_for_db_entities):
727728
)
728729

729730
return client, entity_ids
731+
732+
733+
@pytest.fixture
734+
def create_reserved_ip(test_linode_client):
735+
client = test_linode_client
736+
region = get_region(client, {"Linodes", "Cloud Firewall"}, site_type="core")
737+
reserved_ip = client.networking.reserved_ip_create(
738+
region=region,
739+
tags=["test1"]
740+
)
741+
742+
yield reserved_ip
743+
744+
# Delete only if IP exists (some tests delete it earlier)
745+
if client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address):
746+
reserved_ip.delete()
747+
748+
749+
@pytest.fixture
750+
def create_reserved_ip_assigned(test_linode_client, create_linode):
751+
client = test_linode_client
752+
linode = create_linode
753+
reserved_ip = client.networking.reserved_ip_create(
754+
region=linode.region,
755+
tags=["test", "assigned"],
756+
)
757+
758+
client.networking.ip_addresses_assign(
759+
assignments=[{"address": reserved_ip.address, "linode_id": linode.id}],
760+
region=linode.region,
761+
)
762+
763+
reserved_ip = test_linode_client.load(ReservedIPAddress, reserved_ip.address)
764+
765+
yield linode, reserved_ip
766+
767+
# Delete only if IP exists (some tests delete it earlier)
768+
if client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address):
769+
reserved_ip.delete()

0 commit comments

Comments
 (0)