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
10namespace cc {
11namespace {
12
13class 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
22class IsOddPredicate {
23 public:
24  bool operator()(const Data* data) { return (data->data() % 2) == 1; }
25};
26
27TEST(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
45TEST(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(NULL, v2[0]);
74  EXPECT_EQ(NULL, v2[1]);
75  EXPECT_EQ(NULL, v2[2]);
76}
77
78TEST(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}  // namespace
105}  // namespace cc
106