import 15.2.0 Octopus source

[ceph.git] / ceph / src / test / librbd / mock / MockImageState.h

diff --git a/ceph/src/test/librbd/mock/MockImageState.h b/ceph/src/test/librbd/mock/MockImageState.h
index a817d333e1d5f5434678e803c14300b365628eac..b9c8c8d07590b02c3016a984b1fea95b0d742402 100644 (file)
--- a/ceph/src/test/librbd/mock/MockImageState.h
+++ b/ceph/src/test/librbd/mock/MockImageState.h
@@ -12,6 +12,8 @@ class Context;
 
 namespace librbd {
 
+class UpdateWatchCtx;
+
 struct MockImageState {
   MOCK_CONST_METHOD0(is_refresh_required, bool());
   MOCK_METHOD1(refresh, void(Context*));
@@ -25,6 +27,9 @@ struct MockImageState {
 
   MOCK_METHOD1(prepare_lock, void(Context*));
   MOCK_METHOD0(handle_prepare_lock_complete, void());
+
+  MOCK_METHOD2(register_update_watcher, int(UpdateWatchCtx *, uint64_t *));
+  MOCK_METHOD2(unregister_update_watcher, void(uint64_t, Context *));  
 };
 
 } // namespace librbd