unittests/Demangle/ItaniumDemangleTest.cpp

   1 //===------------------ ItaniumDemangleTest.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/Demangle/ItaniumDemangle.h"
  10 #include "llvm/Support/Allocator.h"
  11 #include "gmock/gmock.h"
  12 #include "gtest/gtest.h"
  13 #include <cstdlib>
  14 #include <vector>
  15
  16 using namespace llvm;
  17 using namespace llvm::itanium_demangle;
  18
  19 namespace {
  20 class TestAllocator {
  21   BumpPtrAllocator Alloc;
  22
  23 public:
  24   void reset() { Alloc.Reset(); }
  25
  26   template <typename T, typename... Args> T *makeNode(Args &&... args) {
  27     return new (Alloc.Allocate(sizeof(T), alignof(T)))
  28         T(std::forward<Args>(args)...);
  29   }
  30
  31   void *allocateNodeArray(size_t sz) {
  32     return Alloc.Allocate(sizeof(Node *) * sz, alignof(Node *));
  33   }
  34 };
  35 } // namespace
  36
  37 TEST(ItaniumDemangle, MethodOverride) {
  38   struct TestParser : AbstractManglingParser<TestParser, TestAllocator> {
  39     std::vector<char> Types;
  40
  41     TestParser(const char *Str)
  42         : AbstractManglingParser(Str, Str + strlen(Str)) {}
  43
  44     Node *parseType() {
  45       Types.push_back(*First);
  46       return AbstractManglingParser<TestParser, TestAllocator>::parseType();
  47     }
  48   };
  49
  50   TestParser Parser("_Z1fIiEjl");
  51   ASSERT_NE(nullptr, Parser.parse());
  52   EXPECT_THAT(Parser.Types, testing::ElementsAre('i', 'j', 'l'));
  53 }