Linter fix

Author: mawilk90

test/integration/conftest.py

@@ -735,14 +735,15 @@ def create_reserved_ip(test_linode_client):
     client = test_linode_client
     region = get_region(client, {"Linodes", "Cloud Firewall"}, site_type="core")
     reserved_ip = client.networking.reserved_ip_create(
-        region=region,
-        tags=["test"]
+        region=region, tags=["test"]
     )
 
     yield reserved_ip
 
     # Delete only if IP exists (some tests delete it earlier)
-    if client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address):
+    if client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    ):
         reserved_ip.delete()
 
 
@@ -760,10 +761,14 @@ def create_reserved_ip_assigned(test_linode_client, create_linode):
         region=linode.region,
     )
 
-    reserved_ip = test_linode_client.load(ReservedIPAddress, reserved_ip.address)
+    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 client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address):
+    if client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    ):
         reserved_ip.delete()

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

@@ -383,11 +383,13 @@ def test_linode_interface_firewalls(e2e_test_firewall, linode_interface_public):
     assert firewall.label == e2e_test_firewall.label
 
 
-@pytest.mark.parametrize("iface_type", [
-    InterfaceGeneration.LEGACY_CONFIG,
-    InterfaceGeneration.LINODE
-])
-def test_linode_interfaces_with_reserved_ips(test_linode_client, e2e_test_firewall, create_reserved_ip, iface_type):
+@pytest.mark.parametrize(
+    "iface_type",
+    [InterfaceGeneration.LEGACY_CONFIG, InterfaceGeneration.LINODE],
+)
+def test_linode_interfaces_with_reserved_ips(
+    test_linode_client, e2e_test_firewall, create_reserved_ip, iface_type
+):
     client = test_linode_client
     reserved_ip = create_reserved_ip
     label = get_test_label(length=8)
@@ -400,10 +402,12 @@ def test_linode_interfaces_with_reserved_ips(test_linode_client, e2e_test_firewa
             label=label,
             firewall=e2e_test_firewall,
             interface_generation=iface_type,
-            ipv4=[reserved_ip.address]
+            ipv4=[reserved_ip.address],
         )
     else:
-        interface = build_interface_public_ipv4(e2e_test_firewall.id, reserved_ip.address)
+        interface = build_interface_public_ipv4(
+            e2e_test_firewall.id, reserved_ip.address
+        )
         linode, _ = client.linode.instance_create(
             "g6-nanode-1",
             reserved_ip.region,
@@ -422,10 +426,14 @@ def test_linode_interfaces_with_reserved_ips(test_linode_client, e2e_test_firewa
     assert linode_ips[0].assigned_entity.id == linode.id
     assert linode_ips[0].assigned_entity.type == "linode"
     assert linode_ips[0].assigned_entity.label == linode.label
-    assert linode_ips[0].assigned_entity.url == f"/v4/linode/instances/{linode.id}"
+    assert (
+        linode_ips[0].assigned_entity.url == f"/v4/linode/instances/{linode.id}"
+    )
 
     linode.delete()
-    reserved_ips_list = client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)
+    reserved_ips_list = client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )
     assert len(reserved_ips_list) == 1
     assert reserved_ips_list[0].reserved == True
     # assert linode_ips[0].tags == ["test"]  # TODO Does not work at the moment - during clarifications with API Team

test/integration/models/linode/test_linode.py

@@ -1159,7 +1159,9 @@ def test_update_linode_maintenance_policy(create_linode, test_linode_client):
     assert linode.maintenance_policy_id == non_default_policy.slug
 
 
-def test_update_linode_with_reserved_ip_in_address(test_linode_client, e2e_test_firewall, create_reserved_ip):
+def test_update_linode_with_reserved_ip_in_address(
+    test_linode_client, e2e_test_firewall, create_reserved_ip
+):
     label = get_test_label(length=8)
     client = test_linode_client
     reserved_ip = create_reserved_ip
@@ -1182,11 +1184,15 @@ def test_update_linode_with_reserved_ip_in_address(test_linode_client, e2e_test_
     assert len(linode_ips) == 2
     assert reserved_ip.address in [ip.address for ip in linode_ips]
 
-    reserved_ip = client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)[0]
+    reserved_ip = client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )[0]
     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}"
