Add enum

Author: zliang-akamai

linode_api4/groups/nodebalancer.py

@@ -1,6 +1,11 @@
 from linode_api4.errors import UnexpectedResponseError
 from linode_api4.groups import Group
-from linode_api4.objects import Base, NodeBalancer, NodeBalancerType
+from linode_api4.objects import (
+    Base,
+    NodeBalancer,
+    NodeBalancerBackendConnectivity,
+    NodeBalancerType,
+)
 
 
 class NodeBalancerGroup(Group):
@@ -48,7 +53,7 @@ def create(self, region, **kwargs):
                      ``undefined`` cannot be sent by clients. This cannot be
                      changed after creation.
                      NOTE: This field may not currently be available to all users.
-        :type backend_connectivity: str
+                :type backend_connectivity: NodeBalancerBackendConnectivity | str
         :param vpcs: VPC attachments for this NodeBalancer. Required when
                      ``backend_connectivity`` is ``vpc``.
         :type vpcs: list[dict]

linode_api4/objects/nodebalancer.py

@@ -7,6 +7,16 @@
 from linode_api4.objects.dbase import DerivedBase
 from linode_api4.objects.networking import Firewall, IPAddress
 from linode_api4.objects.region import Region
+from linode_api4.objects.serializable import StrEnum
+
+
+class NodeBalancerBackendConnectivity(StrEnum):
+    """Backend connectivity modes accepted when creating a NodeBalancer."""
+    
+    IPV6 = "ipv6"
+    LEGACY = "legacy"
+    UNDEFINED = "undefined"
+    VPC = "vpc"
 
 
 class NodeBalancerType(Base):

test/unit/linode_client_test.py

@@ -5,6 +5,7 @@
 from linode_api4.objects.beta import BetaProgram
 from linode_api4.objects.linode import Instance
 from linode_api4.objects.networking import IPAddress
+from linode_api4.objects.nodebalancer import NodeBalancerBackendConnectivity
 from linode_api4.objects.object_storage import (
     ObjectStorageACL,
     ObjectStorageCluster,
@@ -1607,7 +1608,7 @@ def test_create_with_type_and_backend_connectivity(self):
                 "us-east",
                 label="my-premium-nb",
                 type="premium",
-                backend_connectivity="ipv6",
+                backend_connectivity=NodeBalancerBackendConnectivity.IPV6,
                 configs=[
                     {
                         "port": 80,
@@ -1625,6 +1626,7 @@ def test_create_with_type_and_backend_connectivity(self):
             self.assertEqual(m.call_data["region"], "us-east")
             self.assertEqual(m.call_data["type"], "premium")
             self.assertEqual(m.call_data["backend_connectivity"], "ipv6")
+            self.assertIs(type(m.call_data["backend_connectivity"]), str)
             self.assertEqual(
                 m.call_data["configs"][0]["nodes"][0]["address"],
                 "[2001:db8:abcd:0012::1]:80",