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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
/*
* Ceph - scalable distributed file system
*
* Copyright (C) 2011 Greg Farnum <gregory.farnum@dreamhost.com>
*
* This is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License version 2.1, as published by the Free Software
* Foundation. See file COPYING.
*
*/
#include "include/Context.h"
#include "unit.h"
class C_Checker : public Context {
public:
bool *finish_called;
int *result;
C_Checker(bool* _finish_called, int *r) :
finish_called(_finish_called), result(r) {}
void finish(int r) { *finish_called = true; *result = r; }
};
TEST(ContextGather, Constructor) {
C_GatherBuilder gather(g_ceph_context);
EXPECT_FALSE(gather.has_subs());
EXPECT_TRUE(gather.get() == NULL);
}
TEST(ContextGather, OneSub) {
C_GatherBuilder gather(g_ceph_context);
Context *sub = gather.new_sub();
EXPECT_EQ(1, gather.num_subs_created());
EXPECT_EQ(1, gather.num_subs_remaining());
bool finish_called = false;
int result = 0;
C_Checker *checker = new C_Checker(&finish_called, &result);
gather.set_finisher(checker);
gather.activate();
sub->finish(0);
delete sub;
EXPECT_TRUE(finish_called);
EXPECT_EQ(0, result);
}
TEST(ContextGather, ManySubs) {
bool finish_called = false;
int result = 0;
C_GatherBuilder gather(g_ceph_context, new C_Checker(&finish_called, &result));
int sub_count = 8;
Context* subs[sub_count];
//create subs and test
for (int i = 0; i < sub_count; ++i) {
subs[i] = gather.new_sub();
EXPECT_EQ(i+1, gather.num_subs_created());
EXPECT_EQ(i+1, gather.num_subs_remaining());
}
EXPECT_TRUE(gather.has_subs());
gather.activate();
//finish all except one sub
for (int j = 0; j < sub_count - 1; ++j) {
subs[j]->finish(0);
delete subs[j];
EXPECT_FALSE(finish_called);
}
//finish last one and check asserts
subs[sub_count-1]->finish(0);
delete subs[sub_count-1];
EXPECT_TRUE(finish_called);
}
TEST(ContextGather, AlternatingSubCreateFinish) {
C_GatherBuilder gather(g_ceph_context);
int sub_count = 8;
bool finish_called = false;
int result = 0;
C_Checker *checker = new C_Checker(&finish_called, &result);
gather.set_finisher(checker);
Context* subs[sub_count];
//create half the subs
for (int i = 0; i < sub_count / 2; ++i) {
subs[i] = gather.new_sub();
EXPECT_EQ(i + 1, gather.num_subs_created());
EXPECT_EQ(i + 1, gather.num_subs_remaining());
}
//alternate finishing first half of subs and creating last half of subs
for (int j = 0; j < sub_count / 2; ++j) {
subs[j]->finish(0);
delete subs[j];
subs[sub_count / 2 + j] = gather.new_sub();
}
gather.activate();
//finish last half of subs
for (int k = sub_count / 2; k < sub_count; ++k) {
subs[k]->finish(0);
delete subs[k];
}
EXPECT_TRUE(finish_called);
}
|
