aboutsummaryrefslogtreecommitdiff
path: root/llvm/unittests/DebugInfo/LogicalView/StringPoolTest.cpp
blob: 27ff449c98e6413a81c9a0a605e342a6f3e8e0b0 (plain)
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
//===- llvm/unittest/DebugInfo/LogicalView/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/DebugInfo/LogicalView/Core/LVStringPool.h"
#include <vector>

#include "gtest/gtest.h"

using namespace llvm;
using namespace llvm::logicalview;

namespace {

TEST(StringPoolTest, AddStrings) {
  LVStringPool PoolInstance;
  EXPECT_EQ(0u, PoolInstance.getSize());

  // Get indexes for the initial strings.
  EXPECT_EQ(1u, PoolInstance.getIndex("one"));
  EXPECT_EQ(2u, PoolInstance.getIndex("two"));
  EXPECT_EQ(3u, PoolInstance.getIndex("three"));
  EXPECT_EQ(4u, PoolInstance.getIndex("four"));
  EXPECT_EQ(5u, PoolInstance.getIndex("five"));
  EXPECT_EQ(5u, PoolInstance.getSize());

  // Verify the string returned by the given index.
  EXPECT_EQ("one", PoolInstance.getString(1));
  EXPECT_EQ("two", PoolInstance.getString(2));
  EXPECT_EQ("three", PoolInstance.getString(3));
  EXPECT_EQ("four", PoolInstance.getString(4));
  EXPECT_EQ("five", PoolInstance.getString(5));
  EXPECT_EQ(5u, PoolInstance.getSize());

  // Get indexes for the same initial strings.
  EXPECT_EQ(5u, PoolInstance.getIndex("five"));
  EXPECT_EQ(4u, PoolInstance.getIndex("four"));
  EXPECT_EQ(3u, PoolInstance.getIndex("three"));
  EXPECT_EQ(2u, PoolInstance.getIndex("two"));
  EXPECT_EQ(1u, PoolInstance.getIndex("one"));
  EXPECT_EQ(5u, PoolInstance.getSize());

  // Empty string gets the index zero.
  EXPECT_EQ(0u, PoolInstance.getIndex(""));
  EXPECT_EQ(5u, PoolInstance.getSize());

  // Empty string for invalid index.
  EXPECT_EQ("", PoolInstance.getString(620));

  // Lookup for strings
  EXPECT_EQ(5u, PoolInstance.findIndex("five"));
  EXPECT_TRUE(PoolInstance.isValidIndex(PoolInstance.findIndex("five")));
  EXPECT_FALSE(PoolInstance.isValidIndex(PoolInstance.findIndex("FIVE")));
}

} // namespace