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 }
54