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_mock_fixture.h"
5 #include "test/librbd/test_support.h"
6 #include "test/librbd/mock/MockImageCtx.h"
7 #include "test/journal/mock/MockJournaler.h"
8 #include "common/Mutex.h"
9 #include "cls/journal/cls_journal_types.h"
10 #include "librbd/journal/OpenRequest.h"
11 #include "librbd/journal/Types.h"
17 struct MockTestImageCtx : public MockImageCtx {
18 MockTestImageCtx(librbd::ImageCtx& image_ctx) : MockImageCtx(image_ctx) {
22 } // anonymous namespace
27 struct TypeTraits<MockTestImageCtx> {
28 typedef ::journal::MockJournaler Journaler;
31 } // namespace journal
34 // template definitions
35 #include "librbd/journal/OpenRequest.cc"
36 template class librbd::journal::OpenRequest<librbd::MockTestImageCtx>;
42 using ::testing::DoAll;
43 using ::testing::InSequence;
44 using ::testing::Return;
45 using ::testing::SetArgPointee;
46 using ::testing::WithArg;
48 class TestMockJournalOpenRequest : public TestMockFixture {
50 typedef OpenRequest<MockTestImageCtx> MockOpenRequest;
52 TestMockJournalOpenRequest() : m_lock("m_lock") {
55 void expect_init_journaler(::journal::MockJournaler &mock_journaler, int r) {
56 EXPECT_CALL(mock_journaler, init(_))
57 .WillOnce(CompleteContext(r, static_cast<ContextWQ*>(NULL)));
60 void expect_get_journaler_cached_client(::journal::MockJournaler &mock_journaler,
62 journal::ImageClientMeta image_client_meta;
63 image_client_meta.tag_class = 345;
65 journal::ClientData client_data;
66 client_data.client_meta = image_client_meta;
68 cls::journal::Client client;
69 ::encode(client_data, client.data);
71 EXPECT_CALL(mock_journaler, get_cached_client("", _))
72 .WillOnce(DoAll(SetArgPointee<1>(client),
76 void expect_get_journaler_tags(MockImageCtx &mock_image_ctx,
77 ::journal::MockJournaler &mock_journaler,
79 journal::TagData tag_data;
80 tag_data.mirror_uuid = "remote mirror";
82 bufferlist tag_data_bl;
83 ::encode(tag_data, tag_data_bl);
85 ::journal::Journaler::Tags tags = {{0, 345, {}}, {1, 345, tag_data_bl}};
86 EXPECT_CALL(mock_journaler, get_tags(345, _, _))
87 .WillOnce(DoAll(SetArgPointee<1>(tags),
88 WithArg<2>(CompleteContext(r, mock_image_ctx.image_ctx->op_work_queue))));
92 ImageClientMeta m_client_meta;
93 uint64_t m_tag_tid = 0;
97 TEST_F(TestMockJournalOpenRequest, Success) {
98 REQUIRE_FEATURE(RBD_FEATURE_JOURNALING);
100 librbd::ImageCtx *ictx;
101 ASSERT_EQ(0, open_image(m_image_name, &ictx));
103 MockTestImageCtx mock_image_ctx(*ictx);
104 ::journal::MockJournaler mock_journaler;
106 expect_op_work_queue(mock_image_ctx);
109 expect_init_journaler(mock_journaler, 0);
110 expect_get_journaler_cached_client(mock_journaler, 0);
111 expect_get_journaler_tags(mock_image_ctx, mock_journaler, 0);
114 auto req = MockOpenRequest::create(&mock_image_ctx, &mock_journaler,
115 &m_lock, &m_client_meta, &m_tag_tid,
118 ASSERT_EQ(0, ctx.wait());
119 ASSERT_EQ(345U, m_client_meta.tag_class);
120 ASSERT_EQ(1U, m_tag_tid);
121 ASSERT_EQ("remote mirror", m_tag_data.mirror_uuid);
124 TEST_F(TestMockJournalOpenRequest, InitError) {
125 REQUIRE_FEATURE(RBD_FEATURE_JOURNALING);
127 librbd::ImageCtx *ictx;
128 ASSERT_EQ(0, open_image(m_image_name, &ictx));
130 MockTestImageCtx mock_image_ctx(*ictx);
131 ::journal::MockJournaler mock_journaler;
133 expect_op_work_queue(mock_image_ctx);
136 expect_init_journaler(mock_journaler, -ENOENT);
139 auto req = MockOpenRequest::create(&mock_image_ctx, &mock_journaler,
140 &m_lock, &m_client_meta, &m_tag_tid,
143 ASSERT_EQ(-ENOENT, ctx.wait());
146 TEST_F(TestMockJournalOpenRequest, GetCachedClientError) {
147 REQUIRE_FEATURE(RBD_FEATURE_JOURNALING);
149 librbd::ImageCtx *ictx;
150 ASSERT_EQ(0, open_image(m_image_name, &ictx));
152 MockTestImageCtx mock_image_ctx(*ictx);
153 ::journal::MockJournaler mock_journaler;
155 expect_op_work_queue(mock_image_ctx);
158 expect_init_journaler(mock_journaler, 0);
159 expect_get_journaler_cached_client(mock_journaler, -EINVAL);
162 auto req = MockOpenRequest::create(&mock_image_ctx, &mock_journaler,
163 &m_lock, &m_client_meta, &m_tag_tid,
166 ASSERT_EQ(-EINVAL, ctx.wait());
169 TEST_F(TestMockJournalOpenRequest, GetTagsError) {
170 REQUIRE_FEATURE(RBD_FEATURE_JOURNALING);
172 librbd::ImageCtx *ictx;
173 ASSERT_EQ(0, open_image(m_image_name, &ictx));
175 MockTestImageCtx mock_image_ctx(*ictx);
176 ::journal::MockJournaler mock_journaler;
178 expect_op_work_queue(mock_image_ctx);
181 expect_init_journaler(mock_journaler, 0);
182 expect_get_journaler_cached_client(mock_journaler, 0);
183 expect_get_journaler_tags(mock_image_ctx, mock_journaler, -EBADMSG);
186 auto req = MockOpenRequest::create(&mock_image_ctx, &mock_journaler,
187 &m_lock, &m_client_meta, &m_tag_tid,
190 ASSERT_EQ(-EBADMSG, ctx.wait());
193 } // namespace journal
194 } // namespace librbd