Add unit tests

Author: lgarber-akamai

test/unit/objects/linode_test.py

@@ -541,6 +541,24 @@ def test_interface_ipv4(self):
         self.assertEqual(ipv4.vpc, "10.0.0.1")
         self.assertEqual(ipv4.nat_1_1, "any")
 
+    def test_interface_ipv6(self):
+        json = {
+            "slaac": [{"range": "1234::5678/64", "address": "1234::5678"}],
+            "ranges": [{"range": "1234::5678/64"}],
+            "is_public": True,
+        }
+
+        ipv6 = ConfigInterfaceIPv6.from_json(json)
+
+        assert len(ipv6.slaac) == 1
+        assert ipv6.slaac[0].range == "1234::5678/64"
+        assert ipv6.slaac[0].address == "1234::5678"
+
+        assert len(ipv6.ranges) == 1
+        assert ipv6.ranges[0].range == "1234::5678/64"
+
+        assert ipv6.is_public
+
     def test_config_devices_unwrap(self):
         """
         Tests that config devices can be successfully converted to a dict.
@@ -744,6 +762,11 @@ def test_create_interface_vpc(self):
                 subnet=VPCSubnet(self.client, 789, 123456),
                 primary=True,
                 ipv4=ConfigInterfaceIPv4(vpc="10.0.0.4", nat_1_1="any"),
+                ipv6=ConfigInterfaceIPv6Options(
+                    slaac=[ConfigInterfaceIPv6OptionsSLAAC(range="auto")],
+                    ranges=[ConfigInterfaceIPv6OptionsRange(range="auto")],
+                    is_public=True,
+                ),
                 ip_ranges=["10.0.0.0/24"],
             )
 
@@ -757,6 +780,11 @@ def test_create_interface_vpc(self):
                 "primary": True,
                 "subnet_id": 789,
                 "ipv4": {"vpc": "10.0.0.4", "nat_1_1": "any"},
+                "ipv6": {
+                    "slaac": [{"range": "auto"}],
+                    "ranges": [{"range": "auto"}],
+                    "is_public": True,
+                },
                 "ip_ranges": ["10.0.0.0/24"],
             }
 
@@ -765,15 +793,27 @@ def test_create_interface_vpc(self):
             assert interface.primary
             assert interface.vpc.id == 123456
             assert interface.subnet.id == 789
+
             assert interface.ipv4.vpc == "10.0.0.2"
             assert interface.ipv4.nat_1_1 == "any"
+
+            assert len(interface.ipv6.slaac) == 1
+            assert interface.ipv6.slaac[0].range == "1234::5678/64"
+            assert interface.ipv6.slaac[0].address == "1234::5678"
+
+            assert len(interface.ipv6.ranges) == 1
+            assert interface.ipv6.ranges[0].range == "1234::5678/64"
+
+            assert interface.ipv6.is_public
+
             assert interface.ip_ranges == ["10.0.0.0/24"]
 
     def test_update(self):
         interface = NetworkInterface(self.client, 123, 456789, 123)
         interface._api_get()
 
         interface.ipv4.vpc = "10.0.0.3"
+        interface.ipv6.is_public = False
         interface.primary = False
         interface.ip_ranges = ["10.0.0.2/32"]
 
@@ -791,6 +831,11 @@ def test_update(self):
             assert m.call_data == {
                 "primary": False,
                 "ipv4": {"vpc": "10.0.0.3", "nat_1_1": "any"},
+                "ipv6": {
+                    "slaac": [{"range": "auto"}],
+                    "ranges": [{"range": "auto"}],
+                    "is_public": True,
+                },
                 "ip_ranges": ["10.0.0.2/32"],
             }
 
@@ -811,8 +856,17 @@ def test_get_vpc(self):
         self.assertEqual(interface.purpose, "vpc")
         self.assertEqual(interface.vpc.id, 123456)
         self.assertEqual(interface.subnet.id, 789)
+
         self.assertEqual(interface.ipv4.vpc, "10.0.0.2")
         self.assertEqual(interface.ipv4.nat_1_1, "any")
+
+        self.assertEqual(len(interface.ipv6.slaac), 1)
+        self.assertEqual(interface.ipv6.slaac[0].range, "1234::5678/64")
+        self.assertEqual(interface.ipv6.slaac[0].address, "1234::5678")
+        self.assertEqual(len(interface.ipv6.ranges), 1)
+        self.assertEqual(interface.ipv6.ranges[0].range, "1234::5678/64")
+        self.assertEqual(interface.ipv6.is_public, True)
+
         self.assertEqual(interface.ip_ranges, ["10.0.0.0/24"])
         self.assertEqual(interface.active, True)
 

test/unit/objects/vpc_test.py

@@ -113,6 +113,8 @@ def validate_vpc_123456(self, vpc: VPC):
         self.assertEqual(vpc.created, expected_dt)
         self.assertEqual(vpc.updated, expected_dt)
 
+        self.assertEqual(vpc.ipv6[0].range, "fd71:1140:a9d0::/52")
+
     def validate_vpc_subnet_789(self, subnet: VPCSubnet):
         expected_dt = datetime.datetime.strptime(
             "2018-01-01T00:01:01", DATE_FORMAT
@@ -124,6 +126,8 @@ def validate_vpc_subnet_789(self, subnet: VPCSubnet):
         self.assertEqual(subnet.created, expected_dt)
         self.assertEqual(subnet.updated, expected_dt)
 
+        self.assertEqual(subnet.ipv6[0].range, "fd71:1140:a9d0::/52")
+
     def test_list_vpc_ips(self):
         """
         Test that the ips under a specific VPC can be listed.
@@ -148,3 +152,11 @@ def test_list_vpc_ips(self):
         self.assertEqual(vpc_ip.gateway, "10.0.0.1")
         self.assertEqual(vpc_ip.prefix, 8)
         self.assertEqual(vpc_ip.subnet_mask, "255.0.0.0")
+
+        vpc_ip_2 = vpc_ips[1]
+
+        self.assertEqual(vpc_ip_2.ipv6_range, "fd71:1140:a9d0::/52")
+        self.assertEqual(vpc_ip_2.ipv6_is_public, True)
+        self.assertEqual(
+            vpc_ip_2.ipv6_addresses[0].slaac_address, "fd71:1140:a9d0::/52"
+        )