1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
//===- llvm/unittest/DWARFLinkerParallel/StringPoolTest.cpp ---------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "llvm/DWARFLinkerParallel/StringPool.h"
#include "llvm/Support/Parallel.h"
#include "gtest/gtest.h"
#include <cstdlib>
using namespace llvm;
using namespace dwarflinker_parallel;
namespace {
TEST(StringPoolTest, TestStringPool) {
StringPool Strings;
// StringPool uses PerThreadBumpPtrAllocator which should be accessed from
// threads created by ThreadPoolExecutor. Use TaskGroup to run on
// ThreadPoolExecutor threads.
parallel::TaskGroup tg;
tg.spawn([&]() {
std::pair<StringEntry *, bool> Entry = Strings.insert("test");
EXPECT_TRUE(Entry.second);
EXPECT_TRUE(Entry.first->getKey() == "test");
EXPECT_TRUE(Entry.first->second == nullptr);
StringEntry *EntryPtr = Entry.first;
Entry = Strings.insert("test");
EXPECT_FALSE(Entry.second);
EXPECT_TRUE(Entry.first->getKey() == "test");
EXPECT_TRUE(Entry.first->second == nullptr);
EXPECT_TRUE(EntryPtr == Entry.first);
Entry = Strings.insert("test2");
EXPECT_TRUE(Entry.second);
EXPECT_TRUE(Entry.first->getKey() == "test2");
EXPECT_TRUE(Entry.first->second == nullptr);
EXPECT_TRUE(EntryPtr != Entry.first);
});
}
TEST(StringPoolTest, TestStringPoolParallel) {
StringPool Strings;
// Add data.
parallelFor(0, 1000, [&](size_t Idx) {
std::pair<StringEntry *, bool> Entry = Strings.insert(std::to_string(Idx));
EXPECT_TRUE(Entry.second);
EXPECT_TRUE(Entry.first->getKey() == std::to_string(Idx));
EXPECT_TRUE(Entry.first->second == nullptr);
});
// Check data.
parallelFor(0, 1000, [&](size_t Idx) {
std::pair<StringEntry *, bool> Entry = Strings.insert(std::to_string(Idx));
EXPECT_FALSE(Entry.second);
EXPECT_TRUE(Entry.first->getKey() == std::to_string(Idx));
EXPECT_TRUE(Entry.first->second == nullptr);
});
}
} // anonymous namespace
|
