unittest: add mock method of try_acquire_lock

Signed-off-by: Dongsheng Yang <dongsheng.yang@easystack.cn>

diff --git a/src/test/librbd/mock/MockExclusiveLock.h b/src/test/librbd/mock/MockExclusiveLock.h
index 1b413becb0271491eb6009a8701b8943a1d8f9d9..023cd5edb5f03936e588e199ca7e8d9c1d0cba9d 100644 (file)
--- a/src/test/librbd/mock/MockExclusiveLock.h
+++ b/src/test/librbd/mock/MockExclusiveLock.h
@@ -19,6 +19,7 @@ struct MockExclusiveLock {
   MOCK_METHOD1(shut_down, void(Context*));
 
   MOCK_METHOD0(reacquire_lock, void());
+  MOCK_METHOD1(try_acquire_lock, void(Context*));
 
   MOCK_METHOD1(block_requests, void(int));
   MOCK_METHOD0(unblock_requests, void());

diff --git a/src/test/librbd/mock/MockImageCtx.h b/src/test/librbd/mock/MockImageCtx.h
index 1b2354df1563c440948f7171a81157244dcb4894..1b181bd5f343a5f30617a782d8cf507e679a28a4 100644 (file)
--- a/src/test/librbd/mock/MockImageCtx.h
+++ b/src/test/librbd/mock/MockImageCtx.h
@@ -36,6 +36,8 @@ struct MockImageCtx {
     assert(s_instance != nullptr);
     return s_instance;
   }
+  MOCK_METHOD0(destroy, void());
+
   MockImageCtx(librbd::ImageCtx &image_ctx)
     : image_ctx(&image_ctx),
       cct(image_ctx.cct),