+    assert (
+        reserved_ip.assigned_entity.url == f"/v4/linode/instances/{linode.id}"
+    )
 
     linode.delete()

test/integration/models/networking/test_networking.py

@@ -366,7 +366,9 @@ def test_ip_info(test_linode_client, create_linode):
 
 
 def verify_reserved_ip(reserved_ip):
-    assert isinstance(ipaddress.ip_address(reserved_ip.address), ipaddress.IPv4Address)
+    assert isinstance(
+        ipaddress.ip_address(reserved_ip.address), ipaddress.IPv4Address
+    )
     assert reserved_ip.type == "ipv4"
     assert reserved_ip.public == True
     assert reserved_ip.reserved == True
@@ -375,7 +377,9 @@ def verify_reserved_ip(reserved_ip):
 
 
 def verify_reserved_ip_assigned(reserved_ip, resource):
-    assert isinstance(ipaddress.ip_address(reserved_ip.address), ipaddress.IPv4Address)
+    assert isinstance(
+        ipaddress.ip_address(reserved_ip.address), ipaddress.IPv4Address
+    )
     assert reserved_ip.type == "ipv4"
     assert reserved_ip.public == True
     assert reserved_ip.reserved == True
@@ -384,20 +388,22 @@ def verify_reserved_ip_assigned(reserved_ip, resource):
     assert reserved_ip.assigned_entity.id == resource.id
     assert reserved_ip.assigned_entity.type == "linode"
     assert reserved_ip.assigned_entity.label == resource.label
-    assert reserved_ip.assigned_entity.url == f"/v4/linode/instances/{resource.id}"
+    assert (
+        reserved_ip.assigned_entity.url == f"/v4/linode/instances/{resource.id}"
+    )
 
 
 @pytest.mark.smoke
-@pytest.mark.parametrize("region, tags", [
-    (TEST_REGION, ["test"]),
-    (TEST_REGION, None),
-])
+@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
-    )
+    reserved_ip = client.networking.reserved_ip_create(region=region, tags=tags)
     request.addfinalizer(reserved_ip.delete)
 
     verify_reserved_ip(reserved_ip)
@@ -408,10 +414,7 @@ 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"]
-        )
+        client.networking.reserved_ip_create(region=None, tags=["test"])
 
     error_msg = str(exc_info.value.json)
     assert exc_info.value.status == 400
@@ -426,12 +429,16 @@ def test_update_reserved_ip_tags(test_linode_client, create_reserved_ip):
 
     reserved_ip.tags = ["updated"]
     reserved_ip.save()
-    reserved_ip = client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)[0]
+    reserved_ip = client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )[0]
     verify_reserved_ip(reserved_ip)
     assert reserved_ip.tags == ["updated"]
 
 
-def test_create_reserved_ip_assigned(test_linode_client, create_reserved_ip_assigned):
+def test_create_reserved_ip_assigned(
+    test_linode_client, create_reserved_ip_assigned
+):
     client = test_linode_client
     linode, reserved_ip = create_reserved_ip_assigned
     verify_reserved_ip_assigned(reserved_ip, linode)
@@ -451,7 +458,9 @@ def test_create_reserved_ip_assigned(test_linode_client, create_reserved_ip_assi
     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)
+    reserved_ips_list = client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )
     assert len(reserved_ips_list) == 0
 
     delattr(linode, "_ips")
