clang/unittests/AST/ProfilingTest.cpp

   1 //===- unittests/AST/ProfilingTest.cpp --- Tests for Profiling ------===//
   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 "clang/AST/ASTContext.h"
  10 #include "clang/ASTMatchers/ASTMatchFinder.h"
  11 #include "clang/ASTMatchers/ASTMatchers.h"
  12 #include "clang/Tooling/Tooling.h"
  13 #include "gtest/gtest.h"
  14 #include <utility>
  15
  16 namespace clang {
  17 namespace {
  18 using namespace ast_matchers;
  19
  20 static auto getClassTemplateRedecls() {
  21   std::string Code = R"cpp(
  22     template <class> struct A;
  23     template <class> struct A;
  24     template <class> struct A;
  25   )cpp";
  26   auto AST = tooling::buildASTFromCode(Code);
  27   ASTContext &Ctx = AST->getASTContext();
  28
  29   auto MatchResults = match(classTemplateDecl().bind("id"), Ctx);
  30   SmallVector<ClassTemplateDecl *, 3> Res;
  31   for (BoundNodes &N : MatchResults) {
  32     if (auto *CTD = const_cast<ClassTemplateDecl *>(
  33             N.getNodeAs<ClassTemplateDecl>("id")))
  34       Res.push_back(CTD);
  35   }
  36   assert(Res.size() == 3);
  37 #ifndef NDEBUG
  38   for (auto &&I : Res)
  39     assert(I->getCanonicalDecl() == Res[0]);
  40 #endif
  41   return std::make_tuple(std::move(AST), Res[1], Res[2]);
  42 }
  43
  44 template <class T> static void testTypeNode(const T *T1, const T *T2) {
  45   {
  46     llvm::FoldingSetNodeID ID1, ID2;
  47     T1->Profile(ID1);
  48     T2->Profile(ID2);
  49     ASSERT_NE(ID1, ID2);
  50   }
  51   auto *CT1 = cast<T>(T1->getCanonicalTypeInternal());
  52   auto *CT2 = cast<T>(T2->getCanonicalTypeInternal());
  53   {
  54     llvm::FoldingSetNodeID ID1, ID2;
  55     CT1->Profile(ID1);
  56     CT2->Profile(ID2);
  57     ASSERT_EQ(ID1, ID2);
  58   }
  59 }
  60
  61 TEST(Profiling, DeducedTemplateSpecializationType_Name) {
  62   auto [AST, CTD1, CTD2] = getClassTemplateRedecls();
  63   ASTContext &Ctx = AST->getASTContext();
  64
  65   auto *T1 = cast<DeducedTemplateSpecializationType>(
  66       Ctx.getDeducedTemplateSpecializationType(TemplateName(CTD1), QualType(),
  67                                                false));
  68   auto *T2 = cast<DeducedTemplateSpecializationType>(
  69       Ctx.getDeducedTemplateSpecializationType(TemplateName(CTD2), QualType(),
  70                                                false));
  71   testTypeNode(T1, T2);
  72 }
  73
  74 } // namespace
  75 } // namespace clang