llvm/unittests/DWARFLinkerParallel/StringPoolTest.cpp

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