1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
5 #include "gtest/gtest.h"
7 #include "global/global_context.h"
8 #include "global/global_init.h"
9 #include "common/common_init.h"
11 #include "osd/mClockOpClassQueue.h"
14 int main(int argc, char **argv) {
15 std::vector<const char*> args(argv, argv+argc);
16 auto cct = global_init(nullptr, args, CEPH_ENTITY_TYPE_OSD,
17 CODE_ENVIRONMENT_UTILITY,
18 CINIT_FLAG_NO_DEFAULT_CONFIG_FILE);
19 common_init_finish(g_ceph_context);
21 ::testing::InitGoogleTest(&argc, argv);
22 return RUN_ALL_TESTS();
26 class MClockOpClassQueueTest : public testing::Test {
30 entity_inst_t client1;
31 entity_inst_t client2;
32 entity_inst_t client3;
34 MClockOpClassQueueTest() :
36 client1(entity_name_t(CEPH_ENTITY_TYPE_OSD, 1), entity_addr_t()),
37 client2(entity_name_t(CEPH_ENTITY_TYPE_OSD, 2), entity_addr_t()),
38 client3(entity_name_t(CEPH_ENTITY_TYPE_CLIENT, 1), entity_addr_t())
41 #if 0 // more work needed here
42 Request create_client_op(epoch_t e, const entity_inst_t& owner) {
43 return Request(spg_t(), PGQueueable(OpRequestRef(), e));
47 Request create_snaptrim(epoch_t e, const entity_inst_t& owner) {
48 return Request(spg_t(),
49 PGQueueable(PGSnapTrim(e),
51 utime_t(), owner, e));
54 Request create_scrub(epoch_t e, const entity_inst_t& owner) {
55 return Request(spg_t(),
56 PGQueueable(PGScrub(e),
58 utime_t(), owner, e));
61 Request create_recovery(epoch_t e, const entity_inst_t& owner) {
62 return Request(spg_t(),
63 PGQueueable(PGRecovery(e, 64),
65 utime_t(), owner, e));
70 TEST_F(MClockOpClassQueueTest, TestSize) {
71 ASSERT_TRUE(q.empty());
72 ASSERT_EQ(0u, q.length());
74 q.enqueue(client1, 12, 0, create_snaptrim(100, client1));
75 q.enqueue_strict(client2, 12, create_snaptrim(101, client2));
76 q.enqueue(client2, 12, 0, create_snaptrim(102, client2));
77 q.enqueue_strict(client3, 12, create_snaptrim(103, client3));
78 q.enqueue(client1, 12, 0, create_snaptrim(104, client1));
80 ASSERT_FALSE(q.empty());
81 ASSERT_EQ(5u, q.length());
83 std::list<Request> reqs;
85 reqs.push_back(q.dequeue());
86 reqs.push_back(q.dequeue());
87 reqs.push_back(q.dequeue());
89 ASSERT_FALSE(q.empty());
90 ASSERT_EQ(2u, q.length());
92 q.enqueue_front(client2, 12, 0, reqs.back());
95 q.enqueue_strict_front(client3, 12, reqs.back());
98 q.enqueue_strict_front(client2, 12, reqs.back());
101 ASSERT_FALSE(q.empty());
102 ASSERT_EQ(5u, q.length());
104 for (int i = 0; i < 5; ++i) {
108 ASSERT_TRUE(q.empty());
109 ASSERT_EQ(0u, q.length());
113 TEST_F(MClockOpClassQueueTest, TestEnqueue) {
114 q.enqueue(client1, 12, 0, create_snaptrim(100, client1));
115 q.enqueue(client2, 12, 0, create_snaptrim(101, client2));
116 q.enqueue(client2, 12, 0, create_snaptrim(102, client2));
117 q.enqueue(client3, 12, 0, create_snaptrim(103, client3));
118 q.enqueue(client1, 12, 0, create_snaptrim(104, client1));
120 Request r = q.dequeue();
121 ASSERT_EQ(100u, r.second.get_map_epoch());
124 ASSERT_EQ(101u, r.second.get_map_epoch());
127 ASSERT_EQ(102u, r.second.get_map_epoch());
130 ASSERT_EQ(103u, r.second.get_map_epoch());
133 ASSERT_EQ(104u, r.second.get_map_epoch());
137 TEST_F(MClockOpClassQueueTest, TestEnqueueStrict) {
138 q.enqueue_strict(client1, 12, create_snaptrim(100, client1));
139 q.enqueue_strict(client2, 13, create_snaptrim(101, client2));
140 q.enqueue_strict(client2, 16, create_snaptrim(102, client2));
141 q.enqueue_strict(client3, 14, create_snaptrim(103, client3));
142 q.enqueue_strict(client1, 15, create_snaptrim(104, client1));
144 Request r = q.dequeue();
145 ASSERT_EQ(102u, r.second.get_map_epoch());
148 ASSERT_EQ(104u, r.second.get_map_epoch());
151 ASSERT_EQ(103u, r.second.get_map_epoch());
154 ASSERT_EQ(101u, r.second.get_map_epoch());
157 ASSERT_EQ(100u, r.second.get_map_epoch());
161 TEST_F(MClockOpClassQueueTest, TestRemoveByClass) {
162 q.enqueue(client1, 12, 0, create_snaptrim(100, client1));
163 q.enqueue_strict(client2, 12, create_snaptrim(101, client2));
164 q.enqueue(client2, 12, 0, create_snaptrim(102, client2));
165 q.enqueue_strict(client3, 12, create_snaptrim(103, client3));
166 q.enqueue(client1, 12, 0, create_snaptrim(104, client1));
168 std::list<Request> filtered_out;
169 q.remove_by_class(client2, &filtered_out);
171 ASSERT_EQ(2u, filtered_out.size());
172 while (!filtered_out.empty()) {
173 auto e = filtered_out.front().second.get_map_epoch() ;
174 ASSERT_TRUE(e == 101 || e == 102);
175 filtered_out.pop_front();
178 ASSERT_EQ(3u, q.length());
179 Request r = q.dequeue();
180 ASSERT_EQ(103u, r.second.get_map_epoch());
183 ASSERT_EQ(100u, r.second.get_map_epoch());
186 ASSERT_EQ(104u, r.second.get_map_epoch());