summaryrefslogtreecommitdiffstats
path: root/src/test/librbd/mock/io/MockImageDispatcher.h
blob: 9e95f1d5cf8a48aca71dd9e8a56bb5f4a33ffee2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab

#ifndef CEPH_TEST_LIBRBD_MOCK_IO_IMAGE_DISPATCHER_H
#define CEPH_TEST_LIBRBD_MOCK_IO_IMAGE_DISPATCHER_H

#include "gmock/gmock.h"
#include "include/Context.h"
#include "librbd/io/ImageDispatcher.h"
#include "librbd/io/ImageDispatchSpec.h"
#include "librbd/io/Types.h"

class Context;

namespace librbd {
namespace io {

struct ImageDispatchInterface;

struct MockImageDispatcher : public ImageDispatcherInterface {
public:
  MOCK_METHOD1(shut_down, void(Context*));

  MOCK_METHOD1(register_dispatch, void(ImageDispatchInterface*));
  MOCK_METHOD1(exists, bool(ImageDispatchLayer));
  MOCK_METHOD2(shut_down_dispatch, void(ImageDispatchLayer, Context*));
  MOCK_METHOD1(invalidate_cache, void(Context *));

  MOCK_METHOD1(send, void(ImageDispatchSpec*));
  MOCK_METHOD3(finish, void(int r, ImageDispatchLayer, uint64_t));

  MOCK_METHOD1(apply_qos_schedule_tick_min, void(uint64_t));
  MOCK_METHOD4(apply_qos_limit, void(uint64_t, uint64_t, uint64_t, uint64_t));

  MOCK_CONST_METHOD0(writes_blocked, bool());
  MOCK_METHOD0(block_writes, int());
  MOCK_METHOD1(block_writes, void(Context*));

  MOCK_METHOD0(unblock_writes, void());
  MOCK_METHOD1(wait_on_writes_unblocked, void(Context*));

  MOCK_METHOD2(remap_extents, void(Extents&, ImageExtentsMapType));
};

} // namespace io
} // namespace librbd

#endif // CEPH_TEST_LIBRBD_MOCK_IO_IMAGE_DISPATCHER_H