2009-01-08 07:35:39 +00:00
|
|
|
//===- llvm/unittest/ADT/StringMapMap.cpp - StringMap unit tests ----------===//
|
2009-01-08 04:48:20 +00:00
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "llvm/ADT/StringMap.h"
|
2010-11-29 18:16:10 +00:00
|
|
|
#include "llvm/Support/DataTypes.h"
|
2014-06-23 18:28:53 +00:00
|
|
|
#include <tuple>
|
2009-01-08 04:48:20 +00:00
|
|
|
using namespace llvm;
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
|
|
|
|
// Test fixture
|
|
|
|
class StringMapTest : public testing::Test {
|
|
|
|
protected:
|
|
|
|
StringMap<uint32_t> testMap;
|
|
|
|
|
|
|
|
static const char testKey[];
|
|
|
|
static const uint32_t testValue;
|
|
|
|
static const char* testKeyFirst;
|
2009-07-23 18:17:34 +00:00
|
|
|
static size_t testKeyLength;
|
2009-01-08 04:48:20 +00:00
|
|
|
static const std::string testKeyStr;
|
|
|
|
|
|
|
|
void assertEmptyMap() {
|
|
|
|
// Size tests
|
|
|
|
EXPECT_EQ(0u, testMap.size());
|
|
|
|
EXPECT_TRUE(testMap.empty());
|
|
|
|
|
|
|
|
// Iterator tests
|
|
|
|
EXPECT_TRUE(testMap.begin() == testMap.end());
|
|
|
|
|
|
|
|
// Lookup tests
|
|
|
|
EXPECT_EQ(0u, testMap.count(testKey));
|
2009-07-23 18:17:34 +00:00
|
|
|
EXPECT_EQ(0u, testMap.count(StringRef(testKeyFirst, testKeyLength)));
|
2009-01-08 04:48:20 +00:00
|
|
|
EXPECT_EQ(0u, testMap.count(testKeyStr));
|
|
|
|
EXPECT_TRUE(testMap.find(testKey) == testMap.end());
|
2009-07-23 18:17:34 +00:00
|
|
|
EXPECT_TRUE(testMap.find(StringRef(testKeyFirst, testKeyLength)) ==
|
|
|
|
testMap.end());
|
2009-01-08 04:48:20 +00:00
|
|
|
EXPECT_TRUE(testMap.find(testKeyStr) == testMap.end());
|
|
|
|
}
|
|
|
|
|
|
|
|
void assertSingleItemMap() {
|
|
|
|
// Size tests
|
|
|
|
EXPECT_EQ(1u, testMap.size());
|
|
|
|
EXPECT_FALSE(testMap.begin() == testMap.end());
|
|
|
|
EXPECT_FALSE(testMap.empty());
|
|
|
|
|
|
|
|
// Iterator tests
|
|
|
|
StringMap<uint32_t>::iterator it = testMap.begin();
|
2011-07-14 18:31:43 +00:00
|
|
|
EXPECT_STREQ(testKey, it->first().data());
|
2009-01-08 04:48:20 +00:00
|
|
|
EXPECT_EQ(testValue, it->second);
|
|
|
|
++it;
|
|
|
|
EXPECT_TRUE(it == testMap.end());
|
|
|
|
|
|
|
|
// Lookup tests
|
|
|
|
EXPECT_EQ(1u, testMap.count(testKey));
|
2009-07-23 18:17:34 +00:00
|
|
|
EXPECT_EQ(1u, testMap.count(StringRef(testKeyFirst, testKeyLength)));
|
2009-01-08 04:48:20 +00:00
|
|
|
EXPECT_EQ(1u, testMap.count(testKeyStr));
|
|
|
|
EXPECT_TRUE(testMap.find(testKey) == testMap.begin());
|
2009-07-23 18:17:34 +00:00
|
|
|
EXPECT_TRUE(testMap.find(StringRef(testKeyFirst, testKeyLength)) ==
|
|
|
|
testMap.begin());
|
2009-01-08 04:48:20 +00:00
|
|
|
EXPECT_TRUE(testMap.find(testKeyStr) == testMap.begin());
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
const char StringMapTest::testKey[] = "key";
|
|
|
|
const uint32_t StringMapTest::testValue = 1u;
|
|
|
|
const char* StringMapTest::testKeyFirst = testKey;
|
2009-07-23 18:17:34 +00:00
|
|
|
size_t StringMapTest::testKeyLength = sizeof(testKey) - 1;
|
2009-01-08 04:48:20 +00:00
|
|
|
const std::string StringMapTest::testKeyStr(testKey);
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// Empty map tests.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, EmptyMapTest) {
|
|
|
|
assertEmptyMap();
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// Constant map tests.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, ConstEmptyMapTest) {
|
|
|
|
const StringMap<uint32_t>& constTestMap = testMap;
|
|
|
|
|
|
|
|
// Size tests
|
|
|
|
EXPECT_EQ(0u, constTestMap.size());
|
|
|
|
EXPECT_TRUE(constTestMap.empty());
|
|
|
|
|
|
|
|
// Iterator tests
|
|
|
|
EXPECT_TRUE(constTestMap.begin() == constTestMap.end());
|
|
|
|
|
|
|
|
// Lookup tests
|
|
|
|
EXPECT_EQ(0u, constTestMap.count(testKey));
|
2009-07-23 18:17:34 +00:00
|
|
|
EXPECT_EQ(0u, constTestMap.count(StringRef(testKeyFirst, testKeyLength)));
|
2009-01-08 04:48:20 +00:00
|
|
|
EXPECT_EQ(0u, constTestMap.count(testKeyStr));
|
|
|
|
EXPECT_TRUE(constTestMap.find(testKey) == constTestMap.end());
|
2009-07-23 18:17:34 +00:00
|
|
|
EXPECT_TRUE(constTestMap.find(StringRef(testKeyFirst, testKeyLength)) ==
|
2009-01-08 09:31:36 +00:00
|
|
|
constTestMap.end());
|
2009-01-08 04:48:20 +00:00
|
|
|
EXPECT_TRUE(constTestMap.find(testKeyStr) == constTestMap.end());
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// A map with a single entry.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, SingleEntryMapTest) {
|
|
|
|
testMap[testKey] = testValue;
|
|
|
|
assertSingleItemMap();
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// Test clear() method.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, ClearTest) {
|
|
|
|
testMap[testKey] = testValue;
|
|
|
|
testMap.clear();
|
|
|
|
assertEmptyMap();
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// Test erase(iterator) method.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, EraseIteratorTest) {
|
|
|
|
testMap[testKey] = testValue;
|
|
|
|
testMap.erase(testMap.begin());
|
|
|
|
assertEmptyMap();
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// Test erase(value) method.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, EraseValueTest) {
|
|
|
|
testMap[testKey] = testValue;
|
|
|
|
testMap.erase(testKey);
|
|
|
|
assertEmptyMap();
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// Test inserting two values and erasing one.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, InsertAndEraseTest) {
|
|
|
|
testMap[testKey] = testValue;
|
|
|
|
testMap["otherKey"] = 2;
|
|
|
|
testMap.erase("otherKey");
|
|
|
|
assertSingleItemMap();
|
|
|
|
}
|
|
|
|
|
2012-06-19 17:40:35 +00:00
|
|
|
TEST_F(StringMapTest, SmallFullMapTest) {
|
|
|
|
// StringMap has a tricky corner case when the map is small (<8 buckets) and
|
|
|
|
// it fills up through a balanced pattern of inserts and erases. This can
|
|
|
|
// lead to inf-loops in some cases (PR13148) so we test it explicitly here.
|
|
|
|
llvm::StringMap<int> Map(2);
|
|
|
|
|
|
|
|
Map["eins"] = 1;
|
|
|
|
Map["zwei"] = 2;
|
|
|
|
Map["drei"] = 3;
|
|
|
|
Map.erase("drei");
|
|
|
|
Map.erase("eins");
|
|
|
|
Map["veir"] = 4;
|
|
|
|
Map["funf"] = 5;
|
|
|
|
|
|
|
|
EXPECT_EQ(3u, Map.size());
|
|
|
|
EXPECT_EQ(0, Map.lookup("eins"));
|
|
|
|
EXPECT_EQ(2, Map.lookup("zwei"));
|
|
|
|
EXPECT_EQ(0, Map.lookup("drei"));
|
|
|
|
EXPECT_EQ(4, Map.lookup("veir"));
|
|
|
|
EXPECT_EQ(5, Map.lookup("funf"));
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// A more complex iteration test.
|
2009-01-08 04:48:20 +00:00
|
|
|
TEST_F(StringMapTest, IterationTest) {
|
|
|
|
bool visited[100];
|
|
|
|
|
|
|
|
// Insert 100 numbers into the map
|
|
|
|
for (int i = 0; i < 100; ++i) {
|
|
|
|
std::stringstream ss;
|
|
|
|
ss << "key_" << i;
|
|
|
|
testMap[ss.str()] = i;
|
|
|
|
visited[i] = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Iterate over all numbers and mark each one found.
|
|
|
|
for (StringMap<uint32_t>::iterator it = testMap.begin();
|
|
|
|
it != testMap.end(); ++it) {
|
|
|
|
std::stringstream ss;
|
|
|
|
ss << "key_" << it->second;
|
2011-07-14 18:31:43 +00:00
|
|
|
ASSERT_STREQ(ss.str().c_str(), it->first().data());
|
2009-01-08 04:48:20 +00:00
|
|
|
visited[it->second] = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure every number was visited.
|
|
|
|
for (int i = 0; i < 100; ++i) {
|
|
|
|
ASSERT_TRUE(visited[i]) << "Entry #" << i << " was never visited";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
// Test StringMapEntry::Create() method.
|
|
|
|
TEST_F(StringMapTest, StringMapEntryTest) {
|
|
|
|
StringMap<uint32_t>::value_type* entry =
|
|
|
|
StringMap<uint32_t>::value_type::Create(
|
2014-06-11 05:35:56 +00:00
|
|
|
StringRef(testKeyFirst, testKeyLength), 1u);
|
2011-07-14 18:31:43 +00:00
|
|
|
EXPECT_STREQ(testKey, entry->first().data());
|
2009-01-08 09:31:36 +00:00
|
|
|
EXPECT_EQ(1u, entry->second);
|
2010-02-11 07:16:13 +00:00
|
|
|
free(entry);
|
2009-01-08 09:31:36 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Test insert() method.
|
|
|
|
TEST_F(StringMapTest, InsertTest) {
|
|
|
|
SCOPED_TRACE("InsertTest");
|
|
|
|
testMap.insert(
|
|
|
|
StringMap<uint32_t>::value_type::Create(
|
2014-06-11 05:35:56 +00:00
|
|
|
StringRef(testKeyFirst, testKeyLength),
|
2009-07-23 18:17:34 +00:00
|
|
|
testMap.getAllocator(), 1u));
|
2009-01-08 09:31:36 +00:00
|
|
|
assertSingleItemMap();
|
2009-01-08 04:48:20 +00:00
|
|
|
}
|
2009-01-08 09:31:36 +00:00
|
|
|
|
2014-06-23 18:28:53 +00:00
|
|
|
// Test insert(pair<K, V>) method
|
|
|
|
TEST_F(StringMapTest, InsertPairTest) {
|
|
|
|
bool Inserted;
|
|
|
|
StringMap<uint32_t>::iterator NewIt;
|
|
|
|
std::tie(NewIt, Inserted) =
|
|
|
|
testMap.insert(std::make_pair(testKeyFirst, testValue));
|
|
|
|
EXPECT_EQ(1u, testMap.size());
|
|
|
|
EXPECT_EQ(testValue, testMap[testKeyFirst]);
|
|
|
|
EXPECT_EQ(testKeyFirst, NewIt->first());
|
|
|
|
EXPECT_EQ(testValue, NewIt->second);
|
|
|
|
EXPECT_TRUE(Inserted);
|
|
|
|
|
|
|
|
StringMap<uint32_t>::iterator ExistingIt;
|
|
|
|
std::tie(ExistingIt, Inserted) =
|
|
|
|
testMap.insert(std::make_pair(testKeyFirst, testValue + 1));
|
|
|
|
EXPECT_EQ(1u, testMap.size());
|
|
|
|
EXPECT_EQ(testValue, testMap[testKeyFirst]);
|
|
|
|
EXPECT_FALSE(Inserted);
|
|
|
|
EXPECT_EQ(NewIt, ExistingIt);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test insert(pair<K, V>) method when rehashing occurs
|
|
|
|
TEST_F(StringMapTest, InsertRehashingPairTest) {
|
|
|
|
// Check that the correct iterator is returned when the inserted element is
|
|
|
|
// moved to a different bucket during internal rehashing. This depends on
|
|
|
|
// the particular key, and the implementation of StringMap and HashString.
|
|
|
|
// Changes to those might result in this test not actually checking that.
|
|
|
|
StringMap<uint32_t> t(1);
|
|
|
|
EXPECT_EQ(1u, t.getNumBuckets());
|
|
|
|
|
|
|
|
StringMap<uint32_t>::iterator It =
|
|
|
|
t.insert(std::make_pair("abcdef", 42)).first;
|
|
|
|
EXPECT_EQ(2u, t.getNumBuckets());
|
|
|
|
EXPECT_EQ("abcdef", It->first());
|
|
|
|
EXPECT_EQ(42u, It->second);
|
|
|
|
}
|
|
|
|
|
2014-01-02 23:57:28 +00:00
|
|
|
// Create a non-default constructable value
|
2014-01-03 00:00:41 +00:00
|
|
|
struct StringMapTestStruct {
|
|
|
|
StringMapTestStruct(int i) : i(i) {}
|
|
|
|
StringMapTestStruct() LLVM_DELETED_FUNCTION;
|
|
|
|
int i;
|
|
|
|
};
|
2014-01-02 23:57:28 +00:00
|
|
|
|
2014-01-03 00:00:41 +00:00
|
|
|
TEST_F(StringMapTest, NonDefaultConstructable) {
|
2014-01-02 23:57:28 +00:00
|
|
|
StringMap<StringMapTestStruct> t;
|
2014-11-19 05:49:42 +00:00
|
|
|
t.insert(std::make_pair("Test", StringMapTestStruct(123)));
|
2014-01-02 23:57:28 +00:00
|
|
|
StringMap<StringMapTestStruct>::iterator iter = t.find("Test");
|
|
|
|
ASSERT_NE(iter, t.end());
|
|
|
|
ASSERT_EQ(iter->second.i, 123);
|
|
|
|
}
|
|
|
|
|
2014-11-14 00:41:46 +00:00
|
|
|
struct Immovable {
|
|
|
|
Immovable() {}
|
|
|
|
Immovable(Immovable&&) LLVM_DELETED_FUNCTION; // will disable the other special members
|
|
|
|
};
|
|
|
|
|
2014-05-08 21:52:23 +00:00
|
|
|
struct MoveOnly {
|
|
|
|
int i;
|
|
|
|
MoveOnly(int i) : i(i) {}
|
2014-11-14 00:41:46 +00:00
|
|
|
MoveOnly(const Immovable&) : i(0) {}
|
2014-05-09 02:26:36 +00:00
|
|
|
MoveOnly(MoveOnly &&RHS) : i(RHS.i) {}
|
|
|
|
MoveOnly &operator=(MoveOnly &&RHS) {
|
|
|
|
i = RHS.i;
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
2014-06-23 18:28:53 +00:00
|
|
|
MoveOnly(const MoveOnly &) LLVM_DELETED_FUNCTION;
|
|
|
|
MoveOnly &operator=(const MoveOnly &) LLVM_DELETED_FUNCTION;
|
2014-05-08 21:52:23 +00:00
|
|
|
};
|
|
|
|
|
2014-11-14 00:41:46 +00:00
|
|
|
TEST_F(StringMapTest, MoveOnly) {
|
2014-05-08 21:52:23 +00:00
|
|
|
StringMap<MoveOnly> t;
|
2014-11-19 05:49:42 +00:00
|
|
|
t.insert(std::make_pair("Test", MoveOnly(42)));
|
2014-05-08 21:52:23 +00:00
|
|
|
StringRef Key = "Test";
|
2014-06-11 05:35:56 +00:00
|
|
|
StringMapEntry<MoveOnly>::Create(Key, MoveOnly(42))
|
2014-05-08 21:53:33 +00:00
|
|
|
->Destroy();
|
2014-05-08 21:52:23 +00:00
|
|
|
}
|
|
|
|
|
2014-11-14 00:41:46 +00:00
|
|
|
TEST_F(StringMapTest, CtorArg) {
|
|
|
|
StringRef Key = "Test";
|
|
|
|
StringMapEntry<MoveOnly>::Create(Key, Immovable())
|
|
|
|
->Destroy();
|
|
|
|
}
|
|
|
|
|
2014-05-08 21:52:29 +00:00
|
|
|
TEST_F(StringMapTest, MoveConstruct) {
|
|
|
|
StringMap<int> A;
|
2014-11-19 05:49:42 +00:00
|
|
|
A["x"] = 42;
|
2014-05-08 21:52:29 +00:00
|
|
|
StringMap<int> B = std::move(A);
|
|
|
|
ASSERT_EQ(A.size(), 0u);
|
|
|
|
ASSERT_EQ(B.size(), 1u);
|
|
|
|
ASSERT_EQ(B["x"], 42);
|
|
|
|
ASSERT_EQ(B.count("y"), 0u);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_F(StringMapTest, MoveAssignment) {
|
|
|
|
StringMap<int> A;
|
|
|
|
A["x"] = 42;
|
|
|
|
StringMap<int> B;
|
|
|
|
B["y"] = 117;
|
|
|
|
A = std::move(B);
|
|
|
|
ASSERT_EQ(A.size(), 1u);
|
|
|
|
ASSERT_EQ(B.size(), 0u);
|
|
|
|
ASSERT_EQ(A["y"], 117);
|
|
|
|
ASSERT_EQ(B.count("x"), 0u);
|
|
|
|
}
|
|
|
|
|
|
|
|
struct Countable {
|
|
|
|
int &InstanceCount;
|
|
|
|
int Number;
|
2014-05-08 21:53:33 +00:00
|
|
|
Countable(int Number, int &InstanceCount)
|
|
|
|
: InstanceCount(InstanceCount), Number(Number) {
|
2014-05-08 21:52:29 +00:00
|
|
|
++InstanceCount;
|
|
|
|
}
|
|
|
|
Countable(Countable &&C) : InstanceCount(C.InstanceCount), Number(C.Number) {
|
|
|
|
++InstanceCount;
|
|
|
|
C.Number = -1;
|
|
|
|
}
|
2014-05-08 21:53:33 +00:00
|
|
|
Countable(const Countable &C)
|
|
|
|
: InstanceCount(C.InstanceCount), Number(C.Number) {
|
2014-05-08 21:52:29 +00:00
|
|
|
++InstanceCount;
|
|
|
|
}
|
|
|
|
Countable &operator=(Countable C) {
|
|
|
|
Number = C.Number;
|
|
|
|
return *this;
|
|
|
|
}
|
2014-05-08 21:53:33 +00:00
|
|
|
~Countable() { --InstanceCount; }
|
2014-05-08 21:52:29 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
TEST_F(StringMapTest, MoveDtor) {
|
|
|
|
int InstanceCount = 0;
|
|
|
|
StringMap<Countable> A;
|
2014-11-19 05:49:42 +00:00
|
|
|
A.insert(std::make_pair("x", Countable(42, InstanceCount)));
|
2014-05-08 21:52:29 +00:00
|
|
|
ASSERT_EQ(InstanceCount, 1);
|
|
|
|
auto I = A.find("x");
|
|
|
|
ASSERT_NE(I, A.end());
|
|
|
|
ASSERT_EQ(I->second.Number, 42);
|
|
|
|
|
|
|
|
StringMap<Countable> B;
|
|
|
|
B = std::move(A);
|
|
|
|
ASSERT_EQ(InstanceCount, 1);
|
|
|
|
ASSERT_TRUE(A.empty());
|
|
|
|
I = B.find("x");
|
|
|
|
ASSERT_NE(I, B.end());
|
|
|
|
ASSERT_EQ(I->second.Number, 42);
|
|
|
|
|
|
|
|
B = StringMap<Countable>();
|
|
|
|
ASSERT_EQ(InstanceCount, 0);
|
|
|
|
ASSERT_TRUE(B.empty());
|
|
|
|
}
|
|
|
|
|
2009-01-08 09:31:36 +00:00
|
|
|
} // end anonymous namespace
|