Add a test for Lock model