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
|
// Copyright 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <set>
#include "cc/base/scoped_ptr_vector.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace cc {
namespace {
class Data {
public:
static scoped_ptr<Data> Create(int i) { return make_scoped_ptr(new Data(i)); }
int data() const { return data_; }
private:
explicit Data(int i) : data_(i) {}
int data_;
};
class IsOddPredicate {
public:
bool operator()(const Data* data) { return (data->data() % 2) == 1; }
};
TEST(ScopedPtrVectorTest, PushBack) {
ScopedPtrVector<Data> v;
// Insert 5 things into the vector.
v.push_back(Data::Create(1));
v.push_back(Data::Create(2));
v.push_back(Data::Create(3));
v.push_back(Data::Create(4));
v.push_back(Data::Create(5));
EXPECT_EQ(5u, v.size());
EXPECT_EQ(1, v[0]->data());
EXPECT_EQ(2, v[1]->data());
EXPECT_EQ(3, v[2]->data());
EXPECT_EQ(4, v[3]->data());
EXPECT_EQ(5, v[4]->data());
}
TEST(ScopedPtrVectorTest, InsertAndTake) {
// Insert 3 things into each vector.
ScopedPtrVector<Data> v;
v.push_back(Data::Create(1));
v.push_back(Data::Create(2));
v.push_back(Data::Create(6));
ScopedPtrVector<Data> v2;
v2.push_back(Data::Create(3));
v2.push_back(Data::Create(4));
v2.push_back(Data::Create(5));
ScopedPtrVector<Data>::iterator it = v.begin();
++it;
++it;
EXPECT_EQ(6, (*it)->data());
v.insert_and_take(it, v2);
EXPECT_EQ(6u, v.size());
EXPECT_EQ(1, v[0]->data());
EXPECT_EQ(2, v[1]->data());
EXPECT_EQ(3, v[2]->data());
EXPECT_EQ(4, v[3]->data());
EXPECT_EQ(5, v[4]->data());
EXPECT_EQ(6, v[5]->data());
EXPECT_EQ(3u, v2.size());
EXPECT_EQ(NULL, v2[0]);
EXPECT_EQ(NULL, v2[1]);
EXPECT_EQ(NULL, v2[2]);
}
TEST(ScopedPtrVectorTest, Partition) {
ScopedPtrVector<Data> v;
v.push_back(Data::Create(1));
v.push_back(Data::Create(2));
v.push_back(Data::Create(3));
v.push_back(Data::Create(4));
v.push_back(Data::Create(5));
ScopedPtrVector<Data>::iterator it = v.partition(IsOddPredicate());
std::set<int> odd_numbers;
for (ScopedPtrVector<Data>::iterator second_it = v.begin();
second_it != it;
++second_it) {
EXPECT_EQ(1, (*second_it)->data() % 2);
odd_numbers.insert((*second_it)->data());
}
EXPECT_EQ(3u, odd_numbers.size());
std::set<int> even_numbers;
for (; it != v.end(); ++it) {
EXPECT_EQ(0, (*it)->data() % 2);
even_numbers.insert((*it)->data());
}
EXPECT_EQ(2u, even_numbers.size());
}
} // namespace
} // namespace cc
|
