Add resource lock support for NodeBalancer

Author: zliang-akamai

linode_api4/objects/nodebalancer.py

@@ -252,6 +252,7 @@ class NodeBalancer(Base):
         "transfer": Property(),
         "tags": Property(mutable=True, unordered=True),
         "client_udp_sess_throttle": Property(mutable=True),
+        "locks": Property(unordered=True),
     }
 
     # create derived objects

test/fixtures/nodebalancers.json

@@ -10,7 +10,8 @@
       "updated": "2018-01-01T00:01:01",
       "label": "balancer123456",
       "client_conn_throttle": 0,
-      "tags": ["something"]
+      "tags": ["something"],
+      "locks": ["cannot_delete_with_subresources"]
     },
     {
       "created": "2018-01-01T00:01:01",
@@ -22,7 +23,8 @@
       "updated": "2018-01-01T00:01:01",
       "label": "balancer123457",
       "client_conn_throttle": 0,
-      "tags": []
+      "tags": [],
+      "locks": []
     }
   ],
   "results": 2,

test/fixtures/nodebalancers_123456.json

@@ -10,5 +10,8 @@
   "client_conn_throttle": 0,
   "tags": [
     "something"
+  ],
+  "locks": [
+    "cannot_delete_with_subresources"
   ]
 }

test/unit/objects/nodebalancers_test.py

@@ -175,6 +175,24 @@ def test_update(self):
                 },
             )
 
+    def test_locks_not_in_put(self):
+        """
+        Test that locks are not included in PUT request when updating a NodeBalancer.
+        Locks are managed through the separate /v4/locks endpoint.
+        """
+        nb = NodeBalancer(self.client, 123456)
+        # Access locks to ensure it's loaded
+        self.assertEqual(nb.locks, ["cannot_delete_with_subresources"])
+
+        nb.label = "new-label"
+
+        with self.mock_put("nodebalancers/123456") as m:
+            nb.save()
+            self.assertEqual(m.call_url, "/nodebalancers/123456")
+            # Verify locks is NOT in the PUT data
+            self.assertNotIn("locks", m.call_data)
+            self.assertEqual(m.call_data["label"], "new-label")
+
     def test_firewalls(self):
         """
         Test that you can get the firewalls for the requested NodeBalancer.