]>
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 | ||
7c673cae FG |
7 | #include "librbd/Utils.h" |
8 | #include "gmock/gmock.h" | |
9 | ||
10 | namespace librbd { | |
11 | ||
12 | struct MockObjectMap { | |
81eedcae TL |
13 | MOCK_METHOD1(at, uint8_t(uint64_t)); |
14 | uint8_t operator[](uint64_t object_no) { | |
15 | return at(object_no); | |
16 | } | |
17 | ||
11fdf7f2 TL |
18 | MOCK_CONST_METHOD0(size, uint64_t()); |
19 | ||
7c673cae FG |
20 | MOCK_METHOD1(open, void(Context *on_finish)); |
21 | MOCK_METHOD1(close, void(Context *on_finish)); | |
22 | ||
23 | MOCK_METHOD3(aio_resize, void(uint64_t new_size, uint8_t default_object_state, | |
24 | Context *on_finish)); | |
25 | ||
9f95a23c TL |
26 | void get() {} |
27 | void put() {} | |
28 | ||
3efd9988 | 29 | template <typename T, void(T::*MF)(int) = &T::complete> |
7c673cae FG |
30 | bool aio_update(uint64_t snap_id, uint64_t start_object_no, uint8_t new_state, |
31 | const boost::optional<uint8_t> ¤t_state, | |
91327a77 AA |
32 | const ZTracer::Trace &parent_trace, bool ignore_enoent, |
33 | T *callback_object) { | |
7c673cae | 34 | return aio_update<T, MF>(snap_id, start_object_no, start_object_no + 1, |
31f18b77 | 35 | new_state, current_state, parent_trace, |
91327a77 | 36 | ignore_enoent, callback_object); |
7c673cae FG |
37 | } |
38 | ||
3efd9988 | 39 | template <typename T, void(T::*MF)(int) = &T::complete> |
7c673cae FG |
40 | bool aio_update(uint64_t snap_id, uint64_t start_object_no, |
41 | uint64_t end_object_no, uint8_t new_state, | |
42 | const boost::optional<uint8_t> ¤t_state, | |
91327a77 AA |
43 | const ZTracer::Trace &parent_trace, bool ignore_enoent, |
44 | T *callback_object) { | |
3efd9988 FG |
45 | auto ctx = util::create_context_callback<T, MF>(callback_object); |
46 | bool updated = aio_update(snap_id, start_object_no, end_object_no, | |
91327a77 AA |
47 | new_state, current_state, parent_trace, |
48 | ignore_enoent, ctx); | |
3efd9988 FG |
49 | if (!updated) { |
50 | delete ctx; | |
51 | } | |
52 | return updated; | |
7c673cae | 53 | } |
91327a77 | 54 | MOCK_METHOD8(aio_update, bool(uint64_t snap_id, uint64_t start_object_no, |
7c673cae FG |
55 | uint64_t end_object_no, uint8_t new_state, |
56 | const boost::optional<uint8_t> ¤t_state, | |
31f18b77 | 57 | const ZTracer::Trace &parent_trace, |
91327a77 | 58 | bool ignore_enoent, Context *on_finish)); |
3efd9988 | 59 | |
7c673cae FG |
60 | MOCK_METHOD2(snapshot_add, void(uint64_t snap_id, Context *on_finish)); |
61 | MOCK_METHOD2(snapshot_remove, void(uint64_t snap_id, Context *on_finish)); | |
62 | MOCK_METHOD2(rollback, void(uint64_t snap_id, Context *on_finish)); | |
3efd9988 FG |
63 | |
64 | MOCK_CONST_METHOD1(object_may_exist, bool(uint64_t)); | |
65 | ||
7c673cae FG |
66 | }; |
67 | ||
68 | } // namespace librbd | |
69 | ||
70 | #endif // CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H |