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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
//===- unittests/AST/TemplateNameTest.cpp --- Tests for TemplateName ------===//
//
// 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 "ASTPrint.h"
#include "clang/AST/ASTContext.h"
#include "clang/ASTMatchers/ASTMatchers.h"
#include "llvm/Support/raw_ostream.h"
#include "gtest/gtest.h"
namespace clang {
namespace {
using namespace ast_matchers;
std::string printTemplateName(TemplateName TN, const PrintingPolicy &Policy,
TemplateName::Qualified Qual) {
std::string Result;
llvm::raw_string_ostream Out(Result);
TN.print(Out, Policy, Qual);
return Out.str();
}
TEST(TemplateName, PrintUsingTemplate) {
std::string Code = R"cpp(
namespace std {
template <typename> struct vector {};
}
namespace absl { using std::vector; }
template<template <typename> class T> class X;
using absl::vector;
using A = X<vector>;
)cpp";
auto AST = tooling::buildASTFromCode(Code);
ASTContext &Ctx = AST->getASTContext();
// Match the template argument vector in X<vector>.
auto MatchResults = match(templateArgumentLoc().bind("id"), Ctx);
const auto *Template = selectFirst<TemplateArgumentLoc>("id", MatchResults);
ASSERT_TRUE(Template);
TemplateName TN = Template->getArgument().getAsTemplate();
EXPECT_EQ(TN.getKind(), TemplateName::UsingTemplate);
EXPECT_EQ(TN.getAsUsingShadowDecl()->getTargetDecl(), TN.getAsTemplateDecl());
EXPECT_EQ(printTemplateName(TN, Ctx.getPrintingPolicy(),
TemplateName::Qualified::Fully),
"std::vector");
EXPECT_EQ(printTemplateName(TN, Ctx.getPrintingPolicy(),
TemplateName::Qualified::AsWritten),
"vector");
EXPECT_EQ(printTemplateName(TN, Ctx.getPrintingPolicy(),
TemplateName::Qualified::None),
"vector");
}
TEST(TemplateName, QualifiedUsingTemplate) {
std::string Code = R"cpp(
namespace std {
template <typename> struct vector {};
}
namespace absl { using std::vector; }
template<template <typename> class T> class X;
using A = X<absl::vector>; // QualifiedTemplateName in a template argument.
)cpp";
auto AST = tooling::buildASTFromCode(Code);
// Match the template argument absl::vector in X<absl::vector>.
auto Matcher = templateArgumentLoc().bind("id");
auto MatchResults = match(Matcher, AST->getASTContext());
const auto *TAL = MatchResults.front().getNodeAs<TemplateArgumentLoc>("id");
ASSERT_TRUE(TAL);
TemplateName TN = TAL->getArgument().getAsTemplate();
EXPECT_EQ(TN.getKind(), TemplateName::QualifiedTemplate);
const auto *QTN = TN.getAsQualifiedTemplateName();
// Verify that we have the Using template name in the QualifiedTemplateName.
const auto *USD = QTN->getUnderlyingTemplate().getAsUsingShadowDecl();
EXPECT_TRUE(USD);
EXPECT_EQ(USD->getTargetDecl(), TN.getAsTemplateDecl());
EXPECT_EQ(TN.getAsUsingShadowDecl(), USD);
}
TEST(TemplateName, UsingTemplate) {
auto AST = tooling::buildASTFromCode(R"cpp(
namespace std {
template <typename T> struct vector { vector(T); };
}
namespace absl { using std::vector; }
// The "absl::vector<int>" is an elaborated TemplateSpecializationType with
// an inner Using TemplateName (not a Qualified TemplateName, the qualifiers
// are rather part of the ElaboratedType)!
absl::vector<int> v(123);
)cpp");
auto Matcher = elaboratedTypeLoc(
hasNamedTypeLoc(loc(templateSpecializationType().bind("id"))));
auto MatchResults = match(Matcher, AST->getASTContext());
const auto *TST =
MatchResults.front().getNodeAs<TemplateSpecializationType>("id");
ASSERT_TRUE(TST);
EXPECT_EQ(TST->getTemplateName().getKind(), TemplateName::UsingTemplate);
AST = tooling::buildASTFromCodeWithArgs(R"cpp(
namespace std {
template <typename T> struct vector { vector(T); };
}
namespace absl { using std::vector; }
// Similiar to the TemplateSpecializationType, absl::vector is an elaborated
// DeducedTemplateSpecializationType with an inner Using TemplateName!
absl::vector DTST(123);
)cpp",
{"-std=c++17"});
Matcher = elaboratedTypeLoc(
hasNamedTypeLoc(loc(deducedTemplateSpecializationType().bind("id"))));
MatchResults = match(Matcher, AST->getASTContext());
const auto *DTST =
MatchResults.front().getNodeAs<DeducedTemplateSpecializationType>("id");
ASSERT_TRUE(DTST);
EXPECT_EQ(DTST->getTemplateName().getKind(), TemplateName::UsingTemplate);
}
} // namespace
} // namespace clang
|
