projects / ceph.git / blame
| 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 | #ifndef CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H | |
| 5 | #define CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H | |
| 6 | ||
| 7 | #include "common/RWLock.h" | |
| 8 | #include "librbd/Utils.h" | |
| 9 | #include "gmock/gmock.h" | |
| 10 | ||
| 11 | namespace librbd { | |
| 12 | ||
| 13 | struct MockObjectMap { | |
| 14 | MOCK_CONST_METHOD1(enabled, bool(const RWLock &object_map_lock)); | |
| 15 | ||
| 16 | MOCK_METHOD1(open, void(Context *on_finish)); | |
| 17 | MOCK_METHOD1(close, void(Context *on_finish)); | |
| 18 | ||
| 19 | MOCK_METHOD3(aio_resize, void(uint64_t new_size, uint8_t default_object_state, | |
| 20 | Context *on_finish)); | |
| 21 | ||
| 22 | template <typename T, void(T::*MF)(int)> | |
| 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, | |
| 31f18b77 | 25 | const ZTracer::Trace &parent_trace, T *callback_object) { |
| 7c673cae | 26 | return aio_update<T, MF>(snap_id, start_object_no, start_object_no + 1, |
| 31f18b77 FG |
27 | new_state, current_state, parent_trace, |
| 28 | callback_object); | |
| 7c673cae FG |
29 | } |
| 30 | ||
| 31 | template <typename T, void(T::*MF)(int)> | |
| 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, | |
| 31f18b77 | 35 | const ZTracer::Trace &parent_trace, T *callback_object) { |
| 7c673cae | 36 | return aio_update(snap_id, start_object_no, end_object_no, new_state, |
| 31f18b77 | 37 | current_state, parent_trace, |
| 7c673cae FG |
38 | util::create_context_callback<T, MF>(callback_object)); |
| 39 | } | |
| 31f18b77 | 40 | MOCK_METHOD7(aio_update, bool(uint64_t snap_id, uint64_t start_object_no, |
| 7c673cae FG |
41 | uint64_t end_object_no, uint8_t new_state, |
| 42 | const boost::optional<uint8_t> ¤t_state, | |
| 31f18b77 | 43 | const ZTracer::Trace &parent_trace, |
| 7c673cae FG |
44 | Context *on_finish)); |
| 45 | MOCK_METHOD2(snapshot_add, void(uint64_t snap_id, Context *on_finish)); | |
| 46 | MOCK_METHOD2(snapshot_remove, void(uint64_t snap_id, Context *on_finish)); | |
| 47 | MOCK_METHOD2(rollback, void(uint64_t snap_id, Context *on_finish)); | |
| 48 | }; | |
| 49 | ||
| 50 | } // namespace librbd | |
| 51 | ||
| 52 | #endif // CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H |