Fix tests

Author: zliang-akamai

test/integration/models/lock/test_lock.py

@@ -149,7 +149,7 @@ def test_delete_lock(test_linode_client, linode_for_lock):
 
     # Verify the lock no longer exists
     locks = test_linode_client.locks()
-    lock_ids = [l.id for l in locks]
+    lock_ids = [lk.id for lk in locks]
     assert lock_id not in lock_ids
 
 
@@ -173,7 +173,7 @@ def test_delete_lock_by_id(test_linode_client, linode_for_lock):
 
     # Verify the lock no longer exists
     locks = test_linode_client.locks()
-    lock_ids = [l.id for l in locks]
+    lock_ids = [lk.id for lk in locks]
     assert lock_id not in lock_ids
 
 
@@ -195,5 +195,5 @@ def test_lock_object_delete(test_linode_client, linode_for_lock):
 
     # Verify the lock no longer exists
     locks = test_linode_client.locks()
-    lock_ids = [l.id for l in locks]
+    lock_ids = [lk.id for lk in locks]
     assert lock_id not in lock_ids

test/unit/groups/lock_test.py

@@ -62,17 +62,22 @@ def test_create_lock_with_subresources(self):
                 m.call_data["lock_type"], "cannot_delete_with_subresources"
             )
 
+            self.assertEqual(lock.id, 1)
+            self.assertEqual(lock.lock_type, "cannot_delete")
+            self.assertIsNotNone(lock.entity)
+            self.assertEqual(lock.entity.id, 123)
+
     def test_create_lock_default_type(self):
         """
         Tests that creating a lock without specifying lock_type uses cannot_delete
         """
-        with self.mock_post("/locks/1") as m:
-            self.client.locks.create(
+        with self.mock_post("/locks/1"):
+            lock = self.client.locks.create(
                 entity_type="linode",
                 entity_id=123,
             )
 
-            self.assertEqual(m.call_data["lock_type"], "cannot_delete")
+            self.assertEqual(lock.lock_type, "cannot_delete")
 
     def test_delete_lock_by_object(self):
         """