1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
2 // vim: ts=8 sw=2 smarttab
4 #include "test/librbd/test_fixture.h"
5 #include "test/librbd/test_support.h"
6 #include "include/rbd_types.h"
7 #include "librbd/MirroringWatcher.h"
8 #include "gtest/gtest.h"
9 #include "gmock/gmock.h"
12 void register_test_mirroring_watcher() {
19 struct MockMirroringWatcher : public MirroringWatcher<> {
22 MockMirroringWatcher(ImageCtx &image_ctx)
23 : MirroringWatcher<>(image_ctx.md_ctx, image_ctx.op_work_queue) {
26 MOCK_METHOD1(handle_mode_updated, void(cls::rbd::MirrorMode));
27 MOCK_METHOD3(handle_image_updated, void(cls::rbd::MirrorImageState,
29 const std::string &));
32 } // anonymous namespace
35 using ::testing::AtLeast;
36 using ::testing::Invoke;
37 using ::testing::StrEq;
38 using ::testing::WithArg;
40 class TestMirroringWatcher : public TestFixture {
42 void SetUp() override {
46 ASSERT_EQ(0, m_ioctx.write_full(RBD_MIRRORING, bl));
48 librbd::ImageCtx *ictx;
49 ASSERT_EQ(0, open_image(m_image_name, &ictx));
51 m_image_watcher = new MockMirroringWatcher(*ictx);
53 m_image_watcher->register_watch(&ctx);
54 if (ctx.wait() != 0) {
55 delete m_image_watcher;
56 m_image_watcher = nullptr;
61 void TearDown() override {
62 if (m_image_watcher != nullptr) {
64 m_image_watcher->unregister_watch(&ctx);
65 ASSERT_EQ(0, ctx.wait());
66 delete m_image_watcher;
69 TestFixture::TearDown();
72 MockMirroringWatcher *m_image_watcher = nullptr;
75 TEST_F(TestMirroringWatcher, ModeUpdated) {
76 EXPECT_CALL(*m_image_watcher,
77 handle_mode_updated(cls::rbd::MIRROR_MODE_DISABLED))
81 MockMirroringWatcher::notify_mode_updated(
82 m_ioctx, cls::rbd::MIRROR_MODE_DISABLED, &ctx);
83 ASSERT_EQ(0, ctx.wait());
86 TEST_F(TestMirroringWatcher, ImageStatusUpdated) {
87 EXPECT_CALL(*m_image_watcher,
88 handle_image_updated(cls::rbd::MIRROR_IMAGE_STATE_ENABLED,
90 StrEq("global image id")))
94 MockMirroringWatcher::notify_image_updated(
95 m_ioctx, cls::rbd::MIRROR_IMAGE_STATE_ENABLED, "image id",
96 "global image id", &ctx);
97 ASSERT_EQ(0, ctx.wait());
100 } // namespace librbd