blob: df283acd3a9e6a23cb0d46a8698c567d9e3572fc [file] [log] [blame]
/*
* Copyright (C) 2011 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "wtf/Vector.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "wtf/HashSet.h"
#include "wtf/OwnPtr.h"
#include "wtf/PassOwnPtr.h"
#include "wtf/text/WTFString.h"
namespace WTF {
namespace {
TEST(VectorTest, Basic)
{
Vector<int> intVector;
EXPECT_TRUE(intVector.isEmpty());
EXPECT_EQ(0ul, intVector.size());
EXPECT_EQ(0ul, intVector.capacity());
}
TEST(VectorTest, Reverse)
{
Vector<int> intVector;
intVector.append(10);
intVector.append(11);
intVector.append(12);
intVector.append(13);
intVector.reverse();
EXPECT_EQ(13, intVector[0]);
EXPECT_EQ(12, intVector[1]);
EXPECT_EQ(11, intVector[2]);
EXPECT_EQ(10, intVector[3]);
intVector.append(9);
intVector.reverse();
EXPECT_EQ(9, intVector[0]);
EXPECT_EQ(10, intVector[1]);
EXPECT_EQ(11, intVector[2]);
EXPECT_EQ(12, intVector[3]);
EXPECT_EQ(13, intVector[4]);
}
TEST(VectorTest, Remove)
{
Vector<int> intVector;
intVector.append(0);
intVector.append(1);
intVector.append(2);
intVector.append(3);
EXPECT_EQ(4u, intVector.size());
EXPECT_EQ(0, intVector[0]);
EXPECT_EQ(1, intVector[1]);
EXPECT_EQ(2, intVector[2]);
EXPECT_EQ(3, intVector[3]);
intVector.remove(2, 0);
EXPECT_EQ(4u, intVector.size());
EXPECT_EQ(2, intVector[2]);
intVector.remove(2, 1);
EXPECT_EQ(3u, intVector.size());
EXPECT_EQ(3, intVector[2]);
intVector.remove(0, 0);
EXPECT_EQ(3u, intVector.size());
EXPECT_EQ(0, intVector[0]);
intVector.remove(0);
EXPECT_EQ(2u, intVector.size());
EXPECT_EQ(1, intVector[0]);
}
TEST(VectorTest, Iterator)
{
Vector<int> intVector;
intVector.append(10);
intVector.append(11);
intVector.append(12);
intVector.append(13);
Vector<int>::iterator it = intVector.begin();
Vector<int>::iterator end = intVector.end();
EXPECT_TRUE(end != it);
EXPECT_EQ(10, *it);
++it;
EXPECT_EQ(11, *it);
++it;
EXPECT_EQ(12, *it);
++it;
EXPECT_EQ(13, *it);
++it;
EXPECT_TRUE(end == it);
}
TEST(VectorTest, ReverseIterator)
{
Vector<int> intVector;
intVector.append(10);
intVector.append(11);
intVector.append(12);
intVector.append(13);
Vector<int>::reverse_iterator it = intVector.rbegin();
Vector<int>::reverse_iterator end = intVector.rend();
EXPECT_TRUE(end != it);
EXPECT_EQ(13, *it);
++it;
EXPECT_EQ(12, *it);
++it;
EXPECT_EQ(11, *it);
++it;
EXPECT_EQ(10, *it);
++it;
EXPECT_TRUE(end == it);
}
class DestructCounter {
public:
explicit DestructCounter(int i, int* destructNumber)
: m_i(i)
, m_destructNumber(destructNumber)
{ }
~DestructCounter() { ++(*m_destructNumber); }
int get() const { return m_i; }
private:
int m_i;
int* m_destructNumber;
};
typedef WTF::Vector<OwnPtr<DestructCounter>> OwnPtrVector;
TEST(VectorTest, OwnPtr)
{
int destructNumber = 0;
OwnPtrVector vector;
vector.append(adoptPtr(new DestructCounter(0, &destructNumber)));
vector.append(adoptPtr(new DestructCounter(1, &destructNumber)));
EXPECT_EQ(2u, vector.size());
OwnPtr<DestructCounter>& counter0 = vector.first();
ASSERT_EQ(0, counter0->get());
int counter1 = vector.last()->get();
ASSERT_EQ(1, counter1);
ASSERT_EQ(0, destructNumber);
size_t index = 0;
for (OwnPtrVector::iterator iter = vector.begin(); iter != vector.end(); ++iter) {
OwnPtr<DestructCounter>* refCounter = iter;
EXPECT_EQ(index, static_cast<size_t>(refCounter->get()->get()));
EXPECT_EQ(index, static_cast<size_t>((*refCounter)->get()));
index++;
}
EXPECT_EQ(0, destructNumber);
for (index = 0; index < vector.size(); index++) {
OwnPtr<DestructCounter>& refCounter = vector[index];
EXPECT_EQ(index, static_cast<size_t>(refCounter->get()));
index++;
}
EXPECT_EQ(0, destructNumber);
EXPECT_EQ(0, vector[0]->get());
EXPECT_EQ(1, vector[1]->get());
vector.remove(0);
EXPECT_EQ(1, vector[0]->get());
EXPECT_EQ(1u, vector.size());
EXPECT_EQ(1, destructNumber);
OwnPtr<DestructCounter> ownCounter1 = vector[0].release();
vector.remove(0);
ASSERT_EQ(counter1, ownCounter1->get());
ASSERT_EQ(0u, vector.size());
ASSERT_EQ(1, destructNumber);
ownCounter1.clear();
EXPECT_EQ(2, destructNumber);
size_t count = 1025;
destructNumber = 0;
for (size_t i = 0; i < count; i++)
vector.prepend(adoptPtr(new DestructCounter(i, &destructNumber)));
// Vector relocation must not destruct OwnPtr element.
EXPECT_EQ(0, destructNumber);
EXPECT_EQ(count, vector.size());
OwnPtrVector copyVector;
vector.swap(copyVector);
EXPECT_EQ(0, destructNumber);
EXPECT_EQ(count, copyVector.size());
EXPECT_EQ(0u, vector.size());
copyVector.clear();
EXPECT_EQ(count, static_cast<size_t>(destructNumber));
}
// WrappedInt class will fail if it was memmoved or memcpyed.
static HashSet<void*> constructedWrappedInts;
class WrappedInt {
public:
WrappedInt(int i = 0)
: m_originalThisPtr(this)
, m_i(i)
{
constructedWrappedInts.add(this);
}
WrappedInt(const WrappedInt& other)
: m_originalThisPtr(this)
, m_i(other.m_i)
{
constructedWrappedInts.add(this);
}
WrappedInt& operator=(const WrappedInt& other)
{
m_i = other.m_i;
return *this;
}
~WrappedInt()
{
EXPECT_EQ(m_originalThisPtr, this);
EXPECT_TRUE(constructedWrappedInts.contains(this));
constructedWrappedInts.remove(this);
}
int get() const { return m_i; }
private:
void* m_originalThisPtr;
int m_i;
};
TEST(VectorTest, SwapWithInlineCapacity)
{
const size_t inlineCapacity = 2;
Vector<WrappedInt, inlineCapacity> vectorA;
vectorA.append(WrappedInt(1));
Vector<WrappedInt, inlineCapacity> vectorB;
vectorB.append(WrappedInt(2));
EXPECT_EQ(vectorA.size(), vectorB.size());
vectorA.swap(vectorB);
EXPECT_EQ(1u, vectorA.size());
EXPECT_EQ(2, vectorA.at(0).get());
EXPECT_EQ(1u, vectorB.size());
EXPECT_EQ(1, vectorB.at(0).get());
vectorA.append(WrappedInt(3));
EXPECT_GT(vectorA.size(), vectorB.size());
vectorA.swap(vectorB);
EXPECT_EQ(1u, vectorA.size());
EXPECT_EQ(1, vectorA.at(0).get());
EXPECT_EQ(2u, vectorB.size());
EXPECT_EQ(2, vectorB.at(0).get());
EXPECT_EQ(3, vectorB.at(1).get());
EXPECT_LT(vectorA.size(), vectorB.size());
vectorA.swap(vectorB);
EXPECT_EQ(2u, vectorA.size());
EXPECT_EQ(2, vectorA.at(0).get());
EXPECT_EQ(3, vectorA.at(1).get());
EXPECT_EQ(1u, vectorB.size());
EXPECT_EQ(1, vectorB.at(0).get());
vectorA.append(WrappedInt(4));
EXPECT_GT(vectorA.size(), inlineCapacity);
vectorA.swap(vectorB);
EXPECT_EQ(1u, vectorA.size());
EXPECT_EQ(1, vectorA.at(0).get());
EXPECT_EQ(3u, vectorB.size());
EXPECT_EQ(2, vectorB.at(0).get());
EXPECT_EQ(3, vectorB.at(1).get());
EXPECT_EQ(4, vectorB.at(2).get());
vectorB.swap(vectorA);
}
#if defined(ANNOTATE_CONTIGUOUS_CONTAINER)
TEST(VectorTest, ContainerAnnotations)
{
Vector<int> vectorA;
vectorA.append(10);
vectorA.reserveCapacity(32);
volatile int* intPointerA = vectorA.data();
EXPECT_DEATH(intPointerA[1] = 11, "container-overflow");
vectorA.append(11);
intPointerA[1] = 11;
EXPECT_DEATH(intPointerA[2] = 12, "container-overflow");
EXPECT_DEATH((void)intPointerA[2], "container-overflow");
vectorA.shrinkToFit();
vectorA.reserveCapacity(16);
intPointerA = vectorA.data();
EXPECT_DEATH((void)intPointerA[2], "container-overflow");
Vector<int> vectorB(vectorA);
vectorB.reserveCapacity(16);
volatile int* intPointerB = vectorB.data();
EXPECT_DEATH((void)intPointerB[2], "container-overflow");
Vector<int> vectorC((Vector<int>(vectorA)));
volatile int* intPointerC = vectorC.data();
EXPECT_DEATH((void)intPointerC[2], "container-overflow");
vectorC.append(13);
vectorC.swap(vectorB);
volatile int* intPointerB2 = vectorB.data();
volatile int* intPointerC2 = vectorC.data();
intPointerB2[2] = 13;
EXPECT_DEATH((void)intPointerB2[3], "container-overflow");
EXPECT_DEATH((void)intPointerC2[2], "container-overflow");
vectorB = vectorC;
volatile int* intPointerB3 = vectorB.data();
EXPECT_DEATH((void)intPointerB3[2], "container-overflow");
}
#endif // defined(ANNOTATE_CONTIGUOUS_CONTAINER)
class Comparable {
};
bool operator==(const Comparable& a, const Comparable& b) { return true; }
template<typename T> void compare()
{
EXPECT_TRUE(Vector<T>() == Vector<T>());
EXPECT_FALSE(Vector<T>(1) == Vector<T>(0));
EXPECT_FALSE(Vector<T>() == Vector<T>(1));
EXPECT_TRUE(Vector<T>(1) == Vector<T>(1));
Vector<T, 1> vectorWithInlineCapacity;
EXPECT_TRUE(vectorWithInlineCapacity == Vector<T>());
EXPECT_FALSE(vectorWithInlineCapacity == Vector<T>(1));
}
TEST(VectorTest, Compare)
{
compare<int>();
compare<Comparable>();
compare<WTF::String>();
}
} // anonymous namespace
} // namespace WTF