@@ -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