X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fceph%2Fsrc%2Ftest%2Frbd_mirror%2Ftest_mock_LeaderWatcher.cc;fp=src%2Fceph%2Fsrc%2Ftest%2Frbd_mirror%2Ftest_mock_LeaderWatcher.cc;h=2141a00f7a01d3d9df43240a5dca228b155056ba;hb=812ff6ca9fcd3e629e49d4328905f33eee8ca3f5;hp=0000000000000000000000000000000000000000;hpb=15280273faafb77777eab341909a3f495cf248d9;p=stor4nfv.git

diff --git a/src/ceph/src/test/rbd_mirror/test_mock_LeaderWatcher.cc b/src/ceph/src/test/rbd_mirror/test_mock_LeaderWatcher.cc
new file mode 100644
index 0000000..2141a00
--- /dev/null
+++ b/src/ceph/src/test/rbd_mirror/test_mock_LeaderWatcher.cc
@@ -0,0 +1,677 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#include "librbd/Utils.h"
+#include "test/librbd/mock/MockImageCtx.h"
+#include "test/rbd_mirror/test_mock_fixture.h"
+#include "tools/rbd_mirror/LeaderWatcher.h"
+#include "tools/rbd_mirror/Threads.h"
+
+using librbd::util::create_async_context_callback;
+
+namespace librbd {
+
+namespace {
+
+struct MockTestImageCtx : public MockImageCtx {
+  MockTestImageCtx(librbd::ImageCtx &image_ctx)
+    : librbd::MockImageCtx(image_ctx) {
+  }
+};
+
+} // anonymous namespace
+
+struct MockManagedLock {
+  static MockManagedLock *s_instance;
+  static MockManagedLock &get_instance() {
+    assert(s_instance != nullptr);
+    return *s_instance;
+  }
+
+  MockManagedLock() {
+    s_instance = this;
+  }
+
+  bool m_release_lock_on_shutdown = false;
+
+  MOCK_METHOD0(construct, void());
+  MOCK_METHOD0(destroy, void());
+
+  MOCK_CONST_METHOD0(is_lock_owner, bool());
+
+  MOCK_METHOD1(shut_down, void(Context *));
+  MOCK_METHOD1(try_acquire_lock, void(Context *));
+  MOCK_METHOD1(release_lock, void(Context *));
+  MOCK_METHOD3(break_lock, void(const managed_lock::Locker &, bool, Context *));
+  MOCK_METHOD2(get_locker, void(managed_lock::Locker *, Context *));
+
+  MOCK_METHOD0(set_state_post_acquiring, void());
+
+  MOCK_CONST_METHOD0(is_shutdown, bool());
+
+  MOCK_CONST_METHOD0(is_state_post_acquiring, bool());
+  MOCK_CONST_METHOD0(is_state_pre_releasing, bool());
+  MOCK_CONST_METHOD0(is_state_locked, bool());
+};
+
+MockManagedLock *MockManagedLock::s_instance = nullptr;
+
+template <>
+struct ManagedLock<MockTestImageCtx> {
+  ManagedLock(librados::IoCtx& ioctx, ContextWQ *work_queue,
+              const std::string& oid, librbd::Watcher *watcher,
+              managed_lock::Mode  mode, bool blacklist_on_break_lock,
+              uint32_t blacklist_expire_seconds)
+    : m_work_queue(work_queue), m_lock("ManagedLock::m_lock") {
+    MockManagedLock::get_instance().construct();
+  }
+
+  virtual ~ManagedLock() {
+    MockManagedLock::get_instance().destroy();
+  }
+
+  ContextWQ *m_work_queue;
+
+  mutable Mutex m_lock;
+
+  bool is_lock_owner() const {
+    return MockManagedLock::get_instance().is_lock_owner();
+  }
+
+  void shut_down(Context *on_shutdown) {
+    if (MockManagedLock::get_instance().m_release_lock_on_shutdown) {
+      on_shutdown = new FunctionContext(
+        [this, on_shutdown](int r) {
+          MockManagedLock::get_instance().m_release_lock_on_shutdown = false;
+          shut_down(on_shutdown);
+        });
+      release_lock(on_shutdown);
+      return;
+    }
+
+    MockManagedLock::get_instance().shut_down(on_shutdown);
+  }
+
+  void try_acquire_lock(Context *on_acquired) {
+    Context *post_acquire_ctx = create_async_context_callback(
+      m_work_queue, new FunctionContext(
+        [this, on_acquired](int r) {
+          post_acquire_lock_handler(r, on_acquired);
+        }));
+    MockManagedLock::get_instance().try_acquire_lock(post_acquire_ctx);
+  }
+
+  void release_lock(Context *on_released) {
+    Context *post_release_ctx = new FunctionContext(
+      [this, on_released](int r) {
+        post_release_lock_handler(false, r, on_released);
+      });
+
+    Context *release_ctx = new FunctionContext(
+      [this, on_released, post_release_ctx](int r) {
+        if (r < 0) {
+          on_released->complete(r);
+        } else {
+          MockManagedLock::get_instance().release_lock(post_release_ctx);
+        }
+      });
+
+    Context *pre_release_ctx = new FunctionContext(
+      [this, release_ctx](int r) {
+        bool shutting_down =
+          MockManagedLock::get_instance().m_release_lock_on_shutdown;
+        pre_release_lock_handler(shutting_down, release_ctx);
+      });
+
+    m_work_queue->queue(pre_release_ctx, 0);
+  }
+
+  void get_locker(managed_lock::Locker *locker, Context *on_finish) {
+    MockManagedLock::get_instance().get_locker(locker, on_finish);
+  }
+
+  void break_lock(const managed_lock::Locker &locker, bool force_break_lock,
+                  Context *on_finish) {
+    MockManagedLock::get_instance().break_lock(locker, force_break_lock,
+                                               on_finish);
+  }
+
+  void set_state_post_acquiring() {
+    MockManagedLock::get_instance().set_state_post_acquiring();
+  }
+
+  bool is_shutdown() const {
+    return MockManagedLock::get_instance().is_shutdown();
+  }
+
+  bool is_state_post_acquiring() const {
+    return MockManagedLock::get_instance().is_state_post_acquiring();
+  }
+
+  bool is_state_pre_releasing() const {
+    return MockManagedLock::get_instance().is_state_pre_releasing();
+  }
+
+  bool is_state_locked() const {
+    return MockManagedLock::get_instance().is_state_locked();
+  }
+
+  virtual void post_acquire_lock_handler(int r, Context *on_finish) = 0;
+  virtual void pre_release_lock_handler(bool shutting_down,
+                                        Context *on_finish) = 0;
+  virtual void post_release_lock_handler(bool shutting_down, int r,
+                                         Context *on_finish) = 0;
+};
+
+} // namespace librbd
+
+namespace rbd {
+namespace mirror {
+
+template <>
+struct Threads<librbd::MockTestImageCtx> {
+  Mutex &timer_lock;
+  SafeTimer *timer;
+  ContextWQ *work_queue;
+
+  Threads(Threads<librbd::ImageCtx> *threads)
+    : timer_lock(threads->timer_lock), timer(threads->timer),
+      work_queue(threads->work_queue) {
+  }
+};
+
+template <>
+struct MirrorStatusWatcher<librbd::MockTestImageCtx> {
+  static MirrorStatusWatcher* s_instance;
+
+  static MirrorStatusWatcher *create(librados::IoCtx &io_ctx,
+                                     ContextWQ *work_queue) {
+    assert(s_instance != nullptr);
+    return s_instance;
+  }
+
+  MirrorStatusWatcher() {
+    assert(s_instance == nullptr);
+    s_instance = this;
+  }
+
+  ~MirrorStatusWatcher() {
+    assert(s_instance == this);
+    s_instance = nullptr;
+  }
+
+  MOCK_METHOD0(destroy, void());
+  MOCK_METHOD1(init, void(Context *));
+  MOCK_METHOD1(shut_down, void(Context *));
+};
+
+MirrorStatusWatcher<librbd::MockTestImageCtx> *MirrorStatusWatcher<librbd::MockTestImageCtx>::s_instance = nullptr;
+
+template <>
+struct Instances<librbd::MockTestImageCtx> {
+  static Instances* s_instance;
+
+  static Instances *create(Threads<librbd::MockTestImageCtx> *threads,
+                           librados::IoCtx &ioctx) {
+    assert(s_instance != nullptr);
+    return s_instance;
+  }
+
+  Instances() {
+    assert(s_instance == nullptr);
+    s_instance = this;
+  }
+
+  ~Instances() {
+    assert(s_instance == this);
+    s_instance = nullptr;
+  }
+
+  MOCK_METHOD0(destroy, void());
+  MOCK_METHOD1(init, void(Context *));
+  MOCK_METHOD1(shut_down, void(Context *));
+  MOCK_METHOD1(notify, void(const std::string &));
+};
+
+Instances<librbd::MockTestImageCtx> *Instances<librbd::MockTestImageCtx>::s_instance = nullptr;
+
+} // namespace mirror
+} // namespace rbd
+
+
+// template definitions
+#include "tools/rbd_mirror/LeaderWatcher.cc"
+
+namespace rbd {
+namespace mirror {
+
+using ::testing::_;
+using ::testing::AtLeast;
+using ::testing::DoAll;
+using ::testing::InSequence;
+using ::testing::Invoke;
+using ::testing::Return;
+
+using librbd::MockManagedLock;
+
+struct MockListener : public LeaderWatcher<librbd::MockTestImageCtx>::Listener {
+  static MockListener* s_instance;
+
+  MockListener() {
+    assert(s_instance == nullptr);
+    s_instance = this;
+  }
+
+  ~MockListener() override {
+    assert(s_instance == this);
+    s_instance = nullptr;
+  }
+
+  MOCK_METHOD1(post_acquire_handler, void(Context *));
+  MOCK_METHOD1(pre_release_handler, void(Context *));
+
+  MOCK_METHOD1(update_leader_handler, void(const std::string &));
+};
+
+MockListener *MockListener::s_instance = nullptr;
+
+class TestMockLeaderWatcher : public TestMockFixture {
+public:
+  typedef MirrorStatusWatcher<librbd::MockTestImageCtx> MockMirrorStatusWatcher;
+  typedef Instances<librbd::MockTestImageCtx> MockInstances;
+  typedef LeaderWatcher<librbd::MockTestImageCtx> MockLeaderWatcher;
+  typedef Threads<librbd::MockTestImageCtx> MockThreads;
+
+  void SetUp() override {
+    TestMockFixture::SetUp();
+    m_mock_threads = new MockThreads(m_threads);
+  }
+
+  void TearDown() override {
+    delete m_mock_threads;
+    TestMockFixture::TearDown();
+  }
+
+  void expect_construct(MockManagedLock &mock_managed_lock) {
+    EXPECT_CALL(mock_managed_lock, construct());
+  }
+
+  void expect_destroy(MockManagedLock &mock_managed_lock) {
+    EXPECT_CALL(mock_managed_lock, destroy());
+  }
+
+  void expect_is_lock_owner(MockManagedLock &mock_managed_lock, bool owner) {
+    EXPECT_CALL(mock_managed_lock, is_lock_owner())
+      .WillOnce(Return(owner));
+  }
+
+  void expect_shut_down(MockManagedLock &mock_managed_lock,
+                        bool release_lock_on_shutdown, int r) {
+    mock_managed_lock.m_release_lock_on_shutdown = release_lock_on_shutdown;
+    EXPECT_CALL(mock_managed_lock, shut_down(_))
+      .WillOnce(CompleteContext(r));
+  }
+
+  void expect_try_acquire_lock(MockManagedLock &mock_managed_lock, int r) {
+    EXPECT_CALL(mock_managed_lock, try_acquire_lock(_))
+      .WillOnce(CompleteContext(r));
+    if (r == 0) {
+      expect_set_state_post_acquiring(mock_managed_lock);
+    }
+  }
+
+  void expect_release_lock(MockManagedLock &mock_managed_lock, int r,
+                           Context *on_finish = nullptr) {
+    EXPECT_CALL(mock_managed_lock, release_lock(_))
+      .WillOnce(Invoke([on_finish, r](Context *ctx) {
+                         ctx->complete(r);
+                         if (on_finish != nullptr) {
+                           on_finish->complete(0);
+                         }
+                       }));
+  }
+
+  void expect_get_locker(MockManagedLock &mock_managed_lock,
+                         const librbd::managed_lock::Locker &locker, int r) {
+    EXPECT_CALL(mock_managed_lock, get_locker(_, _))
+      .WillOnce(Invoke([r, locker](librbd::managed_lock::Locker *out,
+                                   Context *ctx) {
+                         if (r == 0) {
+                           *out = locker;
+                         }
+                         ctx->complete(r);
+                       }));
+  }
+
+  void expect_break_lock(MockManagedLock &mock_managed_lock,
+                         const librbd::managed_lock::Locker &locker, int r,
+                         Context *on_finish) {
+    EXPECT_CALL(mock_managed_lock, break_lock(locker, true, _))
+      .WillOnce(Invoke([on_finish, r](const librbd::managed_lock::Locker &,
+                                      bool, Context *ctx) {
+                         ctx->complete(r);
+                         on_finish->complete(0);
+                       }));
+  }
+
+  void expect_set_state_post_acquiring(MockManagedLock &mock_managed_lock) {
+    EXPECT_CALL(mock_managed_lock, set_state_post_acquiring());
+  }
+
+  void expect_is_shutdown(MockManagedLock &mock_managed_lock) {
+    EXPECT_CALL(mock_managed_lock, is_shutdown())
+      .Times(AtLeast(0)).WillRepeatedly(Return(false));
+  }
+
+  void expect_is_leader(MockManagedLock &mock_managed_lock, bool post_acquiring,
+                        bool locked) {
+    EXPECT_CALL(mock_managed_lock, is_state_post_acquiring())
+      .WillOnce(Return(post_acquiring));
+    if (!post_acquiring) {
+      EXPECT_CALL(mock_managed_lock, is_state_locked())
+        .WillOnce(Return(locked));
+    }
+  }
+
+  void expect_is_leader(MockManagedLock &mock_managed_lock) {
+    EXPECT_CALL(mock_managed_lock, is_state_post_acquiring())
+      .Times(AtLeast(0)).WillRepeatedly(Return(false));
+    EXPECT_CALL(mock_managed_lock, is_state_locked())
+      .Times(AtLeast(0)).WillRepeatedly(Return(false));
+    EXPECT_CALL(mock_managed_lock, is_state_pre_releasing())
+      .Times(AtLeast(0)).WillRepeatedly(Return(false));
+  }
+
+  void expect_notify_heartbeat(MockManagedLock &mock_managed_lock,
+                               Context *on_finish) {
+    // is_leader in notify_heartbeat
+    EXPECT_CALL(mock_managed_lock, is_state_post_acquiring())
+      .WillOnce(Return(false));
+    EXPECT_CALL(mock_managed_lock, is_state_locked())
+      .WillOnce(Return(true));
+
+    // is_leader in handle_notify_heartbeat
+    EXPECT_CALL(mock_managed_lock, is_state_post_acquiring())
+      .WillOnce(Return(false));
+    EXPECT_CALL(mock_managed_lock, is_state_locked())
+      .WillOnce(DoAll(Invoke([on_finish]() {
+                        on_finish->complete(0);
+                      }),
+                      Return(true)));
+  }
+
+  void expect_destroy(MockMirrorStatusWatcher &mock_mirror_status_watcher) {
+    EXPECT_CALL(mock_mirror_status_watcher, destroy());
+  }
+
+  void expect_init(MockMirrorStatusWatcher &mock_mirror_status_watcher, int r) {
+    EXPECT_CALL(mock_mirror_status_watcher, init(_))
+      .WillOnce(CompleteContext(m_mock_threads->work_queue, r));
+  }
+
+  void expect_shut_down(MockMirrorStatusWatcher &mock_mirror_status_watcher, int r) {
+    EXPECT_CALL(mock_mirror_status_watcher, shut_down(_))
+      .WillOnce(CompleteContext(m_mock_threads->work_queue, r));
+    expect_destroy(mock_mirror_status_watcher);
+  }
+
+  void expect_destroy(MockInstances &mock_instances) {
+    EXPECT_CALL(mock_instances, destroy());
+  }
+
+  void expect_init(MockInstances &mock_instances, int r) {
+    EXPECT_CALL(mock_instances, init(_))
+      .WillOnce(CompleteContext(m_mock_threads->work_queue, r));
+  }
+
+  void expect_shut_down(MockInstances &mock_instances, int r) {
+    EXPECT_CALL(mock_instances, shut_down(_))
+      .WillOnce(CompleteContext(m_mock_threads->work_queue, r));
+    expect_destroy(mock_instances);
+  }
+
+  void expect_acquire_notify(MockManagedLock &mock_managed_lock,
+                             MockListener &mock_listener, int r) {
+    expect_is_leader(mock_managed_lock, true, false);
+    EXPECT_CALL(mock_listener, post_acquire_handler(_))
+      .WillOnce(CompleteContext(r));
+    expect_is_leader(mock_managed_lock, true, false);
+  }
+
+  void expect_release_notify(MockManagedLock &mock_managed_lock,
+                             MockListener &mock_listener, int r) {
+    expect_is_leader(mock_managed_lock, false, false);
+    EXPECT_CALL(mock_listener, pre_release_handler(_))
+      .WillOnce(CompleteContext(r));
+    expect_is_leader(mock_managed_lock, false, false);
+  }
+
+  MockThreads *m_mock_threads;
+};
+
+TEST_F(TestMockLeaderWatcher, InitShutdown) {
+  MockManagedLock mock_managed_lock;
+  MockMirrorStatusWatcher mock_mirror_status_watcher;
+  MockInstances mock_instances;
+  MockListener listener;
+
+  expect_is_shutdown(mock_managed_lock);
+  expect_destroy(mock_managed_lock);
+
+  InSequence seq;
+
+  expect_construct(mock_managed_lock);
+  MockLeaderWatcher leader_watcher(m_mock_threads, m_local_io_ctx, &listener);
+
+  // Inint
+  C_SaferCond on_heartbeat_finish;
+  expect_is_leader(mock_managed_lock, false, false);
+  expect_try_acquire_lock(mock_managed_lock, 0);
+  expect_init(mock_mirror_status_watcher, 0);
+  expect_init(mock_instances, 0);
+  expect_acquire_notify(mock_managed_lock, listener, 0);
+  expect_notify_heartbeat(mock_managed_lock, &on_heartbeat_finish);
+
+  ASSERT_EQ(0, leader_watcher.init());
+  ASSERT_EQ(0, on_heartbeat_finish.wait());
+
+  // Shutdown
+  expect_release_notify(mock_managed_lock, listener, 0);
+  expect_shut_down(mock_instances, 0);
+  expect_shut_down(mock_mirror_status_watcher, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+  expect_release_lock(mock_managed_lock, 0);
+  expect_shut_down(mock_managed_lock, true, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+
+  leader_watcher.shut_down();
+}
+
+TEST_F(TestMockLeaderWatcher, InitReleaseShutdown) {
+  MockManagedLock mock_managed_lock;
+  MockMirrorStatusWatcher mock_mirror_status_watcher;
+  MockInstances mock_instances;
+  MockListener listener;
+
+  expect_is_shutdown(mock_managed_lock);
+  expect_destroy(mock_managed_lock);
+
+  InSequence seq;
+
+  expect_construct(mock_managed_lock);
+  MockLeaderWatcher leader_watcher(m_mock_threads, m_local_io_ctx, &listener);
+
+  // Inint
+  C_SaferCond on_heartbeat_finish;
+  expect_is_leader(mock_managed_lock, false, false);
+  expect_try_acquire_lock(mock_managed_lock, 0);
+  expect_init(mock_mirror_status_watcher, 0);
+  expect_init(mock_instances, 0);
+  expect_acquire_notify(mock_managed_lock, listener, 0);
+  expect_notify_heartbeat(mock_managed_lock, &on_heartbeat_finish);
+
+  ASSERT_EQ(0, leader_watcher.init());
+  ASSERT_EQ(0, on_heartbeat_finish.wait());
+
+  // Release
+  expect_is_leader(mock_managed_lock, false, true);
+  expect_release_notify(mock_managed_lock, listener, 0);
+  expect_shut_down(mock_instances, 0);
+  expect_shut_down(mock_mirror_status_watcher, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+  C_SaferCond on_release;
+  expect_release_lock(mock_managed_lock, 0, &on_release);
+
+  leader_watcher.release_leader();
+  ASSERT_EQ(0, on_release.wait());
+
+  // Shutdown
+  expect_shut_down(mock_managed_lock, false, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+
+  leader_watcher.shut_down();
+}
+
+TEST_F(TestMockLeaderWatcher, AcquireError) {
+  MockManagedLock mock_managed_lock;
+  MockMirrorStatusWatcher mock_mirror_status_watcher;
+  MockInstances mock_instances;
+  MockListener listener;
+
+  expect_is_shutdown(mock_managed_lock);
+  expect_is_leader(mock_managed_lock);
+  expect_destroy(mock_managed_lock);
+
+  InSequence seq;
+
+  expect_construct(mock_managed_lock);
+  MockLeaderWatcher leader_watcher(m_mock_threads, m_local_io_ctx, &listener);
+
+  // Inint
+  C_SaferCond on_heartbeat_finish;
+  expect_is_leader(mock_managed_lock, false, false);
+  expect_try_acquire_lock(mock_managed_lock, -EAGAIN);
+  expect_get_locker(mock_managed_lock, librbd::managed_lock::Locker(), -ENOENT);
+  expect_try_acquire_lock(mock_managed_lock, 0);
+  expect_init(mock_mirror_status_watcher, 0);
+  expect_init(mock_instances, 0);
+  expect_acquire_notify(mock_managed_lock, listener, 0);
+  expect_notify_heartbeat(mock_managed_lock, &on_heartbeat_finish);
+
+  ASSERT_EQ(0, leader_watcher.init());
+  ASSERT_EQ(0, on_heartbeat_finish.wait());
+
+  // Shutdown
+  expect_release_notify(mock_managed_lock, listener, 0);
+  expect_shut_down(mock_instances, 0);
+  expect_shut_down(mock_mirror_status_watcher, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+  expect_release_lock(mock_managed_lock, 0);
+  expect_shut_down(mock_managed_lock, true, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+
+  leader_watcher.shut_down();
+}
+
+TEST_F(TestMockLeaderWatcher, ReleaseError) {
+  MockManagedLock mock_managed_lock;
+  MockMirrorStatusWatcher mock_mirror_status_watcher;
+  MockInstances mock_instances;
+  MockListener listener;
+
+  expect_is_shutdown(mock_managed_lock);
+  expect_destroy(mock_managed_lock);
+
+  InSequence seq;
+
+  expect_construct(mock_managed_lock);
+  MockLeaderWatcher leader_watcher(m_mock_threads, m_local_io_ctx, &listener);
+
+  // Inint
+  C_SaferCond on_heartbeat_finish;
+  expect_is_leader(mock_managed_lock, false, false);
+  expect_try_acquire_lock(mock_managed_lock, 0);
+  expect_init(mock_mirror_status_watcher, 0);
+  expect_init(mock_instances, 0);
+  expect_acquire_notify(mock_managed_lock, listener, 0);
+  expect_notify_heartbeat(mock_managed_lock, &on_heartbeat_finish);
+
+  ASSERT_EQ(0, leader_watcher.init());
+  ASSERT_EQ(0, on_heartbeat_finish.wait());
+
+  // Release
+  expect_is_leader(mock_managed_lock, false, true);
+  expect_release_notify(mock_managed_lock, listener, -EINVAL);
+  expect_shut_down(mock_instances, 0);
+  expect_shut_down(mock_mirror_status_watcher, -EINVAL);
+  expect_is_leader(mock_managed_lock, false, false);
+  C_SaferCond on_release;
+  expect_release_lock(mock_managed_lock, -EINVAL, &on_release);
+
+  leader_watcher.release_leader();
+  ASSERT_EQ(0, on_release.wait());
+
+  // Shutdown
+  expect_shut_down(mock_managed_lock, false, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+
+  leader_watcher.shut_down();
+}
+
+TEST_F(TestMockLeaderWatcher, Break) {
+  EXPECT_EQ(0, _rados->conf_set("rbd_mirror_leader_heartbeat_interval", "1"));
+  EXPECT_EQ(0, _rados->conf_set("rbd_mirror_leader_max_missed_heartbeats",
+                                "1"));
+  CephContext *cct = reinterpret_cast<CephContext *>(m_local_io_ctx.cct());
+  int max_acquire_attempts = cct->_conf->get_val<int64_t>(
+    "rbd_mirror_leader_max_acquire_attempts_before_break");
+
+  MockManagedLock mock_managed_lock;
+  MockMirrorStatusWatcher mock_mirror_status_watcher;
+  MockInstances mock_instances;
+  MockListener listener;
+  librbd::managed_lock::Locker
+    locker{entity_name_t::CLIENT(1), "auto 123", "1.2.3.4:0/0", 123};
+
+  expect_is_shutdown(mock_managed_lock);
+  expect_is_leader(mock_managed_lock);
+  expect_destroy(mock_managed_lock);
+  EXPECT_CALL(listener, update_leader_handler(_));
+
+  InSequence seq;
+
+  expect_construct(mock_managed_lock);
+  MockLeaderWatcher leader_watcher(m_mock_threads, m_local_io_ctx, &listener);
+
+  // Init
+  expect_is_leader(mock_managed_lock, false, false);
+  for (int i = 0; i < max_acquire_attempts; i++) {
+    expect_try_acquire_lock(mock_managed_lock, -EAGAIN);
+    expect_get_locker(mock_managed_lock, locker, 0);
+  }
+  C_SaferCond on_break;
+  expect_break_lock(mock_managed_lock, locker, 0, &on_break);
+  C_SaferCond on_heartbeat_finish;
+  expect_try_acquire_lock(mock_managed_lock, 0);
+  expect_init(mock_mirror_status_watcher, 0);
+  expect_init(mock_instances, 0);
+  expect_acquire_notify(mock_managed_lock, listener, 0);
+  expect_notify_heartbeat(mock_managed_lock, &on_heartbeat_finish);
+
+  ASSERT_EQ(0, leader_watcher.init());
+  ASSERT_EQ(0, on_heartbeat_finish.wait());
+
+  // Shutdown
+  expect_release_notify(mock_managed_lock, listener, 0);
+  expect_shut_down(mock_instances, 0);
+  expect_shut_down(mock_mirror_status_watcher, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+  expect_release_lock(mock_managed_lock, 0);
+  expect_shut_down(mock_managed_lock, true, 0);
+  expect_is_leader(mock_managed_lock, false, false);
+
+  leader_watcher.shut_down();
+}
+
+} // namespace mirror
+} // namespace rbd