Index: cc/base/scoped_ptr_vector_unittest.cc |
diff --git a/cc/base/scoped_ptr_vector_unittest.cc b/cc/base/scoped_ptr_vector_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7b9a86bd24183a25c29a0cf65f3bf35ffddebcaf |
--- /dev/null |
+++ b/cc/base/scoped_ptr_vector_unittest.cc |
@@ -0,0 +1,184 @@ |
+// 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(nullptr, v2[0]); |
+ EXPECT_EQ(nullptr, v2[1]); |
+ EXPECT_EQ(nullptr, 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()); |
+} |
+ |
+class DataWithDestruction { |
+ public: |
+ static scoped_ptr<DataWithDestruction> Create(int i, int* destroy_count) { |
+ return make_scoped_ptr(new DataWithDestruction(i, destroy_count)); |
+ } |
+ int data() const { return data_; } |
+ ~DataWithDestruction() { ++(*destroy_count_); } |
+ |
+ private: |
+ explicit DataWithDestruction(int i, int* destroy_count) |
+ : data_(i), destroy_count_(destroy_count) {} |
+ int data_; |
+ int* destroy_count_; |
+}; |
+ |
+TEST(ScopedPtrVectorTest, RemoveIf) { |
+ ScopedPtrVector<DataWithDestruction> v; |
+ int destroyed[6] = {0}; |
+ v.push_back(DataWithDestruction::Create(1, &destroyed[0])); |
+ v.push_back(DataWithDestruction::Create(2, &destroyed[1])); |
+ v.push_back(DataWithDestruction::Create(3, &destroyed[2])); |
+ v.push_back(DataWithDestruction::Create(3, &destroyed[3])); |
+ v.push_back(DataWithDestruction::Create(4, &destroyed[4])); |
+ v.push_back(DataWithDestruction::Create(5, &destroyed[5])); |
+ |
+ int expect_destroyed[6] = {0}; |
+ |
+ // Removing more than one thing that matches. |
+ auto is_three = [](DataWithDestruction* d) { return d->data() == 3; }; |
+ v.erase(v.remove_if(is_three), v.end()); |
+ EXPECT_EQ(4u, v.size()); |
+ expect_destroyed[2]++; |
+ expect_destroyed[3]++; |
+ for (size_t i = 0; i < arraysize(destroyed); ++i) |
+ EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; |
+ { |
+ int expect_data[4] = {1, 2, 4, 5}; |
+ for (size_t i = 0; i < arraysize(expect_data); ++i) |
+ EXPECT_EQ(expect_data[i], v[i]->data()) << i; |
+ } |
+ |
+ // Removing from the back of the vector. |
+ auto is_five = [](DataWithDestruction* d) { return d->data() == 5; }; |
+ v.erase(v.remove_if(is_five), v.end()); |
+ EXPECT_EQ(3u, v.size()); |
+ expect_destroyed[5]++; |
+ for (size_t i = 0; i < arraysize(destroyed); ++i) |
+ EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; |
+ { |
+ int expect_data[3] = {1, 2, 4}; |
+ for (size_t i = 0; i < arraysize(expect_data); ++i) |
+ EXPECT_EQ(expect_data[i], v[i]->data()) << i; |
+ } |
+ |
+ // Removing from the front of the vector. |
+ auto is_one = [](DataWithDestruction* d) { return d->data() == 1; }; |
+ v.erase(v.remove_if(is_one), v.end()); |
+ EXPECT_EQ(2u, v.size()); |
+ expect_destroyed[0]++; |
+ for (size_t i = 0; i < arraysize(destroyed); ++i) |
+ EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; |
+ { |
+ int expect_data[2] = {2, 4}; |
+ for (size_t i = 0; i < arraysize(expect_data); ++i) |
+ EXPECT_EQ(expect_data[i], v[i]->data()) << i; |
+ } |
+ |
+ // Removing things that aren't in the vector does nothing. |
+ v.erase(v.remove_if(is_one), v.end()); |
+ EXPECT_EQ(2u, v.size()); |
+ for (size_t i = 0; i < arraysize(destroyed); ++i) |
+ EXPECT_EQ(expect_destroyed[i], destroyed[i]) << i; |
+ { |
+ int expect_data[2] = {2, 4}; |
+ for (size_t i = 0; i < arraysize(expect_data); ++i) |
+ EXPECT_EQ(expect_data[i], v[i]->data()) << i; |
+ } |
+} |
+ |
+} // namespace |
+} // namespace cc |