ceph/src/test/librbd/mock/MockObjectMap.h

   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_METHOD1(at, uint8_t(uint64_t));
  15   uint8_t operator[](uint64_t object_no) {
  16     return at(object_no);
  17   }
  18
  19   MOCK_CONST_METHOD1(enabled, bool(const RWLock &object_map_lock));
  20
  21   MOCK_CONST_METHOD0(size, uint64_t());
  22
  23   MOCK_METHOD1(open, void(Context *on_finish));
  24   MOCK_METHOD1(close, void(Context *on_finish));
  25
  26   MOCK_METHOD3(aio_resize, void(uint64_t new_size, uint8_t default_object_state,
  27                                 Context *on_finish));
  28
  29   void get() {}
  30   void put() {}
  31
  32   template <typename T, void(T::*MF)(int) = &T::complete>
  33   bool aio_update(uint64_t snap_id, uint64_t start_object_no, uint8_t new_state,
  34                   const boost::optional<uint8_t> &current_state,
  35                   const ZTracer::Trace &parent_trace, bool ignore_enoent,
  36                   T *callback_object) {
  37     return aio_update<T, MF>(snap_id, start_object_no, start_object_no + 1,
  38                              new_state, current_state, parent_trace,
  39                              ignore_enoent, callback_object);
  40   }
  41
  42   template <typename T, void(T::*MF)(int) = &T::complete>
  43   bool aio_update(uint64_t snap_id, uint64_t start_object_no,
  44                   uint64_t end_object_no, uint8_t new_state,
  45                   const boost::optional<uint8_t> &current_state,
  46                   const ZTracer::Trace &parent_trace, bool ignore_enoent,
  47                   T *callback_object) {
  48     auto ctx = util::create_context_callback<T, MF>(callback_object);
  49     bool updated = aio_update(snap_id, start_object_no, end_object_no,
  50                               new_state, current_state, parent_trace,
  51                               ignore_enoent, ctx);
  52     if (!updated) {
  53       delete ctx;
  54     }
  55     return updated;
  56   }
  57   MOCK_METHOD8(aio_update, bool(uint64_t snap_id, uint64_t start_object_no,
  58                                 uint64_t end_object_no, uint8_t new_state,
  59                                 const boost::optional<uint8_t> &current_state,
  60                                 const ZTracer::Trace &parent_trace,
  61                                 bool ignore_enoent, Context *on_finish));
  62
  63   MOCK_METHOD2(snapshot_add, void(uint64_t snap_id, Context *on_finish));
  64   MOCK_METHOD2(snapshot_remove, void(uint64_t snap_id, Context *on_finish));
  65   MOCK_METHOD2(rollback, void(uint64_t snap_id, Context *on_finish));
  66
  67   MOCK_CONST_METHOD1(object_may_exist, bool(uint64_t));
  68
  69 };
  70
  71 } // namespace librbd
  72
  73 #endif // CEPH_TEST_LIBRBD_MOCK_OBJECT_MAP_H