1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
2 // vim: ts=8 sw=2 smarttab
4 #ifndef CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H
5 #define CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H
7 #include "common/RWLock.h"
8 #include "librbd/Utils.h"
9 #include "gmock/gmock.h"
13 struct MockObjectMap {
14 MOCK_CONST_METHOD1(enabled, bool(const RWLock &object_map_lock));
16 MOCK_METHOD1(open, void(Context *on_finish));
17 MOCK_METHOD1(close, void(Context *on_finish));
19 MOCK_METHOD3(aio_resize, void(uint64_t new_size, uint8_t default_object_state,
22 template <typename T, void(T::*MF)(int) = &T::complete>
23 bool aio_update(uint64_t snap_id, uint64_t start_object_no, uint8_t new_state,
24 const boost::optional<uint8_t> ¤t_state,
25 const ZTracer::Trace &parent_trace, T *callback_object) {
26 return aio_update<T, MF>(snap_id, start_object_no, start_object_no + 1,
27 new_state, current_state, parent_trace,
31 template <typename T, void(T::*MF)(int) = &T::complete>
32 bool aio_update(uint64_t snap_id, uint64_t start_object_no,
33 uint64_t end_object_no, uint8_t new_state,
34 const boost::optional<uint8_t> ¤t_state,
35 const ZTracer::Trace &parent_trace, T *callback_object) {
36 auto ctx = util::create_context_callback<T, MF>(callback_object);
37 bool updated = aio_update(snap_id, start_object_no, end_object_no,
38 new_state, current_state, parent_trace, ctx);
44 MOCK_METHOD7(aio_update, bool(uint64_t snap_id, uint64_t start_object_no,
45 uint64_t end_object_no, uint8_t new_state,
46 const boost::optional<uint8_t> ¤t_state,
47 const ZTracer::Trace &parent_trace,
50 MOCK_METHOD2(snapshot_add, void(uint64_t snap_id, Context *on_finish));
51 MOCK_METHOD2(snapshot_remove, void(uint64_t snap_id, Context *on_finish));
52 MOCK_METHOD2(rollback, void(uint64_t snap_id, Context *on_finish));
54 MOCK_CONST_METHOD1(object_may_exist, bool(uint64_t));
60 #endif // CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H