Create test for frontend VPC IPv4

Author: mawilk90

tests/integration/nodebalancers/test_node_balancers.py

@@ -555,23 +555,22 @@ def test_nb_with_backend_vpc_only(get_vpc_with_subnet):
             vpc["region"],
             "--backend_vpcs.subnet_id",
             str(vpc["subnets"][0]["id"]),
-            "--text",
-            "--delimiter",
-            ",",
-            "--no-headers",
+            "--json",
         ]
-    ).split(",")
+    )
+    nb_attrs = json.loads(nb_attrs)
+    nb_id = str(nb_attrs[0]["id"])
 
-    assert isinstance(ipaddress.ip_address(nb_attrs[5]), ipaddress.IPv4Address)
-    assert isinstance(ipaddress.ip_address(nb_attrs[6]), ipaddress.IPv6Address)
-    assert nb_attrs[-2] == "public"
-    assert nb_attrs[-1] == ""
+    assert isinstance(ipaddress.ip_address(nb_attrs[0]["ipv4"]), ipaddress.IPv4Address)
+    assert isinstance(ipaddress.ip_address(nb_attrs[0]["ipv6"]), ipaddress.IPv6Address)
+    assert nb_attrs[0]["frontend_address_type"] == "public"
+    assert nb_attrs[0]["frontend_vpc_subnet_id"] is None
 
     nb_vpcs = exec_test_command(
         BASE_CMDS["nodebalancers"]
         + [
             "vpcs-list",
-            nb_attrs[0],
+            nb_id,
             "--json",
         ]
     )
@@ -584,7 +583,7 @@ def test_nb_with_backend_vpc_only(get_vpc_with_subnet):
         BASE_CMDS["nodebalancers"]
         + [
             "vpc-view",
-            nb_attrs[0],
+            nb_id,
             str(nb_vpcs[0]["id"]),
             "--json",
         ]
@@ -598,7 +597,7 @@ def test_nb_with_backend_vpc_only(get_vpc_with_subnet):
     #     BASE_CMDS["nodebalancers"]
     #     + [
     #         "backend_vpcs-list",
-    #         nb_attrs[0],
+    #         nb_id,
     #         "--json",
     #     ]
     # )
@@ -610,11 +609,65 @@ def test_nb_with_backend_vpc_only(get_vpc_with_subnet):
     #     BASE_CMDS["nodebalancers"]
     #     + [
     #         "frontend_vpcs-list",
-    #         nb_attrs[0],
+    #         nb_id,
     #         "--json",
     #     ]
     # )
     # nb_frontend_vpcs = json.loads(nb_frontend_vpcs)
     # assert len(nb_frontend_vpcs) == 0
 
-    delete_target_id(target="nodebalancers", id=nb_attrs[0])
+    delete_target_id(target="nodebalancers", id=nb_id)
+
+
+def test_nb_with_frontend_ipv4_only(get_vpc_with_subnet):
+    vpc = get_vpc_with_subnet
+    ipv4_address = "10.0.0.2"  # first available address
+
+    nb_attrs = exec_test_command(
+        BASE_CMDS["nodebalancers"]
+        + [
+            "create",
+            "--region",
+            vpc["region"],
+            "--frontend_vpcs.subnet_id",
+            str(vpc["subnets"][0]["id"]),
+            "--frontend_vpcs.ipv4_range",
+            "10.0.0.0/24",
+            "--type",
+            "premium",
+            "--json",
+        ]
+    )
+    nb_attrs = json.loads(nb_attrs)
+    nb_id = str(nb_attrs[0]["id"])
+
+    assert nb_attrs[0]["ipv4"] == ipv4_address
+    assert nb_attrs[0]["ipv6"] is None
+    assert nb_attrs[0]["frontend_address_type"] == "vpc"
+    assert nb_attrs[0]["frontend_vpc_subnet_id"] == vpc["subnets"][0]["id"]
+
+    # TODO: Uncomment when API implementation of /backend_vpcs and /frontend_vpcs endpoints is finished
+    # nb_frontend_vpcs = exec_test_command(
+    #     BASE_CMDS["nodebalancers"]
+    #     + [
+    #         "frontend_vpcs-list",
+    #         nb_id,
+    #         "--json",
+    #     ]
+    # )
+    # nb_frontend_vpcs = json.loads(nb_frontend_vpcs)
+    # assert len(nb_frontend_vpcs) == 1
+    # assert nb_frontend_vpcs[0]["purpose"] == "frontend"
+    #
+    # nb_backend_vpcs = exec_test_command(
+    #     BASE_CMDS["nodebalancers"]
+    #     + [
+    #         "backend_vpcs-list",
+    #         nb_id,
+    #         "--json",
+    #     ]
+    # )
+    # nb_backend_vpcs = json.loads(nb_backend_vpcs)
+    # assert len(nb_backend_vpcs) == 0
+
+    delete_target_id(target="nodebalancers", id=nb_id)