+++ /dev/null
-// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
-// vim: ts=8 sw=2 smarttab
-
-#include "test/rbd_mirror/test_mock_fixture.h"
-#include "include/rbd/librbd.hpp"
-#include "librbd/journal/Types.h"
-#include "librbd/journal/TypeTraits.h"
-#include "test/journal/mock/MockJournaler.h"
-#include "test/librados_test_stub/MockTestMemIoCtxImpl.h"
-#include "test/librbd/mock/MockImageCtx.h"
-#include "test/librbd/mock/MockObjectMap.h"
-#include "tools/rbd_mirror/ImageSync.h"
-#include "tools/rbd_mirror/Threads.h"
-#include "tools/rbd_mirror/image_sync/ImageCopyRequest.h"
-#include "tools/rbd_mirror/image_sync/SnapshotCopyRequest.h"
-#include "tools/rbd_mirror/image_sync/SyncPointCreateRequest.h"
-#include "tools/rbd_mirror/image_sync/SyncPointPruneRequest.h"
-
-namespace librbd {
-
-namespace {
-
-struct MockTestImageCtx : public librbd::MockImageCtx {
- MockTestImageCtx(librbd::ImageCtx &image_ctx)
- : librbd::MockImageCtx(image_ctx) {
- }
-};
-
-} // anonymous namespace
-
-namespace journal {
-
-template <>
-struct TypeTraits<librbd::MockTestImageCtx> {
- typedef ::journal::MockJournaler Journaler;
-};
-
-} // namespace journal
-} // namespace librbd
-
-// template definitions
-template class rbd::mirror::ImageSync<librbd::MockTestImageCtx>;
-#include "tools/rbd_mirror/ImageSync.cc"
-
-namespace rbd {
-namespace mirror {
-
-template<>
-struct InstanceWatcher<librbd::MockTestImageCtx> {
- MOCK_METHOD2(notify_sync_request, void(const std::string, Context *));
- MOCK_METHOD1(cancel_sync_request, bool(const std::string &));
- MOCK_METHOD1(notify_sync_complete, void(const std::string &));
-};
-
-namespace image_sync {
-
-template <>
-class ImageCopyRequest<librbd::MockTestImageCtx> {
-public:
- static ImageCopyRequest* s_instance;
- Context *on_finish;
-
- static ImageCopyRequest* create(librbd::MockTestImageCtx *local_image_ctx,
- librbd::MockTestImageCtx *remote_image_ctx,
- SafeTimer *timer, Mutex *timer_lock,
- journal::MockJournaler *journaler,
- librbd::journal::MirrorPeerClientMeta *client_meta,
- librbd::journal::MirrorPeerSyncPoint *sync_point,
- Context *on_finish,
- rbd::mirror::ProgressContext *progress_ctx = nullptr) {
- assert(s_instance != nullptr);
- s_instance->on_finish = on_finish;
- return s_instance;
- }
-
- ImageCopyRequest() {
- s_instance = this;
- }
-
- void put() {
- }
-
- void get() {
- }
-
- MOCK_METHOD0(cancel, void());
- MOCK_METHOD0(send, void());
-};
-
-template <>
-class SnapshotCopyRequest<librbd::MockTestImageCtx> {
-public:
- static SnapshotCopyRequest* s_instance;
- Context *on_finish;
-
- static SnapshotCopyRequest* create(librbd::MockTestImageCtx *local_image_ctx,
- librbd::MockTestImageCtx *remote_image_ctx,
- SnapshotCopyRequest<librbd::ImageCtx>::SnapMap *snap_map,
- journal::MockJournaler *journaler,
- librbd::journal::MirrorPeerClientMeta *client_meta,
- ContextWQ *work_queue,
- Context *on_finish) {
- assert(s_instance != nullptr);
- s_instance->on_finish = on_finish;
- return s_instance;
- }
-
- SnapshotCopyRequest() {
- s_instance = this;
- }
-
- void put() {
- }
-
- void get() {
- }
-
- MOCK_METHOD0(send, void());
- MOCK_METHOD0(cancel, void());
-};
-
-template <>
-class SyncPointCreateRequest<librbd::MockTestImageCtx> {
-public:
- static SyncPointCreateRequest *s_instance;
- Context *on_finish;
-
- static SyncPointCreateRequest* create(librbd::MockTestImageCtx *remote_image_ctx,
- const std::string &mirror_uuid,
- journal::MockJournaler *journaler,
- librbd::journal::MirrorPeerClientMeta *client_meta,
- Context *on_finish) {
- assert(s_instance != nullptr);
- s_instance->on_finish = on_finish;
- return s_instance;
- }
-
- SyncPointCreateRequest() {
- s_instance = this;
- }
- MOCK_METHOD0(send, void());
-};
-
-template <>
-class SyncPointPruneRequest<librbd::MockTestImageCtx> {
-public:
- static SyncPointPruneRequest *s_instance;
- Context *on_finish;
- bool sync_complete;
-
- static SyncPointPruneRequest* create(librbd::MockTestImageCtx *remote_image_ctx,
- bool sync_complete,
- journal::MockJournaler *journaler,
- librbd::journal::MirrorPeerClientMeta *client_meta,
- Context *on_finish) {
- assert(s_instance != nullptr);
- s_instance->on_finish = on_finish;
- s_instance->sync_complete = sync_complete;
- return s_instance;
- }
-
- SyncPointPruneRequest() {
- s_instance = this;
- }
- MOCK_METHOD0(send, void());
-};
-
-ImageCopyRequest<librbd::MockTestImageCtx>* ImageCopyRequest<librbd::MockTestImageCtx>::s_instance = nullptr;
-SnapshotCopyRequest<librbd::MockTestImageCtx>* SnapshotCopyRequest<librbd::MockTestImageCtx>::s_instance = nullptr;
-SyncPointCreateRequest<librbd::MockTestImageCtx>* SyncPointCreateRequest<librbd::MockTestImageCtx>::s_instance = nullptr;
-SyncPointPruneRequest<librbd::MockTestImageCtx>* SyncPointPruneRequest<librbd::MockTestImageCtx>::s_instance = nullptr;
-
-} // namespace image_sync
-
-using ::testing::_;
-using ::testing::InSequence;
-using ::testing::Invoke;
-using ::testing::Return;
-using ::testing::ReturnNew;
-using ::testing::WithArg;
-using ::testing::InvokeWithoutArgs;
-
-class TestMockImageSync : public TestMockFixture {
-public:
- typedef ImageSync<librbd::MockTestImageCtx> MockImageSync;
- typedef InstanceWatcher<librbd::MockTestImageCtx> MockInstanceWatcher;
- typedef image_sync::ImageCopyRequest<librbd::MockTestImageCtx> MockImageCopyRequest;
- typedef image_sync::SnapshotCopyRequest<librbd::MockTestImageCtx> MockSnapshotCopyRequest;
- typedef image_sync::SyncPointCreateRequest<librbd::MockTestImageCtx> MockSyncPointCreateRequest;
- typedef image_sync::SyncPointPruneRequest<librbd::MockTestImageCtx> MockSyncPointPruneRequest;
-
- void SetUp() override {
- TestMockFixture::SetUp();
-
- librbd::RBD rbd;
- ASSERT_EQ(0, create_image(rbd, m_remote_io_ctx, m_image_name, m_image_size));
- ASSERT_EQ(0, open_image(m_remote_io_ctx, m_image_name, &m_remote_image_ctx));
-
- ASSERT_EQ(0, create_image(rbd, m_local_io_ctx, m_image_name, m_image_size));
- ASSERT_EQ(0, open_image(m_local_io_ctx, m_image_name, &m_local_image_ctx));
- }
-
- void expect_start_op(librbd::MockExclusiveLock &mock_exclusive_lock) {
- EXPECT_CALL(mock_exclusive_lock, start_op()).WillOnce(
- ReturnNew<FunctionContext>([](int) {}));
- }
-
- void expect_notify_sync_request(MockInstanceWatcher &mock_instance_watcher,
- const std::string &sync_id, int r) {
- EXPECT_CALL(mock_instance_watcher, notify_sync_request(sync_id, _))
- .WillOnce(Invoke([this, r](const std::string &, Context *on_sync_start) {
- m_threads->work_queue->queue(on_sync_start, r);
- }));
- }
-
- void expect_cancel_sync_request(MockInstanceWatcher &mock_instance_watcher,
- const std::string &sync_id, bool canceled) {
- EXPECT_CALL(mock_instance_watcher, cancel_sync_request(sync_id))
- .WillOnce(Return(canceled));
- }
-
- void expect_notify_sync_complete(MockInstanceWatcher &mock_instance_watcher,
- const std::string &sync_id) {
- EXPECT_CALL(mock_instance_watcher, notify_sync_complete(sync_id));
- }
-
- void expect_create_sync_point(librbd::MockTestImageCtx &mock_local_image_ctx,
- MockSyncPointCreateRequest &mock_sync_point_create_request,
- int r) {
- EXPECT_CALL(mock_sync_point_create_request, send())
- .WillOnce(Invoke([this, &mock_local_image_ctx, &mock_sync_point_create_request, r]() {
- if (r == 0) {
- mock_local_image_ctx.snap_ids[{cls::rbd::UserSnapshotNamespace(),
- "snap1"}] = 123;
- m_client_meta.sync_points.emplace_back(cls::rbd::UserSnapshotNamespace(),
- "snap1",
- boost::none);
- }
- m_threads->work_queue->queue(mock_sync_point_create_request.on_finish, r);
- }));
- }
-
- void expect_copy_snapshots(MockSnapshotCopyRequest &mock_snapshot_copy_request, int r) {
- EXPECT_CALL(mock_snapshot_copy_request, send())
- .WillOnce(Invoke([this, &mock_snapshot_copy_request, r]() {
- m_threads->work_queue->queue(mock_snapshot_copy_request.on_finish, r);
- }));
- }
-
- void expect_copy_image(MockImageCopyRequest &mock_image_copy_request, int r) {
- EXPECT_CALL(mock_image_copy_request, send())
- .WillOnce(Invoke([this, &mock_image_copy_request, r]() {
- m_threads->work_queue->queue(mock_image_copy_request.on_finish, r);
- }));
- }
-
- void expect_rollback_object_map(librbd::MockObjectMap &mock_object_map, int r) {
- if ((m_local_image_ctx->features & RBD_FEATURE_OBJECT_MAP) != 0) {
- EXPECT_CALL(mock_object_map, rollback(_, _))
- .WillOnce(WithArg<1>(Invoke([this, r](Context *ctx) {
- m_threads->work_queue->queue(ctx, r);
- })));
- }
- }
-
- void expect_create_object_map(librbd::MockTestImageCtx &mock_image_ctx,
- librbd::MockObjectMap *mock_object_map) {
- EXPECT_CALL(mock_image_ctx, create_object_map(CEPH_NOSNAP))
- .WillOnce(Return(mock_object_map));
- }
-
- void expect_open_object_map(librbd::MockTestImageCtx &mock_image_ctx,
- librbd::MockObjectMap &mock_object_map) {
- EXPECT_CALL(mock_object_map, open(_))
- .WillOnce(Invoke([this](Context *ctx) {
- m_threads->work_queue->queue(ctx, 0);
- }));
- }
-
- void expect_prune_sync_point(MockSyncPointPruneRequest &mock_sync_point_prune_request,
- bool sync_complete, int r) {
- EXPECT_CALL(mock_sync_point_prune_request, send())
- .WillOnce(Invoke([this, &mock_sync_point_prune_request, sync_complete, r]() {
- ASSERT_EQ(sync_complete, mock_sync_point_prune_request.sync_complete);
- if (r == 0 && !m_client_meta.sync_points.empty()) {
- if (sync_complete) {
- m_client_meta.sync_points.pop_front();
- } else {
- while (m_client_meta.sync_points.size() > 1) {
- m_client_meta.sync_points.pop_back();
- }
- }
- }
- m_threads->work_queue->queue(mock_sync_point_prune_request.on_finish, r);
- }));
- }
-
- MockImageSync *create_request(librbd::MockTestImageCtx &mock_remote_image_ctx,
- librbd::MockTestImageCtx &mock_local_image_ctx,
- journal::MockJournaler &mock_journaler,
- MockInstanceWatcher &mock_instance_watcher,
- Context *ctx) {
- return new MockImageSync(&mock_local_image_ctx, &mock_remote_image_ctx,
- m_threads->timer, &m_threads->timer_lock,
- "mirror-uuid", &mock_journaler, &m_client_meta,
- m_threads->work_queue, &mock_instance_watcher,
- ctx);
- }
-
- librbd::ImageCtx *m_remote_image_ctx;
- librbd::ImageCtx *m_local_image_ctx;
- librbd::journal::MirrorPeerClientMeta m_client_meta;
-};
-
-TEST_F(TestMockImageSync, SimpleSync) {
- librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
- librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
- journal::MockJournaler mock_journaler;
- MockInstanceWatcher mock_instance_watcher;
- MockImageCopyRequest mock_image_copy_request;
- MockSnapshotCopyRequest mock_snapshot_copy_request;
- MockSyncPointCreateRequest mock_sync_point_create_request;
- MockSyncPointPruneRequest mock_sync_point_prune_request;
-
- librbd::MockExclusiveLock mock_exclusive_lock;
- mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
-
- librbd::MockObjectMap *mock_object_map = new librbd::MockObjectMap();
- mock_local_image_ctx.object_map = mock_object_map;
- expect_test_features(mock_local_image_ctx);
-
- InSequence seq;
- expect_notify_sync_request(mock_instance_watcher, mock_local_image_ctx.id, 0);
- expect_create_sync_point(mock_local_image_ctx, mock_sync_point_create_request, 0);
- expect_copy_snapshots(mock_snapshot_copy_request, 0);
- expect_copy_image(mock_image_copy_request, 0);
- expect_start_op(mock_exclusive_lock);
- expect_rollback_object_map(*mock_object_map, 0);
- expect_create_object_map(mock_local_image_ctx, mock_object_map);
- expect_open_object_map(mock_local_image_ctx, *mock_object_map);
- expect_prune_sync_point(mock_sync_point_prune_request, true, 0);
- expect_notify_sync_complete(mock_instance_watcher, mock_local_image_ctx.id);
-
- C_SaferCond ctx;
- MockImageSync *request = create_request(mock_remote_image_ctx,
- mock_local_image_ctx, mock_journaler,
- mock_instance_watcher, &ctx);
- request->send();
- ASSERT_EQ(0, ctx.wait());
-}
-
-TEST_F(TestMockImageSync, RestartSync) {
- librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
- librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
- journal::MockJournaler mock_journaler;
- MockInstanceWatcher mock_instance_watcher;
- MockImageCopyRequest mock_image_copy_request;
- MockSnapshotCopyRequest mock_snapshot_copy_request;
- MockSyncPointCreateRequest mock_sync_point_create_request;
- MockSyncPointPruneRequest mock_sync_point_prune_request;
-
- m_client_meta.sync_points = {{cls::rbd::UserSnapshotNamespace(), "snap1", boost::none},
- {cls::rbd::UserSnapshotNamespace(), "snap2", "snap1", boost::none}};
- mock_local_image_ctx.snap_ids[{cls::rbd::UserSnapshotNamespace(), "snap1"}] = 123;
- mock_local_image_ctx.snap_ids[{cls::rbd::UserSnapshotNamespace(), "snap2"}] = 234;
-
- librbd::MockExclusiveLock mock_exclusive_lock;
- mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
-
- librbd::MockObjectMap *mock_object_map = new librbd::MockObjectMap();
- mock_local_image_ctx.object_map = mock_object_map;
- expect_test_features(mock_local_image_ctx);
-
- InSequence seq;
- expect_notify_sync_request(mock_instance_watcher, mock_local_image_ctx.id, 0);
- expect_prune_sync_point(mock_sync_point_prune_request, false, 0);
- expect_copy_snapshots(mock_snapshot_copy_request, 0);
- expect_copy_image(mock_image_copy_request, 0);
- expect_start_op(mock_exclusive_lock);
- expect_rollback_object_map(*mock_object_map, 0);
- expect_create_object_map(mock_local_image_ctx, mock_object_map);
- expect_open_object_map(mock_local_image_ctx, *mock_object_map);
- expect_prune_sync_point(mock_sync_point_prune_request, true, 0);
- expect_notify_sync_complete(mock_instance_watcher, mock_local_image_ctx.id);
-
- C_SaferCond ctx;
- MockImageSync *request = create_request(mock_remote_image_ctx,
- mock_local_image_ctx, mock_journaler,
- mock_instance_watcher, &ctx);
- request->send();
- ASSERT_EQ(0, ctx.wait());
-}
-
-TEST_F(TestMockImageSync, CancelNotifySyncRequest) {
- librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
- librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
- journal::MockJournaler mock_journaler;
- MockInstanceWatcher mock_instance_watcher;
-
- InSequence seq;
- Context *on_sync_start = nullptr;
- C_SaferCond notify_sync_ctx;
- EXPECT_CALL(mock_instance_watcher,
- notify_sync_request(mock_local_image_ctx.id, _))
- .WillOnce(Invoke([this, &on_sync_start, ¬ify_sync_ctx](
- const std::string &, Context *ctx) {
- on_sync_start = ctx;
- notify_sync_ctx.complete(0);
- }));
- EXPECT_CALL(mock_instance_watcher,
- cancel_sync_request(mock_local_image_ctx.id))
- .WillOnce(Invoke([this, &on_sync_start](const std::string &) {
- EXPECT_NE(nullptr, on_sync_start);
- on_sync_start->complete(-ECANCELED);
- return true;
- }));
-
- C_SaferCond ctx;
- MockImageSync *request = create_request(mock_remote_image_ctx,
- mock_local_image_ctx, mock_journaler,
- mock_instance_watcher, &ctx);
- request->get();
- request->send();
-
- // cancel the notify sync request once it starts
- ASSERT_EQ(0, notify_sync_ctx.wait());
- request->cancel();
- request->put();
-
- ASSERT_EQ(-ECANCELED, ctx.wait());
-}
-
-TEST_F(TestMockImageSync, CancelImageCopy) {
- librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
- librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
- journal::MockJournaler mock_journaler;
- MockInstanceWatcher mock_instance_watcher;
- MockImageCopyRequest mock_image_copy_request;
- MockSnapshotCopyRequest mock_snapshot_copy_request;
- MockSyncPointCreateRequest mock_sync_point_create_request;
- MockSyncPointPruneRequest mock_sync_point_prune_request;
-
- librbd::MockExclusiveLock mock_exclusive_lock;
- mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
-
- m_client_meta.sync_points = {{cls::rbd::UserSnapshotNamespace(), "snap1", boost::none}};
-
- InSequence seq;
- expect_notify_sync_request(mock_instance_watcher, mock_local_image_ctx.id, 0);
- expect_prune_sync_point(mock_sync_point_prune_request, false, 0);
- expect_copy_snapshots(mock_snapshot_copy_request, 0);
-
- C_SaferCond image_copy_ctx;
- EXPECT_CALL(mock_image_copy_request, send())
- .WillOnce(Invoke([&image_copy_ctx]() {
- image_copy_ctx.complete(0);
- }));
- expect_cancel_sync_request(mock_instance_watcher, mock_local_image_ctx.id,
- false);
- EXPECT_CALL(mock_image_copy_request, cancel());
- expect_notify_sync_complete(mock_instance_watcher, mock_local_image_ctx.id);
-
- C_SaferCond ctx;
- MockImageSync *request = create_request(mock_remote_image_ctx,
- mock_local_image_ctx, mock_journaler,
- mock_instance_watcher, &ctx);
- request->get();
- request->send();
-
- // cancel the image copy once it starts
- ASSERT_EQ(0, image_copy_ctx.wait());
- request->cancel();
- request->put();
- m_threads->work_queue->queue(mock_image_copy_request.on_finish, 0);
-
- ASSERT_EQ(-ECANCELED, ctx.wait());
-}
-
-TEST_F(TestMockImageSync, CancelAfterCopySnapshots) {
- librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
- librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
- journal::MockJournaler mock_journaler;
- MockInstanceWatcher mock_instance_watcher;
- MockSnapshotCopyRequest mock_snapshot_copy_request;
- MockSyncPointCreateRequest mock_sync_point_create_request;
-
- librbd::MockExclusiveLock mock_exclusive_lock;
- mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
-
- librbd::MockObjectMap *mock_object_map = new librbd::MockObjectMap();
- mock_local_image_ctx.object_map = mock_object_map;
- expect_test_features(mock_local_image_ctx);
-
- C_SaferCond ctx;
- MockImageSync *request = create_request(mock_remote_image_ctx,
- mock_local_image_ctx, mock_journaler,
- mock_instance_watcher, &ctx);
- InSequence seq;
- expect_notify_sync_request(mock_instance_watcher, mock_local_image_ctx.id, 0);
- expect_create_sync_point(mock_local_image_ctx, mock_sync_point_create_request, 0);
- EXPECT_CALL(mock_snapshot_copy_request, send())
- .WillOnce((DoAll(InvokeWithoutArgs([request]() {
- request->cancel();
- }),
- Invoke([this, &mock_snapshot_copy_request]() {
- m_threads->work_queue->queue(mock_snapshot_copy_request.on_finish, 0);
- }))));
- expect_cancel_sync_request(mock_instance_watcher, mock_local_image_ctx.id,
- false);
- EXPECT_CALL(mock_snapshot_copy_request, cancel());
- expect_notify_sync_complete(mock_instance_watcher, mock_local_image_ctx.id);
-
- request->send();
- ASSERT_EQ(-ECANCELED, ctx.wait());
-}
-
-TEST_F(TestMockImageSync, CancelAfterCopyImage) {
- librbd::MockTestImageCtx mock_remote_image_ctx(*m_remote_image_ctx);
- librbd::MockTestImageCtx mock_local_image_ctx(*m_local_image_ctx);
- journal::MockJournaler mock_journaler;
- MockInstanceWatcher mock_instance_watcher;
- MockImageCopyRequest mock_image_copy_request;
- MockSnapshotCopyRequest mock_snapshot_copy_request;
- MockSyncPointCreateRequest mock_sync_point_create_request;
- MockSyncPointPruneRequest mock_sync_point_prune_request;
-
- librbd::MockExclusiveLock mock_exclusive_lock;
- mock_local_image_ctx.exclusive_lock = &mock_exclusive_lock;
-
- librbd::MockObjectMap *mock_object_map = new librbd::MockObjectMap();
- mock_local_image_ctx.object_map = mock_object_map;
- expect_test_features(mock_local_image_ctx);
-
- C_SaferCond ctx;
- MockImageSync *request = create_request(mock_remote_image_ctx,
- mock_local_image_ctx, mock_journaler,
- mock_instance_watcher, &ctx);
- InSequence seq;
- expect_notify_sync_request(mock_instance_watcher, mock_local_image_ctx.id, 0);
- expect_create_sync_point(mock_local_image_ctx, mock_sync_point_create_request, 0);
- expect_copy_snapshots(mock_snapshot_copy_request, 0);
- EXPECT_CALL(mock_image_copy_request, send())
- .WillOnce((DoAll(InvokeWithoutArgs([request]() {
- request->cancel();
- }),
- Invoke([this, &mock_image_copy_request]() {
- m_threads->work_queue->queue(mock_image_copy_request.on_finish, 0);
- }))));
- expect_cancel_sync_request(mock_instance_watcher, mock_local_image_ctx.id,
- false);
- EXPECT_CALL(mock_image_copy_request, cancel());
- expect_notify_sync_complete(mock_instance_watcher, mock_local_image_ctx.id);
-
- request->send();
- ASSERT_EQ(-ECANCELED, ctx.wait());
-}
-
-} // namespace mirror
-} // namespace rbd