tests: unit test case for librbd::operation::ResizeRequest

Signed-off-by: Jason Dillaman <[email protected]>
jsaintro · Jan 15, 2016 · ce8fa3d · ce8fa3d
1 parent bf7a1f6
commit ce8fa3d
Show file tree

Hide file tree

Showing 6 changed files with 414 additions and 1 deletion.
diff --git a/src/test/Makefile-client.am b/src/test/Makefile-client.am
@@ -374,6 +374,7 @@ unittest_librbd_SOURCES = \
 	test/librbd/object_map/test_mock_SnapshotRollbackRequest.cc \
 	test/librbd/object_map/test_mock_UnlockRequest.cc \
 	test/librbd/object_map/test_mock_UpdateRequest.cc \
+	test/librbd/operation/test_mock_ResizeRequest.cc \
 	test/librbd/operation/test_mock_SnapshotCreateRequest.cc \
 	test/librbd/operation/test_mock_SnapshotProtectRequest.cc \
 	test/librbd/operation/test_mock_SnapshotRemoveRequest.cc \

diff --git a/src/test/librados_test_stub/MockTestMemIoCtxImpl.h b/src/test/librados_test_stub/MockTestMemIoCtxImpl.h
@@ -87,6 +87,13 @@ class MockTestMemIoCtxImpl : public TestMemIoCtxImpl {
     return TestMemIoCtxImpl::truncate(oid, size, snapc);
   }
 
+  MOCK_METHOD5(write, int(const std::string& oid, bufferlist& bl, size_t len,
+                          uint64_t off, const SnapContext &snapc));
+  int do_write(const std::string& oid, bufferlist& bl, size_t len, uint64_t off,
+                    const SnapContext &snapc) {
+    return TestMemIoCtxImpl::write(oid, bl, len, off, snapc);
+  }
+
   MOCK_METHOD3(write_full, int(const std::string& oid,
                                bufferlist& bl,
                                const SnapContext &snapc));
@@ -105,6 +112,7 @@ class MockTestMemIoCtxImpl : public TestMemIoCtxImpl {
     ON_CALL(*this, selfmanaged_snap_create(_)).WillByDefault(Invoke(this, &MockTestMemIoCtxImpl::do_selfmanaged_snap_create));
     ON_CALL(*this, selfmanaged_snap_remove(_)).WillByDefault(Invoke(this, &MockTestMemIoCtxImpl::do_selfmanaged_snap_remove));
     ON_CALL(*this, truncate(_,_,_)).WillByDefault(Invoke(this, &MockTestMemIoCtxImpl::do_truncate));
+    ON_CALL(*this, write(_, _, _, _, _)).WillByDefault(Invoke(this, &MockTestMemIoCtxImpl::do_write));
     ON_CALL(*this, write_full(_, _, _)).WillByDefault(Invoke(this, &MockTestMemIoCtxImpl::do_write_full));
   }
 

diff --git a/src/test/librbd/mock/MockImageCtx.h b/src/test/librbd/mock/MockImageCtx.h
@@ -18,6 +18,10 @@
 
 namespace librbd {
 
+namespace operation {
+template <typename> class ResizeRequest;
+}
+
 struct MockImageCtx {
   MockImageCtx(librbd::ImageCtx &image_ctx)
     : image_ctx(&image_ctx),
@@ -106,6 +110,7 @@ struct MockImageCtx {
   MOCK_METHOD1(flush, void(Context *));
   MOCK_METHOD1(flush_copyup, void(Context *));
 
+  MOCK_METHOD1(invalidate_cache, void(Context *));
   MOCK_METHOD1(shut_down_cache, void(Context *));
 
   MOCK_CONST_METHOD1(test_features, bool(uint64_t test_features));
@@ -145,6 +150,7 @@ struct MockImageCtx {
 
   ceph_file_layout layout;
 
+  xlist<operation::ResizeRequest<MockImageCtx>*> resize_reqs;
   xlist<AsyncRequest<MockImageCtx>*> async_requests;
   std::list<Context*> async_requests_waiters;
 

diff --git a/src/test/librbd/mock/MockJournal.h b/src/test/librbd/mock/MockJournal.h
@@ -20,7 +20,15 @@ struct MockJournal {
   MOCK_METHOD1(close, void(Context *));
 
   MOCK_METHOD0(allocate_op_tid, uint64_t());
-  MOCK_METHOD2(append_op_event, void(uint64_t, journal::EventEntry&));
+
+  MOCK_METHOD3(append_op_event_mock, void(uint64_t, const journal::EventEntry&,
+                                          Context *));
+  void append_op_event(uint64_t op_tid, journal::EventEntry &&event_entry,
+                       Context *on_safe) {
+    // googlemock doesn't support move semantics
+    append_op_event_mock(op_tid, event_entry, on_safe);
+  }
+
   MOCK_METHOD2(commit_op_event, void(uint64_t, int));
 };
 

diff --git a/src/test/librbd/mock/MockObjectMap.h b/src/test/librbd/mock/MockObjectMap.h
@@ -16,6 +16,9 @@ struct MockObjectMap {
   MOCK_METHOD1(lock, void(Context *on_finish));
   MOCK_METHOD1(unlock, void(Context *on_finish));
 
+  MOCK_METHOD3(aio_resize, void(uint64_t new_size, uint8_t default_object_state,
+                                Context *on_finish));
+
   MOCK_METHOD2(snapshot_add, void(uint64_t snap_id, Context *on_finish));
   MOCK_METHOD2(snapshot_remove, void(uint64_t snap_id, Context *on_finish));
 };