Skip to content

Commit ef5066f

Browse files
committed
Remove default lock type to match API schema; fix tests
1 parent 01d9674 commit ef5066f

File tree

2 files changed

+9
-23
lines changed

2 files changed

+9
-23
lines changed

linode_api4/groups/lock.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def create(
3939
self,
4040
entity_type: str,
4141
entity_id: Union[int, str],
42-
lock_type: Union[LockType, str] = LockType.cannot_delete,
42+
lock_type: Union[LockType, str],
4343
) -> Lock:
4444
"""
4545
Creates a new Resource Lock for the specified entity.

test/unit/groups/lock_test.py

Lines changed: 8 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,11 @@ def test_list_locks(self):
1616

1717
self.assertEqual(len(locks), 2)
1818
self.assertEqual(locks[0].id, 1)
19-
self.assertEqual(locks[0].lock_type, "cannot_delete")
19+
self.assertEqual(locks[0].lock_type, LockType.cannot_delete)
2020
self.assertEqual(locks[0].entity.id, 123)
2121
self.assertEqual(locks[0].entity.type, "linode")
2222
self.assertEqual(locks[1].id, 2)
23-
self.assertEqual(locks[1].lock_type, "cannot_delete_with_subresources")
23+
self.assertEqual(locks[1].lock_type, LockType.cannot_delete_with_subresources)
2424
self.assertEqual(locks[1].entity.id, 456)
2525

2626
def test_create_lock(self):
@@ -37,10 +37,10 @@ def test_create_lock(self):
3737
self.assertEqual(m.call_url, "/locks")
3838
self.assertEqual(m.call_data["entity_type"], "linode")
3939
self.assertEqual(m.call_data["entity_id"], 123)
40-
self.assertEqual(m.call_data["lock_type"], "cannot_delete")
40+
self.assertEqual(m.call_data["lock_type"], LockType.cannot_delete)
4141

4242
self.assertEqual(lock.id, 1)
43-
self.assertEqual(lock.lock_type, "cannot_delete")
43+
self.assertEqual(lock.lock_type, LockType.cannot_delete)
4444
self.assertIsNotNone(lock.entity)
4545
self.assertEqual(lock.entity.id, 123)
4646

@@ -52,27 +52,13 @@ def test_create_lock_with_subresources(self):
5252
lock = self.client.locks.create(
5353
entity_type="linode",
5454
entity_id=456,
55-
lock_type=LockType.cannot_delete,
55+
lock_type=LockType.cannot_delete_with_subresources,
5656
)
5757

5858
self.assertEqual(m.call_url, "/locks")
5959
self.assertEqual(m.call_data["entity_type"], "linode")
6060
self.assertEqual(m.call_data["entity_id"], 456)
61-
self.assertEqual(m.call_data["lock_type"], LockType.cannot_delete)
62-
63-
self.assertEqual(lock.id, 1)
64-
self.assertEqual(lock.lock_type, LockType.cannot_delete)
65-
self.assertIsNotNone(lock.entity)
66-
self.assertEqual(lock.entity.id, 123)
67-
68-
def test_create_lock_default_type(self):
69-
"""
70-
Tests that creating a lock without specifying lock_type uses cannot_delete
71-
"""
72-
with self.mock_post("/locks/1"):
73-
lock = self.client.locks.create(
74-
entity_type="linode",
75-
entity_id=123,
61+
self.assertEqual(
62+
m.call_data["lock_type"],
63+
LockType.cannot_delete_with_subresources,
7664
)
77-
78-
self.assertEqual(lock.lock_type, "cannot_delete")

0 commit comments

Comments
 (0)