@@ -464,7 +473,9 @@ def test_create_reserved_ip_assigned(test_linode_client, create_reserved_ip_assi
 def test_get_reserved_ip_types(test_linode_client, create_reserved_ip):
     client = test_linode_client
     endpoint = client.base_url + "/networking/reserved/ips/types"
-    types = requests.get(endpoint).json()["data"]  # Pricing should be publicly available
+    types = requests.get(endpoint).json()[
+        "data"
+    ]  # Pricing should be publicly available
 
     assert isinstance(types, list)
     assert types[0]["id"] == "reserved-ipv4"
@@ -476,19 +487,28 @@ def test_get_reserved_ip_types(test_linode_client, create_reserved_ip):
 
 
 @pytest.mark.smoke
-@pytest.mark.parametrize("reserved, region", [
-    (True, TEST_REGION),
-    (True, None),
-])
-def test_create_reserved_ip_with_allocate(test_linode_client, create_linode, reserved, region):
+@pytest.mark.parametrize(
+    "reserved, region",
+    [
+        (True, TEST_REGION),
+        (True, None),
+    ],
+)
+def test_create_reserved_ip_with_allocate(
+    test_linode_client, create_linode, reserved, region
+):
     client = test_linode_client
     linode = create_linode
 
     if region:
-        reserved_ip = client.networking.ip_allocate(reserved=reserved, region=TEST_REGION)
+        reserved_ip = client.networking.ip_allocate(
+            reserved=reserved, region=TEST_REGION
+        )
         verify_reserved_ip(reserved_ip)
     else:
-        reserved_ip = client.networking.ip_allocate(reserved=reserved, linode=linode.id)
+        reserved_ip = client.networking.ip_allocate(
+            reserved=reserved, linode=linode.id
+        )
         verify_reserved_ip_assigned(reserved_ip, linode)
 
     assert reserved_ip.tags == []
@@ -515,7 +535,9 @@ def test_reserve_ephemeral_ip(test_linode_client, create_linode):
     assert ip_address.reserved == False
 
 
-def test_convert_unassigned_reserved_ip_to_ephemeral(test_linode_client, create_reserved_ip):
+def test_convert_unassigned_reserved_ip_to_ephemeral(
+    test_linode_client, create_reserved_ip
+):
     client = test_linode_client
     reserved_ip = create_reserved_ip
     verify_reserved_ip(reserved_ip)
@@ -524,5 +546,7 @@ def test_convert_unassigned_reserved_ip_to_ephemeral(test_linode_client, create_
     ip_address.reserved = False
     ip_address.save()
 
-    reserved_ips_list = client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)
+    reserved_ips_list = client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )
     assert len(reserved_ips_list) == 0

test/integration/models/nodebalancer/test_nodebalancer.py

@@ -114,7 +114,9 @@ def test_create_nb(test_linode_client, e2e_test_firewall):
     nb.delete()
 
 
-def test_create_nb_with_reserved_ip(test_linode_client, e2e_test_firewall, create_reserved_ip):
+def test_create_nb_with_reserved_ip(
+    test_linode_client, e2e_test_firewall, create_reserved_ip
+):
     client = test_linode_client
     reserved_ip = create_reserved_ip
     label = get_test_label(8)
@@ -134,7 +136,9 @@ def test_create_nb_with_reserved_ip(test_linode_client, e2e_test_firewall, creat
     assert nb.ipv4.reserved == True
 
     nb.delete()
-    reserved_ip = client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)[0]
+    reserved_ip = client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )[0]
     assert reserved_ip.assigned_entity is None
 
 

test/integration/models/tag/test_tag.py

@@ -20,8 +20,12 @@ def create_tag_with_reserved_ip(test_linode_client, create_reserved_ip):
     unique_tag = get_test_label() + "_tag"
     reserved_ip = create_reserved_ip
 
-    tag = test_linode_client.tags.create(unique_tag, reserved_ipv4_addresses=[reserved_ip.address])
-    reserved_ip = test_linode_client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)[0]
+    tag = test_linode_client.tags.create(
+        unique_tag, reserved_ipv4_addresses=[reserved_ip.address]
+    )
+    reserved_ip = test_linode_client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )[0]
 
     yield tag, reserved_ip
 
@@ -35,7 +39,9 @@ def test_get_tag(test_linode_client, test_tag):
     assert tag.id == test_tag.id
 
 
-def test_get_tag_with_reserved_ip(test_linode_client, create_tag_with_reserved_ip):
+def test_get_tag_with_reserved_ip(
+    test_linode_client, create_tag_with_reserved_ip
+):
     tag, reserved_ip = create_tag_with_reserved_ip
     tag = test_linode_client.load(Tag, tag.id).objects[0]
 
@@ -45,5 +51,7 @@ def test_get_tag_with_reserved_ip(test_linode_client, create_tag_with_reserved_i
     assert tag.tags == reserved_ip.tags
 
     tag.delete()
-    reserved_ip = test_linode_client.networking.reserved_ips(ReservedIPAddress.address==reserved_ip.address)
+    reserved_ip = test_linode_client.networking.reserved_ips(
+        ReservedIPAddress.address == reserved_ip.address
+    )
     assert len(reserved_ip) == 0