Cleanup

Author: zliang-akamai

linode_api4/groups/account.py

@@ -11,7 +11,6 @@
     ChildAccount,
     Event,
     Invoice,
-    Lock,
     Login,
     MappedObject,
     OAuthClient,
@@ -511,58 +510,3 @@ def child_accounts(self, *filters):
         :rtype: PaginatedList of ChildAccount
         """
         return self.client._get_and_filter(ChildAccount, *filters)
-
-    def locks(self, *filters):
-        """
-        Returns a list of all resource locks on the account.
-
-        API Documentation: https://techdocs.akamai.com/linode-api/reference/get-locks
-
-        :param filters: Any number of filters to apply to this query.
-                        See :doc:`Filtering Collections</linode_api4/objects/filtering>`
-                        for more details on filtering.
-
-        :returns: A list of resource locks on the account.
-        :rtype: PaginatedList of Lock
-        """
-        return self.client._get_and_filter(Lock, *filters)
-
-    def lock_create(self, entity_type, entity_id, lock_type, **kwargs):
-        """
-        Creates a resource lock to prevent deletion or modification.
-
-        API Documentation: https://techdocs.akamai.com/linode-api/reference/post-lock
-
-        :param entity_type: The type of entity to lock (e.g., "linode").
-        :type entity_type: str
-        :param entity_id: The ID of the entity to lock.
-        :type entity_id: int
-        :param lock_type: The type of lock (e.g., "cannot_delete").
-        :type lock_type: str or LockType
-
-        :returns: The created resource lock.
-        :rtype: Lock
-        """
-        from linode_api4.objects.lock import (  # pylint: disable=import-outside-toplevel
-            LockType,
-        )
-
-        params = {
-            "entity_type": entity_type,
-            "entity_id": entity_id,
-            "lock_type": (
-                lock_type.value
-                if isinstance(lock_type, LockType)
-                else lock_type
-            ),
-        }
-        params.update(kwargs)
-
-        result = self.client.post("/locks", data=params)
-
-        if "id" not in result:
-            raise UnexpectedResponseError(
-                "Unexpected response when creating lock!", json=result
-            )
-
-        return Lock(self.client, result["id"], result)

test/unit/objects/lock_test.py

@@ -1,6 +1,6 @@
 from test.unit.base import ClientBaseCase
 
-from linode_api4.objects.lock import Lock, LockEntity, LockType
+from linode_api4.objects.lock import Lock, LockEntity
 
 
 class LockTest(ClientBaseCase):
@@ -22,38 +22,9 @@ def test_get_lock(self):
         self.assertEqual(lock.entity.label, "test-linode")
         self.assertEqual(lock.entity.url, "/v4/linode/instances/123")
 
-    def test_get_locks(self):
-        """
-        Tests that locks can be retrieved
-        """
-        locks = self.client.account.locks()
-
-        self.assertEqual(len(locks), 2)
-        self.assertEqual(locks[0].id, 1)
-        self.assertEqual(locks[0].lock_type, "cannot_delete")
-        self.assertEqual(locks[1].id, 2)
-        self.assertEqual(locks[1].lock_type, "cannot_delete_with_subresources")
-
-    def test_create_lock(self):
-        """
-        Tests that a lock can be created
-        """
-        with self.mock_post("/locks/1") as m:
-            lock = self.client.account.lock_create(
-                "linode", 123, LockType.cannot_delete
-            )
-
-            self.assertEqual(m.call_url, "/locks")
-            self.assertEqual(m.call_data["entity_type"], "linode")
-            self.assertEqual(m.call_data["entity_id"], 123)
-            self.assertEqual(m.call_data["lock_type"], "cannot_delete")
-
-            self.assertEqual(lock.id, 1)
-            self.assertEqual(lock.lock_type, "cannot_delete")
-
     def test_delete_lock(self):
         """
-        Tests that a lock can be deleted
+        Tests that a lock can be deleted using the Lock object's delete method
         """
         lock = Lock(self.client, 1)