]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- |
2 | // vim: ts=8 sw=2 smarttab | |
3 | ||
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/librados_test_stub/MockTestMemIoCtxImpl.h" | |
8 | #include "test/librados_test_stub/MockTestMemRadosClient.h" | |
9 | #include "cls/lock/cls_lock_ops.h" | |
10 | #include "librbd/managed_lock/GetLockerRequest.h" | |
11 | #include "librbd/managed_lock/Types.h" | |
12 | #include "librbd/managed_lock/Utils.h" | |
13 | #include "gmock/gmock.h" | |
14 | #include "gtest/gtest.h" | |
15 | #include <arpa/inet.h> | |
16 | #include <list> | |
17 | ||
18 | namespace librbd { | |
19 | namespace { | |
20 | ||
21 | struct MockTestImageCtx : public librbd::MockImageCtx { | |
22 | MockTestImageCtx(librbd::ImageCtx &image_ctx) | |
23 | : librbd::MockImageCtx(image_ctx) { | |
24 | } | |
25 | }; | |
26 | ||
27 | } // anonymous namespace | |
28 | } // namespace librbd | |
29 | ||
30 | // template definitions | |
31 | #include "librbd/managed_lock/GetLockerRequest.cc" | |
32 | ||
33 | namespace librbd { | |
34 | namespace managed_lock { | |
35 | ||
36 | using ::testing::_; | |
37 | using ::testing::DoAll; | |
38 | using ::testing::InSequence; | |
39 | using ::testing::Return; | |
40 | using ::testing::StrEq; | |
41 | using ::testing::WithArg; | |
42 | ||
43 | class TestMockManagedLockGetLockerRequest : public TestMockFixture { | |
44 | public: | |
45 | typedef GetLockerRequest<MockTestImageCtx> MockGetLockerRequest; | |
46 | ||
47 | void expect_get_lock_info(MockTestImageCtx &mock_image_ctx, int r, | |
48 | const entity_name_t &locker_entity, | |
49 | const std::string &locker_address, | |
50 | const std::string &locker_cookie, | |
51 | const std::string &lock_tag, | |
52 | ClsLockType lock_type) { | |
53 | auto &expect = EXPECT_CALL(get_mock_io_ctx(mock_image_ctx.md_ctx), | |
54 | exec(mock_image_ctx.header_oid, _, StrEq("lock"), | |
f67539c2 | 55 | StrEq("get_info"), _, _, _, _)); |
7c673cae FG |
56 | if (r < 0 && r != -ENOENT) { |
57 | expect.WillOnce(Return(r)); | |
58 | } else { | |
59 | entity_name_t entity(locker_entity); | |
60 | entity_addr_t entity_addr; | |
61 | entity_addr.parse(locker_address.c_str(), NULL); | |
62 | ||
63 | cls_lock_get_info_reply reply; | |
64 | if (r != -ENOENT) { | |
11fdf7f2 TL |
65 | reply.lockers.emplace( |
66 | rados::cls::lock::locker_id_t(entity, locker_cookie), | |
67 | rados::cls::lock::locker_info_t(utime_t(), entity_addr, "")); | |
7c673cae FG |
68 | reply.tag = lock_tag; |
69 | reply.lock_type = lock_type; | |
70 | } | |
71 | ||
72 | bufferlist bl; | |
11fdf7f2 | 73 | encode(reply, bl, CEPH_FEATURES_SUPPORTED_DEFAULT); |
7c673cae FG |
74 | |
75 | std::string str(bl.c_str(), bl.length()); | |
76 | expect.WillOnce(DoAll(WithArg<5>(CopyInBufferlist(str)), Return(0))); | |
77 | } | |
78 | } | |
79 | }; | |
80 | ||
81 | TEST_F(TestMockManagedLockGetLockerRequest, SuccessExclusive) { | |
82 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
83 | ||
84 | librbd::ImageCtx *ictx; | |
85 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
86 | ||
87 | MockTestImageCtx mock_image_ctx(*ictx); | |
88 | expect_op_work_queue(mock_image_ctx); | |
89 | ||
90 | InSequence seq; | |
91 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "1.2.3.4", | |
92 | "auto 123", util::get_watcher_lock_tag(), | |
f67539c2 | 93 | ClsLockType::EXCLUSIVE); |
7c673cae FG |
94 | |
95 | C_SaferCond ctx; | |
96 | Locker locker; | |
97 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
98 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
99 | req->send(); | |
100 | ASSERT_EQ(0, ctx.wait()); | |
101 | ||
102 | ASSERT_EQ(entity_name_t::CLIENT(1), locker.entity); | |
103 | ASSERT_EQ("1.2.3.4:0/0", locker.address); | |
104 | ASSERT_EQ("auto 123", locker.cookie); | |
105 | ASSERT_EQ(123U, locker.handle); | |
106 | } | |
107 | ||
108 | TEST_F(TestMockManagedLockGetLockerRequest, SuccessShared) { | |
109 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
110 | ||
111 | librbd::ImageCtx *ictx; | |
112 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
113 | ||
114 | MockTestImageCtx mock_image_ctx(*ictx); | |
115 | expect_op_work_queue(mock_image_ctx); | |
116 | ||
117 | InSequence seq; | |
118 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "1.2.3.4", | |
119 | "auto 123", util::get_watcher_lock_tag(), | |
f67539c2 | 120 | ClsLockType::SHARED); |
7c673cae FG |
121 | |
122 | C_SaferCond ctx; | |
123 | Locker locker; | |
124 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
125 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, false, &locker, &ctx); | |
126 | req->send(); | |
127 | ASSERT_EQ(0, ctx.wait()); | |
128 | ||
129 | ASSERT_EQ(entity_name_t::CLIENT(1), locker.entity); | |
130 | ASSERT_EQ("1.2.3.4:0/0", locker.address); | |
131 | ASSERT_EQ("auto 123", locker.cookie); | |
132 | ASSERT_EQ(123U, locker.handle); | |
133 | } | |
134 | ||
135 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoError) { | |
136 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
137 | ||
138 | librbd::ImageCtx *ictx; | |
139 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
140 | ||
141 | MockTestImageCtx mock_image_ctx(*ictx); | |
142 | expect_op_work_queue(mock_image_ctx); | |
143 | ||
144 | InSequence seq; | |
145 | expect_get_lock_info(mock_image_ctx, -EINVAL, entity_name_t::CLIENT(1), "", | |
f67539c2 | 146 | "", "", ClsLockType::EXCLUSIVE); |
7c673cae FG |
147 | |
148 | C_SaferCond ctx; | |
149 | Locker locker; | |
150 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
151 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
152 | req->send(); | |
153 | ASSERT_EQ(-EINVAL, ctx.wait()); | |
154 | } | |
155 | ||
156 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoEmpty) { | |
157 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
158 | ||
159 | librbd::ImageCtx *ictx; | |
160 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
161 | ||
162 | MockTestImageCtx mock_image_ctx(*ictx); | |
163 | expect_op_work_queue(mock_image_ctx); | |
164 | ||
165 | InSequence seq; | |
166 | expect_get_lock_info(mock_image_ctx, -ENOENT, entity_name_t::CLIENT(1), "", | |
f67539c2 | 167 | "", "", ClsLockType::EXCLUSIVE); |
7c673cae FG |
168 | |
169 | C_SaferCond ctx; | |
170 | Locker locker; | |
171 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
172 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
173 | req->send(); | |
174 | ASSERT_EQ(-ENOENT, ctx.wait()); | |
175 | } | |
176 | ||
177 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoExternalTag) { | |
178 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
179 | ||
180 | librbd::ImageCtx *ictx; | |
181 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
182 | ||
183 | MockTestImageCtx mock_image_ctx(*ictx); | |
184 | expect_op_work_queue(mock_image_ctx); | |
185 | ||
186 | InSequence seq; | |
187 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "1.2.3.4", | |
f67539c2 | 188 | "auto 123", "external tag", ClsLockType::EXCLUSIVE); |
7c673cae FG |
189 | |
190 | C_SaferCond ctx; | |
191 | Locker locker; | |
192 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
193 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
194 | req->send(); | |
195 | ASSERT_EQ(-EBUSY, ctx.wait()); | |
196 | } | |
197 | ||
198 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoIncompatibleShared) { | |
199 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
200 | ||
201 | librbd::ImageCtx *ictx; | |
202 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
203 | ||
204 | MockTestImageCtx mock_image_ctx(*ictx); | |
205 | expect_op_work_queue(mock_image_ctx); | |
206 | ||
207 | InSequence seq; | |
208 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "1.2.3.4", | |
209 | "auto 123", util::get_watcher_lock_tag(), | |
f67539c2 | 210 | ClsLockType::SHARED); |
7c673cae FG |
211 | |
212 | C_SaferCond ctx; | |
213 | Locker locker; | |
214 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
215 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
216 | req->send(); | |
217 | ASSERT_EQ(-EBUSY, ctx.wait()); | |
218 | } | |
219 | ||
220 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoIncompatibleExclusive) { | |
221 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
222 | ||
223 | librbd::ImageCtx *ictx; | |
224 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
225 | ||
226 | MockTestImageCtx mock_image_ctx(*ictx); | |
227 | expect_op_work_queue(mock_image_ctx); | |
228 | ||
229 | InSequence seq; | |
230 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "1.2.3.4", | |
231 | "auto 123", util::get_watcher_lock_tag(), | |
f67539c2 | 232 | ClsLockType::EXCLUSIVE); |
7c673cae FG |
233 | |
234 | C_SaferCond ctx; | |
235 | Locker locker; | |
236 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
237 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, false, &locker, &ctx); | |
238 | req->send(); | |
239 | ASSERT_EQ(-EBUSY, ctx.wait()); | |
240 | } | |
241 | ||
242 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoExternalCookie) { | |
243 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
244 | ||
245 | librbd::ImageCtx *ictx; | |
246 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
247 | ||
248 | MockTestImageCtx mock_image_ctx(*ictx); | |
249 | expect_op_work_queue(mock_image_ctx); | |
250 | ||
251 | InSequence seq; | |
252 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "1.2.3.4", | |
253 | "external cookie", util::get_watcher_lock_tag(), | |
f67539c2 | 254 | ClsLockType::EXCLUSIVE); |
7c673cae FG |
255 | |
256 | C_SaferCond ctx; | |
257 | Locker locker; | |
258 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
259 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
260 | req->send(); | |
261 | ASSERT_EQ(-EBUSY, ctx.wait()); | |
262 | } | |
263 | ||
1e59de90 TL |
264 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoEmptyCookie) { |
265 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
266 | ||
267 | librbd::ImageCtx *ictx; | |
268 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
269 | ||
270 | MockTestImageCtx mock_image_ctx(*ictx); | |
271 | expect_op_work_queue(mock_image_ctx); | |
272 | ||
273 | InSequence seq; | |
274 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "1.2.3.4", | |
275 | "", util::get_watcher_lock_tag(), | |
276 | ClsLockType::EXCLUSIVE); | |
277 | ||
278 | C_SaferCond ctx; | |
279 | Locker locker; | |
280 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
281 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
282 | req->send(); | |
283 | ASSERT_EQ(-EBUSY, ctx.wait()); | |
284 | } | |
285 | ||
286 | TEST_F(TestMockManagedLockGetLockerRequest, GetLockInfoBlankAddress) { | |
287 | REQUIRE_FEATURE(RBD_FEATURE_EXCLUSIVE_LOCK); | |
288 | ||
289 | librbd::ImageCtx *ictx; | |
290 | ASSERT_EQ(0, open_image(m_image_name, &ictx)); | |
291 | ||
292 | MockTestImageCtx mock_image_ctx(*ictx); | |
293 | expect_op_work_queue(mock_image_ctx); | |
294 | ||
295 | InSequence seq; | |
296 | expect_get_lock_info(mock_image_ctx, 0, entity_name_t::CLIENT(1), "", | |
297 | "auto 123", util::get_watcher_lock_tag(), | |
298 | ClsLockType::EXCLUSIVE); | |
299 | ||
300 | C_SaferCond ctx; | |
301 | Locker locker; | |
302 | MockGetLockerRequest *req = MockGetLockerRequest::create( | |
303 | mock_image_ctx.md_ctx, mock_image_ctx.header_oid, true, &locker, &ctx); | |
304 | req->send(); | |
305 | ASSERT_EQ(-EBUSY, ctx.wait()); | |
306 | } | |
307 | ||
7c673cae FG |
308 | } // namespace managed_lock |
309 | } // namespace librbd |