1 //===- StringTableBuilderTest.cpp -----------------------------------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "llvm/DebugInfo/PDB/Native/PDBStringTable.h"
11 #include "llvm/DebugInfo/PDB/Native/PDBStringTableBuilder.h"
12 #include "llvm/Support/BinaryByteStream.h"
13 #include "llvm/Support/BinaryStreamReader.h"
14 #include "llvm/Support/BinaryStreamWriter.h"
15 #include "llvm/Testing/Support/Error.h"
16
17 #include "gtest/gtest.h"
18
19 using namespace llvm;
20 using namespace llvm::pdb;
21 using namespace llvm::support;
22
23 namespace {
24 class StringTableBuilderTest : public ::testing::Test {};
25 }
26
TEST_F(StringTableBuilderTest,Simple)27 TEST_F(StringTableBuilderTest, Simple) {
28 // Create /names table contents.
29 PDBStringTableBuilder Builder;
30
31 // This test case is carefully constructed to ensure that at least one
32 // string gets bucketed into slot 0, *and* to ensure that at least one
33 // has a hash collision at the end of the bucket list so it has to
34 // wrap around.
35 uint32_t FooID = Builder.insert("foo");
36 uint32_t BarID = Builder.insert("bar");
37 uint32_t BazID = Builder.insert("baz");
38 uint32_t BuzzID = Builder.insert("buzz");
39 uint32_t BazzID = Builder.insert("bazz");
40 uint32_t BarrID = Builder.insert("barr");
41
42 // Re-inserting the same item should return the same id.
43 EXPECT_EQ(FooID, Builder.insert("foo"));
44 EXPECT_EQ(BarID, Builder.insert("bar"));
45 EXPECT_EQ(BazID, Builder.insert("baz"));
46 EXPECT_EQ(BuzzID, Builder.insert("buzz"));
47 EXPECT_EQ(BazzID, Builder.insert("bazz"));
48 EXPECT_EQ(BarrID, Builder.insert("barr"));
49
50 // Each ID should be distinct.
51 std::set<uint32_t> Distinct{FooID, BarID, BazID, BuzzID, BazzID, BarrID};
52 EXPECT_EQ(6U, Distinct.size());
53
54 std::vector<uint8_t> Buffer(Builder.calculateSerializedSize());
55 MutableBinaryByteStream OutStream(Buffer, little);
56 BinaryStreamWriter Writer(OutStream);
57 EXPECT_THAT_ERROR(Builder.commit(Writer), Succeeded());
58
59 // Reads the contents back.
60 BinaryByteStream InStream(Buffer, little);
61 BinaryStreamReader Reader(InStream);
62 PDBStringTable Table;
63 EXPECT_THAT_ERROR(Table.reload(Reader), Succeeded());
64
65 EXPECT_EQ(6U, Table.getNameCount());
66 EXPECT_EQ(1U, Table.getHashVersion());
67
68 EXPECT_THAT_EXPECTED(Table.getStringForID(FooID), HasValue("foo"));
69 EXPECT_THAT_EXPECTED(Table.getStringForID(BarID), HasValue("bar"));
70 EXPECT_THAT_EXPECTED(Table.getStringForID(BazID), HasValue("baz"));
71 EXPECT_THAT_EXPECTED(Table.getStringForID(BuzzID), HasValue("buzz"));
72 EXPECT_THAT_EXPECTED(Table.getStringForID(BazzID), HasValue("bazz"));
73 EXPECT_THAT_EXPECTED(Table.getStringForID(BarrID), HasValue("barr"));
74
75 EXPECT_THAT_EXPECTED(Table.getIDForString("foo"), HasValue(FooID));
76 EXPECT_THAT_EXPECTED(Table.getIDForString("bar"), HasValue(BarID));
77 EXPECT_THAT_EXPECTED(Table.getIDForString("baz"), HasValue(BazID));
78 EXPECT_THAT_EXPECTED(Table.getIDForString("buzz"), HasValue(BuzzID));
79 EXPECT_THAT_EXPECTED(Table.getIDForString("bazz"), HasValue(BazzID));
80 EXPECT_THAT_EXPECTED(Table.getIDForString("barr"), HasValue(BarrID));
81 }
82