OLD | NEW |
| (Empty) |
1 // Copyright 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include <set> | |
6 | |
7 #include "cc/base/scoped_ptr_vector.h" | |
8 #include "testing/gmock/include/gmock/gmock.h" | |
9 | |
10 namespace cc { | |
11 namespace { | |
12 | |
13 class Data { | |
14 public: | |
15 static scoped_ptr<Data> Create(int i) { return make_scoped_ptr(new Data(i)); } | |
16 int data() const { return data_; } | |
17 private: | |
18 explicit Data(int i) : data_(i) {} | |
19 int data_; | |
20 }; | |
21 | |
22 class IsOddPredicate { | |
23 public: | |
24 bool operator()(const Data* data) { return (data->data() % 2) == 1; } | |
25 }; | |
26 | |
27 TEST(ScopedPtrVectorTest, PushBack) { | |
28 ScopedPtrVector<Data> v; | |
29 | |
30 // Insert 5 things into the vector. | |
31 v.push_back(Data::Create(1)); | |
32 v.push_back(Data::Create(2)); | |
33 v.push_back(Data::Create(3)); | |
34 v.push_back(Data::Create(4)); | |
35 v.push_back(Data::Create(5)); | |
36 | |
37 EXPECT_EQ(5u, v.size()); | |
38 EXPECT_EQ(1, v[0]->data()); | |
39 EXPECT_EQ(2, v[1]->data()); | |
40 EXPECT_EQ(3, v[2]->data()); | |
41 EXPECT_EQ(4, v[3]->data()); | |
42 EXPECT_EQ(5, v[4]->data()); | |
43 } | |
44 | |
45 TEST(ScopedPtrVectorTest, InsertAndTake) { | |
46 // Insert 3 things into each vector. | |
47 ScopedPtrVector<Data> v; | |
48 v.push_back(Data::Create(1)); | |
49 v.push_back(Data::Create(2)); | |
50 v.push_back(Data::Create(6)); | |
51 | |
52 ScopedPtrVector<Data> v2; | |
53 v2.push_back(Data::Create(3)); | |
54 v2.push_back(Data::Create(4)); | |
55 v2.push_back(Data::Create(5)); | |
56 | |
57 ScopedPtrVector<Data>::iterator it = v.begin(); | |
58 ++it; | |
59 ++it; | |
60 EXPECT_EQ(6, (*it)->data()); | |
61 | |
62 v.insert_and_take(it, &v2); | |
63 | |
64 EXPECT_EQ(6u, v.size()); | |
65 EXPECT_EQ(1, v[0]->data()); | |
66 EXPECT_EQ(2, v[1]->data()); | |
67 EXPECT_EQ(3, v[2]->data()); | |
68 EXPECT_EQ(4, v[3]->data()); | |
69 EXPECT_EQ(5, v[4]->data()); | |
70 EXPECT_EQ(6, v[5]->data()); | |
71 | |
72 EXPECT_EQ(3u, v2.size()); | |
73 EXPECT_EQ(nullptr, v2[0]); | |
74 EXPECT_EQ(nullptr, v2[1]); | |
75 EXPECT_EQ(nullptr, v2[2]); | |
76 } | |
77 | |
78 TEST(ScopedPtrVectorTest, Partition) { | |
79 ScopedPtrVector<Data> v; | |
80 v.push_back(Data::Create(1)); | |
81 v.push_back(Data::Create(2)); | |
82 v.push_back(Data::Create(3)); | |
83 v.push_back(Data::Create(4)); | |
84 v.push_back(Data::Create(5)); | |
85 | |
86 ScopedPtrVector<Data>::iterator it = v.partition(IsOddPredicate()); | |
87 std::set<int> odd_numbers; | |
88 for (ScopedPtrVector<Data>::iterator second_it = v.begin(); | |
89 second_it != it; | |
90 ++second_it) { | |
91 EXPECT_EQ(1, (*second_it)->data() % 2); | |
92 odd_numbers.insert((*second_it)->data()); | |
93 } | |
94 EXPECT_EQ(3u, odd_numbers.size()); | |
95 | |
96 std::set<int> even_numbers; | |
97 for (; it != v.end(); ++it) { | |
98 EXPECT_EQ(0, (*it)->data() % 2); | |
99 even_numbers.insert((*it)->data()); | |
100 } | |
101 EXPECT_EQ(2u, even_numbers.size()); | |
102 } | |
103 | |
104 class DataWithDestruction { | |
105 public: | |
106 static scoped_ptr<DataWithDestruction> Create(int i, int* destroy_count) { | |
107 return make_scoped_ptr(new DataWithDestruction(i, destroy_count)); | |
108 } | |
109 int data() const { return data_; } | |
110 ~DataWithDestruction() { ++(*destroy_count_); } | |
111 | |
112 private: | |
113 explicit DataWithDestruction(int i, int* destroy_count) | |
114 : data_(i), destroy_count_(destroy_count) {} | |
115 int data_; | |
116 int* destroy_count_; | |
117 }; | |
118 | |
119 TEST(ScopedPtrVectorTest, RemoveIf) { | |
120 ScopedPtrVector<DataWithDestruction> v; | |
121 int destroyed[6] = {0}; | |
122 v.push_back(DataWithDestruction::Create(1, &destroyed[0])); | |
123 v.push_back(DataWithDestruction::Create(2, &destroyed[1])); | |
124 v.push_back(DataWithDestruction::Create(3, &destroyed[2])); | |
125 v.push_back(DataWithDestruction::Create(3, &destroyed[3])); | |
126 v.push_back(DataWithDestruction::Create(4, &destroyed[4])); | |
127 v.push_back(DataWithDestruction::Create(5, &destroyed[5])); | |
128 | |
129 int expect_destroyed[6] = {0}; | |
130 | |
131 // Removing more than one thing that matches. | |
132 auto is_three = [](DataWithDestruction* d) { return d->data() == 3; }; | |
133 v.erase(v.remove_if(is_three), v.end()); | |
134 EXPECT_EQ(4u, v.size()); | |
135 expect_destroyed[2]++; | |
136 expect_destroyed[3]++; | |
137 for (size_t i = 0; i < arraysize(destroyed); ++i) | |
138 EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; | |
139 { | |
140 int expect_data[4] = {1, 2, 4, 5}; | |
141 for (size_t i = 0; i < arraysize(expect_data); ++i) | |
142 EXPECT_EQ(expect_data[i], v[i]->data()) << i; | |
143 } | |
144 | |
145 // Removing from the back of the vector. | |
146 auto is_five = [](DataWithDestruction* d) { return d->data() == 5; }; | |
147 v.erase(v.remove_if(is_five), v.end()); | |
148 EXPECT_EQ(3u, v.size()); | |
149 expect_destroyed[5]++; | |
150 for (size_t i = 0; i < arraysize(destroyed); ++i) | |
151 EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; | |
152 { | |
153 int expect_data[3] = {1, 2, 4}; | |
154 for (size_t i = 0; i < arraysize(expect_data); ++i) | |
155 EXPECT_EQ(expect_data[i], v[i]->data()) << i; | |
156 } | |
157 | |
158 // Removing from the front of the vector. | |
159 auto is_one = [](DataWithDestruction* d) { return d->data() == 1; }; | |
160 v.erase(v.remove_if(is_one), v.end()); | |
161 EXPECT_EQ(2u, v.size()); | |
162 expect_destroyed[0]++; | |
163 for (size_t i = 0; i < arraysize(destroyed); ++i) | |
164 EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; | |
165 { | |
166 int expect_data[2] = {2, 4}; | |
167 for (size_t i = 0; i < arraysize(expect_data); ++i) | |
168 EXPECT_EQ(expect_data[i], v[i]->data()) << i; | |
169 } | |
170 | |
171 // Removing things that aren't in the vector does nothing. | |
172 v.erase(v.remove_if(is_one), v.end()); | |
173 EXPECT_EQ(2u, v.size()); | |
174 for (size_t i = 0; i < arraysize(destroyed); ++i) | |
175 EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; | |
176 { | |
177 int expect_data[2] = {2, 4}; | |
178 for (size_t i = 0; i < arraysize(expect_data); ++i) | |
179 EXPECT_EQ(expect_data[i], v[i]->data()) << i; | |
180 } | |
181 } | |
182 | |
183 } // namespace | |
184 } // namespace cc | |
OLD | NEW |