1 /*
2 * Copyright (C) 2012 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #define LOG_TAG "Vector_test"
18
19 #define __STDC_LIMIT_MACROS
20 #include <stdint.h>
21 #include <unistd.h>
22
23 #include <android/log.h>
24 #include <gtest/gtest.h>
25 #include <utils/Vector.h>
26
27 namespace android {
28
29 class VectorTest : public testing::Test {
30 protected:
SetUp()31 virtual void SetUp() {
32 }
33
TearDown()34 virtual void TearDown() {
35 }
36
37 public:
38 };
39
40
TEST_F(VectorTest,CopyOnWrite_CopyAndAddElements)41 TEST_F(VectorTest, CopyOnWrite_CopyAndAddElements) {
42
43 Vector<int> vector;
44 Vector<int> other;
45 vector.setCapacity(8);
46
47 vector.add(1);
48 vector.add(2);
49 vector.add(3);
50
51 EXPECT_EQ(3U, vector.size());
52
53 // copy the vector
54 other = vector;
55
56 EXPECT_EQ(3U, other.size());
57
58 // add an element to the first vector
59 vector.add(4);
60
61 // make sure the sizes are correct
62 EXPECT_EQ(4U, vector.size());
63 EXPECT_EQ(3U, other.size());
64
65 // add an element to the copy
66 other.add(5);
67
68 // make sure the sizes are correct
69 EXPECT_EQ(4U, vector.size());
70 EXPECT_EQ(4U, other.size());
71
72 // make sure the content of both vectors are correct
73 EXPECT_EQ(vector[3], 4);
74 EXPECT_EQ(other[3], 5);
75 }
76
TEST_F(VectorTest,SetCapacity_Overflow)77 TEST_F(VectorTest, SetCapacity_Overflow) {
78 Vector<int> vector;
79 EXPECT_DEATH(vector.setCapacity(SIZE_MAX / sizeof(int) + 1), "Assertion failed");
80 }
81
TEST_F(VectorTest,SetCapacity_ShrinkBelowSize)82 TEST_F(VectorTest, SetCapacity_ShrinkBelowSize) {
83 Vector<int> vector;
84 vector.add(1);
85 vector.add(2);
86 vector.add(3);
87 vector.add(4);
88
89 vector.setCapacity(8);
90 ASSERT_EQ(8U, vector.capacity());
91 vector.setCapacity(2);
92 ASSERT_EQ(8U, vector.capacity());
93 }
94
TEST_F(VectorTest,_grow_OverflowSize)95 TEST_F(VectorTest, _grow_OverflowSize) {
96 Vector<int> vector;
97 vector.add(1);
98
99 // Checks that the size calculation (not the capacity calculation) doesn't
100 // overflow : the size here will be (1 + SIZE_MAX).
101 EXPECT_DEATH(vector.insertArrayAt(nullptr, 0, SIZE_MAX), "new_size overflow");
102 }
103
TEST_F(VectorTest,_grow_OverflowCapacityDoubling)104 TEST_F(VectorTest, _grow_OverflowCapacityDoubling) {
105 Vector<int> vector;
106
107 // This should fail because the calculated capacity will overflow even though
108 // the size of the vector doesn't.
109 EXPECT_DEATH(vector.insertArrayAt(nullptr, 0, (SIZE_MAX - 1)), "new_capacity overflow");
110 }
111
TEST_F(VectorTest,_grow_OverflowBufferAlloc)112 TEST_F(VectorTest, _grow_OverflowBufferAlloc) {
113 Vector<int> vector;
114 // This should fail because the capacity * sizeof(int) overflows, even
115 // though the capacity itself doesn't.
116 EXPECT_DEATH(vector.insertArrayAt(nullptr, 0, (SIZE_MAX / 2)), "new_alloc_size overflow");
117 }
118
TEST_F(VectorTest,editArray_Shared)119 TEST_F(VectorTest, editArray_Shared) {
120 Vector<int> vector1;
121 vector1.add(1);
122 vector1.add(2);
123 vector1.add(3);
124 vector1.add(4);
125
126 Vector<int> vector2 = vector1;
127 ASSERT_EQ(vector1.array(), vector2.array());
128 // We must make a copy here, since we're not the exclusive owners
129 // of this array.
130 ASSERT_NE(vector1.editArray(), vector2.editArray());
131
132 // Vector doesn't implement operator ==.
133 ASSERT_EQ(vector1.size(), vector2.size());
134 for (size_t i = 0; i < vector1.size(); ++i) {
135 EXPECT_EQ(vector1[i], vector2[i]);
136 }
137 }
138
TEST_F(VectorTest,removeItemsAt_overflow)139 TEST_F(VectorTest, removeItemsAt_overflow) {
140 android::Vector<int> v;
141 for (int i = 0; i < 666; i++) v.add(i);
142
143 ASSERT_DEATH(v.removeItemsAt(SIZE_MAX, 666), "overflow");
144 ASSERT_DEATH(v.removeItemsAt(666, SIZE_MAX), "overflow");
145 ASSERT_DEATH(v.removeItemsAt(SIZE_MAX, SIZE_MAX), "overflow");
146 }
147
148 } // namespace android
149