1 //== unittests/ASTMatchers/ASTMatchersNodeTest.cpp - AST matcher unit tests ==//
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 "ASTMatchersTest.h"
10 #include "clang/AST/PrettyPrinter.h"
11 #include "clang/ASTMatchers/ASTMatchFinder.h"
12 #include "clang/ASTMatchers/ASTMatchers.h"
13 #include "clang/Tooling/Tooling.h"
14 #include "llvm/ADT/Triple.h"
15 #include "llvm/Support/Host.h"
16 #include "gtest/gtest.h"
17 
18 namespace clang {
19 namespace ast_matchers {
20 
TEST_P(ASTMatchersTest,Decl_CXX)21 TEST_P(ASTMatchersTest, Decl_CXX) {
22   if (!GetParam().isCXX()) {
23     // FIXME: Add a test for `decl()` that does not depend on C++.
24     return;
25   }
26   EXPECT_TRUE(notMatches("", decl(usingDecl())));
27   EXPECT_TRUE(
28       matches("namespace x { class X {}; } using x::X;", decl(usingDecl())));
29 }
30 
TEST_P(ASTMatchersTest,NameableDeclaration_MatchesVariousDecls)31 TEST_P(ASTMatchersTest, NameableDeclaration_MatchesVariousDecls) {
32   DeclarationMatcher NamedX = namedDecl(hasName("X"));
33   EXPECT_TRUE(matches("typedef int X;", NamedX));
34   EXPECT_TRUE(matches("int X;", NamedX));
35   EXPECT_TRUE(matches("void foo() { int X; }", NamedX));
36   EXPECT_TRUE(matches("enum X { A, B, C };", NamedX));
37 
38   EXPECT_TRUE(notMatches("#define X 1", NamedX));
39 }
40 
TEST_P(ASTMatchersTest,NamedDecl_CXX)41 TEST_P(ASTMatchersTest, NamedDecl_CXX) {
42   if (!GetParam().isCXX()) {
43     return;
44   }
45   DeclarationMatcher NamedX = namedDecl(hasName("X"));
46   EXPECT_TRUE(matches("class foo { virtual void X(); };", NamedX));
47   EXPECT_TRUE(matches("void foo() try { } catch(int X) { }", NamedX));
48   EXPECT_TRUE(matches("namespace X { }", NamedX));
49 }
50 
TEST_P(ASTMatchersTest,MatchesNameRE)51 TEST_P(ASTMatchersTest, MatchesNameRE) {
52   DeclarationMatcher NamedX = namedDecl(matchesName("::X"));
53   EXPECT_TRUE(matches("typedef int Xa;", NamedX));
54   EXPECT_TRUE(matches("int Xb;", NamedX));
55   EXPECT_TRUE(matches("void foo() { int Xgh; }", NamedX));
56   EXPECT_TRUE(matches("enum X { A, B, C };", NamedX));
57 
58   EXPECT_TRUE(notMatches("#define Xkl 1", NamedX));
59 
60   DeclarationMatcher StartsWithNo = namedDecl(matchesName("::no"));
61   EXPECT_TRUE(matches("int no_foo;", StartsWithNo));
62 
63   DeclarationMatcher Abc = namedDecl(matchesName("a.*b.*c"));
64   EXPECT_TRUE(matches("int abc;", Abc));
65   EXPECT_TRUE(matches("int aFOObBARc;", Abc));
66   EXPECT_TRUE(notMatches("int cab;", Abc));
67   EXPECT_TRUE(matches("int cabc;", Abc));
68 
69   DeclarationMatcher StartsWithK = namedDecl(matchesName(":k[^:]*$"));
70   EXPECT_TRUE(matches("int k;", StartsWithK));
71   EXPECT_TRUE(matches("int kAbc;", StartsWithK));
72 }
73 
TEST_P(ASTMatchersTest,MatchesNameRE_CXX)74 TEST_P(ASTMatchersTest, MatchesNameRE_CXX) {
75   if (!GetParam().isCXX()) {
76     return;
77   }
78   DeclarationMatcher NamedX = namedDecl(matchesName("::X"));
79   EXPECT_TRUE(matches("class foo { virtual void Xc(); };", NamedX));
80   EXPECT_TRUE(matches("void foo() try { } catch(int Xdef) { }", NamedX));
81   EXPECT_TRUE(matches("namespace Xij { }", NamedX));
82 
83   DeclarationMatcher StartsWithNo = namedDecl(matchesName("::no"));
84   EXPECT_TRUE(matches("class foo { virtual void nobody(); };", StartsWithNo));
85 
86   DeclarationMatcher StartsWithK = namedDecl(matchesName(":k[^:]*$"));
87   EXPECT_TRUE(matches("namespace x { int kTest; }", StartsWithK));
88   EXPECT_TRUE(matches("class C { int k; };", StartsWithK));
89   EXPECT_TRUE(notMatches("class C { int ckc; };", StartsWithK));
90   EXPECT_TRUE(notMatches("int K;", StartsWithK));
91 
92   DeclarationMatcher StartsWithKIgnoreCase =
93       namedDecl(matchesName(":k[^:]*$", llvm::Regex::IgnoreCase));
94   EXPECT_TRUE(matches("int k;", StartsWithKIgnoreCase));
95   EXPECT_TRUE(matches("int K;", StartsWithKIgnoreCase));
96 }
97 
TEST_P(ASTMatchersTest,DeclarationMatcher_MatchClass)98 TEST_P(ASTMatchersTest, DeclarationMatcher_MatchClass) {
99   if (!GetParam().isCXX()) {
100     return;
101   }
102 
103   DeclarationMatcher ClassX = recordDecl(recordDecl(hasName("X")));
104   EXPECT_TRUE(matches("class X;", ClassX));
105   EXPECT_TRUE(matches("class X {};", ClassX));
106   EXPECT_TRUE(matches("template<class T> class X {};", ClassX));
107   EXPECT_TRUE(notMatches("", ClassX));
108 }
109 
TEST_P(ASTMatchersTest,TranslationUnitDecl)110 TEST_P(ASTMatchersTest, TranslationUnitDecl) {
111   if (!GetParam().isCXX()) {
112     // FIXME: Add a test for `translationUnitDecl()` that does not depend on
113     // C++.
114     return;
115   }
116   StringRef Code = "int MyVar1;\n"
117                    "namespace NameSpace {\n"
118                    "int MyVar2;\n"
119                    "}  // namespace NameSpace\n";
120   EXPECT_TRUE(matches(
121       Code, varDecl(hasName("MyVar1"), hasDeclContext(translationUnitDecl()))));
122   EXPECT_FALSE(matches(
123       Code, varDecl(hasName("MyVar2"), hasDeclContext(translationUnitDecl()))));
124   EXPECT_TRUE(matches(
125       Code,
126       varDecl(hasName("MyVar2"),
127               hasDeclContext(decl(hasDeclContext(translationUnitDecl()))))));
128 }
129 
TEST_P(ASTMatchersTest,LinkageSpecDecl)130 TEST_P(ASTMatchersTest, LinkageSpecDecl) {
131   if (!GetParam().isCXX()) {
132     return;
133   }
134   EXPECT_TRUE(matches("extern \"C\" { void foo() {}; }", linkageSpecDecl()));
135   EXPECT_TRUE(notMatches("void foo() {};", linkageSpecDecl()));
136 }
137 
TEST_P(ASTMatchersTest,ClassTemplateDecl_DoesNotMatchClass)138 TEST_P(ASTMatchersTest, ClassTemplateDecl_DoesNotMatchClass) {
139   if (!GetParam().isCXX()) {
140     return;
141   }
142   DeclarationMatcher ClassX = classTemplateDecl(hasName("X"));
143   EXPECT_TRUE(notMatches("class X;", ClassX));
144   EXPECT_TRUE(notMatches("class X {};", ClassX));
145 }
146 
TEST_P(ASTMatchersTest,ClassTemplateDecl_MatchesClassTemplate)147 TEST_P(ASTMatchersTest, ClassTemplateDecl_MatchesClassTemplate) {
148   if (!GetParam().isCXX()) {
149     return;
150   }
151   DeclarationMatcher ClassX = classTemplateDecl(hasName("X"));
152   EXPECT_TRUE(matches("template<typename T> class X {};", ClassX));
153   EXPECT_TRUE(matches("class Z { template<class T> class X {}; };", ClassX));
154 }
155 
TEST_P(ASTMatchersTest,ClassTemplateDecl_DoesNotMatchClassTemplateExplicitSpecialization)156 TEST_P(ASTMatchersTest,
157        ClassTemplateDecl_DoesNotMatchClassTemplateExplicitSpecialization) {
158   if (!GetParam().isCXX()) {
159     return;
160   }
161   EXPECT_TRUE(notMatches(
162       "template<typename T> class X { };"
163       "template<> class X<int> { int a; };",
164       classTemplateDecl(hasName("X"), hasDescendant(fieldDecl(hasName("a"))))));
165 }
166 
TEST_P(ASTMatchersTest,ClassTemplateDecl_DoesNotMatchClassTemplatePartialSpecialization)167 TEST_P(ASTMatchersTest,
168        ClassTemplateDecl_DoesNotMatchClassTemplatePartialSpecialization) {
169   if (!GetParam().isCXX()) {
170     return;
171   }
172   EXPECT_TRUE(notMatches(
173       "template<typename T, typename U> class X { };"
174       "template<typename T> class X<T, int> { int a; };",
175       classTemplateDecl(hasName("X"), hasDescendant(fieldDecl(hasName("a"))))));
176 }
177 
TEST(ASTMatchersTestCUDA,CUDAKernelCallExpr)178 TEST(ASTMatchersTestCUDA, CUDAKernelCallExpr) {
179   EXPECT_TRUE(matchesWithCuda("__global__ void f() { }"
180                               "void g() { f<<<1, 2>>>(); }",
181                               cudaKernelCallExpr()));
182   EXPECT_TRUE(notMatchesWithCuda("void f() {}", cudaKernelCallExpr()));
183 }
184 
TEST(ASTMatchersTestCUDA,HasAttrCUDA)185 TEST(ASTMatchersTestCUDA, HasAttrCUDA) {
186   EXPECT_TRUE(matchesWithCuda("__attribute__((device)) void f() {}",
187                               hasAttr(clang::attr::CUDADevice)));
188   EXPECT_FALSE(notMatchesWithCuda("__attribute__((global)) void f() {}",
189                                   hasAttr(clang::attr::CUDAGlobal)));
190 }
191 
TEST_P(ASTMatchersTest,ValueDecl)192 TEST_P(ASTMatchersTest, ValueDecl) {
193   if (!GetParam().isCXX()) {
194     // FIXME: Fix this test in non-C++ language modes.
195     return;
196   }
197   EXPECT_TRUE(matches("enum EnumType { EnumValue };",
198                       valueDecl(hasType(asString("enum EnumType")))));
199   EXPECT_TRUE(matches("void FunctionDecl();",
200                       valueDecl(hasType(asString("void (void)")))));
201 }
202 
TEST_P(ASTMatchersTest,FriendDecl)203 TEST_P(ASTMatchersTest, FriendDecl) {
204   if (!GetParam().isCXX()) {
205     return;
206   }
207   EXPECT_TRUE(matches("class Y { friend class X; };",
208                       friendDecl(hasType(asString("class X")))));
209   EXPECT_TRUE(matches("class Y { friend class X; };",
210                       friendDecl(hasType(recordDecl(hasName("X"))))));
211 
212   EXPECT_TRUE(matches("class Y { friend void f(); };",
213                       functionDecl(hasName("f"), hasParent(friendDecl()))));
214 }
215 
TEST_P(ASTMatchersTest,EnumDecl_DoesNotMatchClasses)216 TEST_P(ASTMatchersTest, EnumDecl_DoesNotMatchClasses) {
217   if (!GetParam().isCXX()) {
218     return;
219   }
220   EXPECT_TRUE(notMatches("class X {};", enumDecl(hasName("X"))));
221 }
222 
TEST_P(ASTMatchersTest,EnumDecl_MatchesEnums)223 TEST_P(ASTMatchersTest, EnumDecl_MatchesEnums) {
224   if (!GetParam().isCXX()) {
225     // FIXME: Fix this test in non-C++ language modes.
226     return;
227   }
228   EXPECT_TRUE(matches("enum X {};", enumDecl(hasName("X"))));
229 }
230 
TEST_P(ASTMatchersTest,EnumConstantDecl)231 TEST_P(ASTMatchersTest, EnumConstantDecl) {
232   if (!GetParam().isCXX()) {
233     // FIXME: Fix this test in non-C++ language modes.
234     return;
235   }
236   DeclarationMatcher Matcher = enumConstantDecl(hasName("A"));
237   EXPECT_TRUE(matches("enum X{ A };", Matcher));
238   EXPECT_TRUE(notMatches("enum X{ B };", Matcher));
239   EXPECT_TRUE(notMatches("enum X {};", Matcher));
240 }
241 
TEST_P(ASTMatchersTest,TagDecl)242 TEST_P(ASTMatchersTest, TagDecl) {
243   if (!GetParam().isCXX()) {
244     // FIXME: Fix this test in non-C++ language modes.
245     return;
246   }
247   EXPECT_TRUE(matches("struct X {};", tagDecl(hasName("X"))));
248   EXPECT_TRUE(matches("union U {};", tagDecl(hasName("U"))));
249   EXPECT_TRUE(matches("enum E {};", tagDecl(hasName("E"))));
250 }
251 
TEST_P(ASTMatchersTest,TagDecl_CXX)252 TEST_P(ASTMatchersTest, TagDecl_CXX) {
253   if (!GetParam().isCXX()) {
254     return;
255   }
256   EXPECT_TRUE(matches("class C {};", tagDecl(hasName("C"))));
257 }
258 
TEST_P(ASTMatchersTest,UnresolvedLookupExpr)259 TEST_P(ASTMatchersTest, UnresolvedLookupExpr) {
260   if (!GetParam().isCXX() || GetParam().hasDelayedTemplateParsing()) {
261     // FIXME: Fix this test to work with delayed template parsing.
262     return;
263   }
264 
265   EXPECT_TRUE(matches("template<typename T>"
266                       "T foo() { T a; return a; }"
267                       "template<typename T>"
268                       "void bar() {"
269                       "  foo<T>();"
270                       "}",
271                       unresolvedLookupExpr()));
272 }
273 
TEST_P(ASTMatchersTest,UsesADL)274 TEST_P(ASTMatchersTest, UsesADL) {
275   if (!GetParam().isCXX()) {
276     return;
277   }
278 
279   StatementMatcher ADLMatch = callExpr(usesADL());
280   StatementMatcher ADLMatchOper = cxxOperatorCallExpr(usesADL());
281   StringRef NS_Str = R"cpp(
282   namespace NS {
283     struct X {};
284     void f(X);
285     void operator+(X, X);
286   }
287   struct MyX {};
288   void f(...);
289   void operator+(MyX, MyX);
290 )cpp";
291 
292   auto MkStr = [&](StringRef Body) {
293     return (NS_Str + "void test_fn() { " + Body + " }").str();
294   };
295 
296   EXPECT_TRUE(matches(MkStr("NS::X x; f(x);"), ADLMatch));
297   EXPECT_TRUE(notMatches(MkStr("NS::X x; NS::f(x);"), ADLMatch));
298   EXPECT_TRUE(notMatches(MkStr("MyX x; f(x);"), ADLMatch));
299   EXPECT_TRUE(notMatches(MkStr("NS::X x; using NS::f; f(x);"), ADLMatch));
300 
301   // Operator call expressions
302   EXPECT_TRUE(matches(MkStr("NS::X x; x + x;"), ADLMatch));
303   EXPECT_TRUE(matches(MkStr("NS::X x; x + x;"), ADLMatchOper));
304   EXPECT_TRUE(notMatches(MkStr("MyX x; x + x;"), ADLMatch));
305   EXPECT_TRUE(notMatches(MkStr("MyX x; x + x;"), ADLMatchOper));
306   EXPECT_TRUE(matches(MkStr("NS::X x; operator+(x, x);"), ADLMatch));
307   EXPECT_TRUE(notMatches(MkStr("NS::X x; NS::operator+(x, x);"), ADLMatch));
308 }
309 
TEST_P(ASTMatchersTest,CallExpr_CXX)310 TEST_P(ASTMatchersTest, CallExpr_CXX) {
311   if (!GetParam().isCXX()) {
312     // FIXME: Add a test for `callExpr()` that does not depend on C++.
313     return;
314   }
315   // FIXME: Do we want to overload Call() to directly take
316   // Matcher<Decl>, too?
317   StatementMatcher MethodX =
318       callExpr(hasDeclaration(cxxMethodDecl(hasName("x"))));
319 
320   EXPECT_TRUE(matches("class Y { void x() { x(); } };", MethodX));
321   EXPECT_TRUE(notMatches("class Y { void x() {} };", MethodX));
322 
323   StatementMatcher MethodOnY =
324       cxxMemberCallExpr(on(hasType(recordDecl(hasName("Y")))));
325 
326   EXPECT_TRUE(matches("class Y { public: void x(); }; void z() { Y y; y.x(); }",
327                       MethodOnY));
328   EXPECT_TRUE(matches("class Y { public: void x(); }; void z(Y &y) { y.x(); }",
329                       MethodOnY));
330   EXPECT_TRUE(notMatches(
331       "class Y { public: void x(); }; void z(Y *&y) { y->x(); }", MethodOnY));
332   EXPECT_TRUE(notMatches(
333       "class Y { public: void x(); }; void z(Y y[]) { y->x(); }", MethodOnY));
334   EXPECT_TRUE(notMatches(
335       "class Y { public: void x(); }; void z() { Y *y; y->x(); }", MethodOnY));
336 
337   StatementMatcher MethodOnYPointer =
338       cxxMemberCallExpr(on(hasType(pointsTo(recordDecl(hasName("Y"))))));
339 
340   EXPECT_TRUE(
341       matches("class Y { public: void x(); }; void z() { Y *y; y->x(); }",
342               MethodOnYPointer));
343   EXPECT_TRUE(
344       matches("class Y { public: void x(); }; void z(Y *&y) { y->x(); }",
345               MethodOnYPointer));
346   EXPECT_TRUE(
347       matches("class Y { public: void x(); }; void z(Y y[]) { y->x(); }",
348               MethodOnYPointer));
349   EXPECT_TRUE(
350       notMatches("class Y { public: void x(); }; void z() { Y y; y.x(); }",
351                  MethodOnYPointer));
352   EXPECT_TRUE(
353       notMatches("class Y { public: void x(); }; void z(Y &y) { y.x(); }",
354                  MethodOnYPointer));
355 }
356 
TEST_P(ASTMatchersTest,LambdaExpr)357 TEST_P(ASTMatchersTest, LambdaExpr) {
358   if (!GetParam().isCXX11OrLater()) {
359     return;
360   }
361   EXPECT_TRUE(matches("auto f = [] (int i) { return i; };", lambdaExpr()));
362 }
363 
TEST_P(ASTMatchersTest,CXXForRangeStmt)364 TEST_P(ASTMatchersTest, CXXForRangeStmt) {
365   EXPECT_TRUE(
366       notMatches("void f() { for (int i; i<5; ++i); }", cxxForRangeStmt()));
367 }
368 
TEST_P(ASTMatchersTest,CXXForRangeStmt_CXX11)369 TEST_P(ASTMatchersTest, CXXForRangeStmt_CXX11) {
370   if (!GetParam().isCXX11OrLater()) {
371     return;
372   }
373   EXPECT_TRUE(matches("int as[] = { 1, 2, 3 };"
374                       "void f() { for (auto &a : as); }",
375                       cxxForRangeStmt()));
376 }
377 
TEST_P(ASTMatchersTest,SubstNonTypeTemplateParmExpr)378 TEST_P(ASTMatchersTest, SubstNonTypeTemplateParmExpr) {
379   if (!GetParam().isCXX()) {
380     return;
381   }
382   EXPECT_FALSE(matches("template<int N>\n"
383                        "struct A {  static const int n = 0; };\n"
384                        "struct B : public A<42> {};",
385                        traverse(TK_AsIs, substNonTypeTemplateParmExpr())));
386   EXPECT_TRUE(matches("template<int N>\n"
387                       "struct A {  static const int n = N; };\n"
388                       "struct B : public A<42> {};",
389                       traverse(TK_AsIs, substNonTypeTemplateParmExpr())));
390 }
391 
TEST_P(ASTMatchersTest,NonTypeTemplateParmDecl)392 TEST_P(ASTMatchersTest, NonTypeTemplateParmDecl) {
393   if (!GetParam().isCXX()) {
394     return;
395   }
396   EXPECT_TRUE(matches("template <int N> void f();",
397                       nonTypeTemplateParmDecl(hasName("N"))));
398   EXPECT_TRUE(
399       notMatches("template <typename T> void f();", nonTypeTemplateParmDecl()));
400 }
401 
TEST_P(ASTMatchersTest,TemplateTypeParmDecl)402 TEST_P(ASTMatchersTest, TemplateTypeParmDecl) {
403   if (!GetParam().isCXX()) {
404     return;
405   }
406   EXPECT_TRUE(matches("template <typename T> void f();",
407                       templateTypeParmDecl(hasName("T"))));
408   EXPECT_TRUE(notMatches("template <int N> void f();", templateTypeParmDecl()));
409 }
410 
TEST_P(ASTMatchersTest,TemplateTemplateParmDecl)411 TEST_P(ASTMatchersTest, TemplateTemplateParmDecl) {
412   if (!GetParam().isCXX())
413     return;
414   EXPECT_TRUE(matches("template <template <typename> class Z> void f();",
415                       templateTemplateParmDecl(hasName("Z"))));
416   EXPECT_TRUE(notMatches("template <typename, int> void f();",
417                          templateTemplateParmDecl()));
418 }
419 
TEST_P(ASTMatchersTest,UserDefinedLiteral)420 TEST_P(ASTMatchersTest, UserDefinedLiteral) {
421   if (!GetParam().isCXX11OrLater()) {
422     return;
423   }
424   EXPECT_TRUE(matches("constexpr char operator \"\" _inc (const char i) {"
425                       "  return i + 1;"
426                       "}"
427                       "char c = 'a'_inc;",
428                       userDefinedLiteral()));
429 }
430 
TEST_P(ASTMatchersTest,FlowControl)431 TEST_P(ASTMatchersTest, FlowControl) {
432   EXPECT_TRUE(matches("void f() { while(1) { break; } }", breakStmt()));
433   EXPECT_TRUE(matches("void f() { while(1) { continue; } }", continueStmt()));
434   EXPECT_TRUE(matches("void f() { goto FOO; FOO: ;}", gotoStmt()));
435   EXPECT_TRUE(matches("void f() { goto FOO; FOO: ;}",
436                       labelStmt(hasDeclaration(labelDecl(hasName("FOO"))))));
437   EXPECT_TRUE(matches("void f() { FOO: ; void *ptr = &&FOO; goto *ptr; }",
438                       addrLabelExpr()));
439   EXPECT_TRUE(matches("void f() { return; }", returnStmt()));
440 }
441 
TEST_P(ASTMatchersTest,CXXOperatorCallExpr)442 TEST_P(ASTMatchersTest, CXXOperatorCallExpr) {
443   if (!GetParam().isCXX()) {
444     return;
445   }
446 
447   StatementMatcher OpCall = cxxOperatorCallExpr();
448   // Unary operator
449   EXPECT_TRUE(matches("class Y { }; "
450                       "bool operator!(Y x) { return false; }; "
451                       "Y y; bool c = !y;",
452                       OpCall));
453   // No match -- special operators like "new", "delete"
454   // FIXME: operator new takes size_t, for which we need stddef.h, for which
455   // we need to figure out include paths in the test.
456   // EXPECT_TRUE(NotMatches("#include <stddef.h>\n"
457   //             "class Y { }; "
458   //             "void *operator new(size_t size) { return 0; } "
459   //             "Y *y = new Y;", OpCall));
460   EXPECT_TRUE(notMatches("class Y { }; "
461                          "void operator delete(void *p) { } "
462                          "void a() {Y *y = new Y; delete y;}",
463                          OpCall));
464   // Binary operator
465   EXPECT_TRUE(matches("class Y { }; "
466                       "bool operator&&(Y x, Y y) { return true; }; "
467                       "Y a; Y b; bool c = a && b;",
468                       OpCall));
469   // No match -- normal operator, not an overloaded one.
470   EXPECT_TRUE(notMatches("bool x = true, y = true; bool t = x && y;", OpCall));
471   EXPECT_TRUE(notMatches("int t = 5 << 2;", OpCall));
472 }
473 
TEST_P(ASTMatchersTest,ThisPointerType)474 TEST_P(ASTMatchersTest, ThisPointerType) {
475   if (!GetParam().isCXX()) {
476     return;
477   }
478 
479   StatementMatcher MethodOnY = traverse(
480       TK_AsIs, cxxMemberCallExpr(thisPointerType(recordDecl(hasName("Y")))));
481 
482   EXPECT_TRUE(matches("class Y { public: void x(); }; void z() { Y y; y.x(); }",
483                       MethodOnY));
484   EXPECT_TRUE(matches("class Y { public: void x(); }; void z(Y &y) { y.x(); }",
485                       MethodOnY));
486   EXPECT_TRUE(matches(
487       "class Y { public: void x(); }; void z(Y *&y) { y->x(); }", MethodOnY));
488   EXPECT_TRUE(matches(
489       "class Y { public: void x(); }; void z(Y y[]) { y->x(); }", MethodOnY));
490   EXPECT_TRUE(matches(
491       "class Y { public: void x(); }; void z() { Y *y; y->x(); }", MethodOnY));
492 
493   EXPECT_TRUE(matches("class Y {"
494                       "  public: virtual void x();"
495                       "};"
496                       "class X : public Y {"
497                       "  public: virtual void x();"
498                       "};"
499                       "void z() { X *x; x->Y::x(); }",
500                       MethodOnY));
501 }
502 
TEST_P(ASTMatchersTest,DeclRefExpr)503 TEST_P(ASTMatchersTest, DeclRefExpr) {
504   if (!GetParam().isCXX()) {
505     // FIXME: Add a test for `declRefExpr()` that does not depend on C++.
506     return;
507   }
508   StatementMatcher Reference = declRefExpr(to(varDecl(hasInitializer(
509       cxxMemberCallExpr(thisPointerType(recordDecl(hasName("Y"))))))));
510 
511   EXPECT_TRUE(matches("class Y {"
512                       " public:"
513                       "  bool x() const;"
514                       "};"
515                       "void z(const Y &y) {"
516                       "  bool b = y.x();"
517                       "  if (b) {}"
518                       "}",
519                       Reference));
520 
521   EXPECT_TRUE(notMatches("class Y {"
522                          " public:"
523                          "  bool x() const;"
524                          "};"
525                          "void z(const Y &y) {"
526                          "  bool b = y.x();"
527                          "}",
528                          Reference));
529 }
530 
TEST_P(ASTMatchersTest,CXXMemberCallExpr)531 TEST_P(ASTMatchersTest, CXXMemberCallExpr) {
532   if (!GetParam().isCXX()) {
533     return;
534   }
535   StatementMatcher CallOnVariableY =
536       cxxMemberCallExpr(on(declRefExpr(to(varDecl(hasName("y"))))));
537 
538   EXPECT_TRUE(matches("class Y { public: void x() { Y y; y.x(); } };",
539                       CallOnVariableY));
540   EXPECT_TRUE(matches("class Y { public: void x() const { Y y; y.x(); } };",
541                       CallOnVariableY));
542   EXPECT_TRUE(matches("class Y { public: void x(); };"
543                       "class X : public Y { void z() { X y; y.x(); } };",
544                       CallOnVariableY));
545   EXPECT_TRUE(matches("class Y { public: void x(); };"
546                       "class X : public Y { void z() { X *y; y->x(); } };",
547                       CallOnVariableY));
548   EXPECT_TRUE(notMatches(
549       "class Y { public: void x(); };"
550       "class X : public Y { void z() { unsigned long y; ((X*)y)->x(); } };",
551       CallOnVariableY));
552 }
553 
TEST_P(ASTMatchersTest,UnaryExprOrTypeTraitExpr)554 TEST_P(ASTMatchersTest, UnaryExprOrTypeTraitExpr) {
555   EXPECT_TRUE(
556       matches("void x() { int a = sizeof(a); }", unaryExprOrTypeTraitExpr()));
557 }
558 
TEST_P(ASTMatchersTest,AlignOfExpr)559 TEST_P(ASTMatchersTest, AlignOfExpr) {
560   EXPECT_TRUE(
561       notMatches("void x() { int a = sizeof(a); }", alignOfExpr(anything())));
562   // FIXME: Uncomment once alignof is enabled.
563   // EXPECT_TRUE(matches("void x() { int a = alignof(a); }",
564   //                     unaryExprOrTypeTraitExpr()));
565   // EXPECT_TRUE(notMatches("void x() { int a = alignof(a); }",
566   //                        sizeOfExpr()));
567 }
568 
TEST_P(ASTMatchersTest,MemberExpr_DoesNotMatchClasses)569 TEST_P(ASTMatchersTest, MemberExpr_DoesNotMatchClasses) {
570   if (!GetParam().isCXX()) {
571     return;
572   }
573   EXPECT_TRUE(notMatches("class Y { void x() {} };", memberExpr()));
574   EXPECT_TRUE(notMatches("class Y { void x() {} };", unresolvedMemberExpr()));
575   EXPECT_TRUE(
576       notMatches("class Y { void x() {} };", cxxDependentScopeMemberExpr()));
577 }
578 
TEST_P(ASTMatchersTest,MemberExpr_MatchesMemberFunctionCall)579 TEST_P(ASTMatchersTest, MemberExpr_MatchesMemberFunctionCall) {
580   if (!GetParam().isCXX() || GetParam().hasDelayedTemplateParsing()) {
581     // FIXME: Fix this test to work with delayed template parsing.
582     return;
583   }
584   EXPECT_TRUE(matches("class Y { void x() { x(); } };", memberExpr()));
585   EXPECT_TRUE(matches("class Y { template <class T> void x() { x<T>(); } };",
586                       unresolvedMemberExpr()));
587   EXPECT_TRUE(matches("template <class T> void x() { T t; t.f(); }",
588                       cxxDependentScopeMemberExpr()));
589 }
590 
TEST_P(ASTMatchersTest,MemberExpr_MatchesVariable)591 TEST_P(ASTMatchersTest, MemberExpr_MatchesVariable) {
592   if (!GetParam().isCXX() || GetParam().hasDelayedTemplateParsing()) {
593     // FIXME: Fix this test to work with delayed template parsing.
594     return;
595   }
596   EXPECT_TRUE(
597       matches("class Y { void x() { this->y; } int y; };", memberExpr()));
598   EXPECT_TRUE(matches("class Y { void x() { y; } int y; };", memberExpr()));
599   EXPECT_TRUE(
600       matches("class Y { void x() { Y y; y.y; } int y; };", memberExpr()));
601   EXPECT_TRUE(matches("template <class T>"
602                       "class X : T { void f() { this->T::v; } };",
603                       cxxDependentScopeMemberExpr()));
604   EXPECT_TRUE(matches("template <class T> class X : T { void f() { T::v; } };",
605                       cxxDependentScopeMemberExpr()));
606   EXPECT_TRUE(matches("template <class T> void x() { T t; t.v; }",
607                       cxxDependentScopeMemberExpr()));
608 }
609 
TEST_P(ASTMatchersTest,MemberExpr_MatchesStaticVariable)610 TEST_P(ASTMatchersTest, MemberExpr_MatchesStaticVariable) {
611   if (!GetParam().isCXX()) {
612     return;
613   }
614   EXPECT_TRUE(matches("class Y { void x() { this->y; } static int y; };",
615                       memberExpr()));
616   EXPECT_TRUE(
617       notMatches("class Y { void x() { y; } static int y; };", memberExpr()));
618   EXPECT_TRUE(notMatches("class Y { void x() { Y::y; } static int y; };",
619                          memberExpr()));
620 }
621 
TEST_P(ASTMatchersTest,FunctionDecl)622 TEST_P(ASTMatchersTest, FunctionDecl) {
623   StatementMatcher CallFunctionF = callExpr(callee(functionDecl(hasName("f"))));
624 
625   EXPECT_TRUE(matches("void f() { f(); }", CallFunctionF));
626   EXPECT_TRUE(notMatches("void f() { }", CallFunctionF));
627 
628   EXPECT_TRUE(notMatches("void f(int);", functionDecl(isVariadic())));
629   EXPECT_TRUE(notMatches("void f();", functionDecl(isVariadic())));
630   EXPECT_TRUE(matches("void f(int, ...);", functionDecl(parameterCountIs(1))));
631 }
632 
TEST_P(ASTMatchersTest,FunctionDecl_C)633 TEST_P(ASTMatchersTest, FunctionDecl_C) {
634   if (!GetParam().isC()) {
635     return;
636   }
637   EXPECT_TRUE(notMatches("void f();", functionDecl(isVariadic())));
638   EXPECT_TRUE(matches("void f();", functionDecl(parameterCountIs(0))));
639 }
640 
TEST_P(ASTMatchersTest,FunctionDecl_CXX)641 TEST_P(ASTMatchersTest, FunctionDecl_CXX) {
642   if (!GetParam().isCXX()) {
643     return;
644   }
645 
646   StatementMatcher CallFunctionF = callExpr(callee(functionDecl(hasName("f"))));
647 
648   if (!GetParam().hasDelayedTemplateParsing()) {
649     // FIXME: Fix this test to work with delayed template parsing.
650     // Dependent contexts, but a non-dependent call.
651     EXPECT_TRUE(
652         matches("void f(); template <int N> void g() { f(); }", CallFunctionF));
653     EXPECT_TRUE(
654         matches("void f(); template <int N> struct S { void g() { f(); } };",
655                 CallFunctionF));
656   }
657 
658   // Depedent calls don't match.
659   EXPECT_TRUE(
660       notMatches("void f(int); template <typename T> void g(T t) { f(t); }",
661                  CallFunctionF));
662   EXPECT_TRUE(
663       notMatches("void f(int);"
664                  "template <typename T> struct S { void g(T t) { f(t); } };",
665                  CallFunctionF));
666 
667   EXPECT_TRUE(matches("void f(...);", functionDecl(isVariadic())));
668   EXPECT_TRUE(matches("void f(...);", functionDecl(parameterCountIs(0))));
669 }
670 
TEST_P(ASTMatchersTest,FunctionDecl_CXX11)671 TEST_P(ASTMatchersTest, FunctionDecl_CXX11) {
672   if (!GetParam().isCXX11OrLater()) {
673     return;
674   }
675 
676   EXPECT_TRUE(notMatches("template <typename... Ts> void f(Ts...);",
677                          functionDecl(isVariadic())));
678 }
679 
TEST_P(ASTMatchersTest,FunctionTemplateDecl_MatchesFunctionTemplateDeclarations)680 TEST_P(ASTMatchersTest,
681        FunctionTemplateDecl_MatchesFunctionTemplateDeclarations) {
682   if (!GetParam().isCXX()) {
683     return;
684   }
685   EXPECT_TRUE(matches("template <typename T> void f(T t) {}",
686                       functionTemplateDecl(hasName("f"))));
687 }
688 
TEST_P(ASTMatchersTest,FunctionTemplate_DoesNotMatchFunctionDeclarations)689 TEST_P(ASTMatchersTest, FunctionTemplate_DoesNotMatchFunctionDeclarations) {
690   EXPECT_TRUE(
691       notMatches("void f(double d);", functionTemplateDecl(hasName("f"))));
692   EXPECT_TRUE(
693       notMatches("void f(int t) {}", functionTemplateDecl(hasName("f"))));
694 }
695 
TEST_P(ASTMatchersTest,FunctionTemplateDecl_DoesNotMatchFunctionTemplateSpecializations)696 TEST_P(ASTMatchersTest,
697        FunctionTemplateDecl_DoesNotMatchFunctionTemplateSpecializations) {
698   if (!GetParam().isCXX()) {
699     return;
700   }
701   EXPECT_TRUE(notMatches(
702       "void g(); template <typename T> void f(T t) {}"
703       "template <> void f(int t) { g(); }",
704       functionTemplateDecl(hasName("f"), hasDescendant(declRefExpr(to(
705                                              functionDecl(hasName("g"))))))));
706 }
707 
TEST_P(ASTMatchersTest,ClassTemplateSpecializationDecl)708 TEST_P(ASTMatchersTest, ClassTemplateSpecializationDecl) {
709   if (!GetParam().isCXX()) {
710     return;
711   }
712   EXPECT_TRUE(matches("template<typename T> struct A {};"
713                       "template<> struct A<int> {};",
714                       classTemplateSpecializationDecl()));
715   EXPECT_TRUE(matches("template<typename T> struct A {}; A<int> a;",
716                       classTemplateSpecializationDecl()));
717   EXPECT_TRUE(notMatches("template<typename T> struct A {};",
718                          classTemplateSpecializationDecl()));
719 }
720 
TEST_P(ASTMatchersTest,DeclaratorDecl)721 TEST_P(ASTMatchersTest, DeclaratorDecl) {
722   EXPECT_TRUE(matches("int x;", declaratorDecl()));
723   EXPECT_TRUE(notMatches("struct A {};", declaratorDecl()));
724 }
725 
TEST_P(ASTMatchersTest,DeclaratorDecl_CXX)726 TEST_P(ASTMatchersTest, DeclaratorDecl_CXX) {
727   if (!GetParam().isCXX()) {
728     return;
729   }
730   EXPECT_TRUE(notMatches("class A {};", declaratorDecl()));
731 }
732 
TEST_P(ASTMatchersTest,ParmVarDecl)733 TEST_P(ASTMatchersTest, ParmVarDecl) {
734   EXPECT_TRUE(matches("void f(int x);", parmVarDecl()));
735   EXPECT_TRUE(notMatches("void f();", parmVarDecl()));
736 }
737 
TEST_P(ASTMatchersTest,Matcher_ConstructorCall)738 TEST_P(ASTMatchersTest, Matcher_ConstructorCall) {
739   if (!GetParam().isCXX()) {
740     return;
741   }
742 
743   StatementMatcher Constructor = traverse(TK_AsIs, cxxConstructExpr());
744 
745   EXPECT_TRUE(
746       matches("class X { public: X(); }; void x() { X x; }", Constructor));
747   EXPECT_TRUE(matches("class X { public: X(); }; void x() { X x = X(); }",
748                       Constructor));
749   EXPECT_TRUE(matches("class X { public: X(int); }; void x() { X x = 0; }",
750                       Constructor));
751   EXPECT_TRUE(matches("class X {}; void x(int) { X x; }", Constructor));
752 }
753 
TEST_P(ASTMatchersTest,Match_ConstructorInitializers)754 TEST_P(ASTMatchersTest, Match_ConstructorInitializers) {
755   if (!GetParam().isCXX()) {
756     return;
757   }
758   EXPECT_TRUE(matches("class C { int i; public: C(int ii) : i(ii) {} };",
759                       cxxCtorInitializer(forField(hasName("i")))));
760 }
761 
TEST_P(ASTMatchersTest,Matcher_ThisExpr)762 TEST_P(ASTMatchersTest, Matcher_ThisExpr) {
763   if (!GetParam().isCXX()) {
764     return;
765   }
766   EXPECT_TRUE(
767       matches("struct X { int a; int f () { return a; } };", cxxThisExpr()));
768   EXPECT_TRUE(
769       notMatches("struct X { int f () { int a; return a; } };", cxxThisExpr()));
770 }
771 
TEST_P(ASTMatchersTest,Matcher_BindTemporaryExpression)772 TEST_P(ASTMatchersTest, Matcher_BindTemporaryExpression) {
773   if (!GetParam().isCXX()) {
774     return;
775   }
776 
777   StatementMatcher TempExpression = traverse(TK_AsIs, cxxBindTemporaryExpr());
778 
779   StringRef ClassString = "class string { public: string(); ~string(); }; ";
780 
781   EXPECT_TRUE(matches(
782       ClassString + "string GetStringByValue();"
783                     "void FunctionTakesString(string s);"
784                     "void run() { FunctionTakesString(GetStringByValue()); }",
785       TempExpression));
786 
787   EXPECT_TRUE(notMatches(ClassString +
788                              "string* GetStringPointer(); "
789                              "void FunctionTakesStringPtr(string* s);"
790                              "void run() {"
791                              "  string* s = GetStringPointer();"
792                              "  FunctionTakesStringPtr(GetStringPointer());"
793                              "  FunctionTakesStringPtr(s);"
794                              "}",
795                          TempExpression));
796 
797   EXPECT_TRUE(notMatches("class no_dtor {};"
798                          "no_dtor GetObjByValue();"
799                          "void ConsumeObj(no_dtor param);"
800                          "void run() { ConsumeObj(GetObjByValue()); }",
801                          TempExpression));
802 }
803 
TEST_P(ASTMatchersTest,MaterializeTemporaryExpr_MatchesTemporaryCXX11CXX14)804 TEST_P(ASTMatchersTest, MaterializeTemporaryExpr_MatchesTemporaryCXX11CXX14) {
805   if (GetParam().Language != Lang_CXX11 && GetParam().Language != Lang_CXX14) {
806     return;
807   }
808 
809   StatementMatcher TempExpression =
810       traverse(TK_AsIs, materializeTemporaryExpr());
811 
812   EXPECT_TRUE(matches("class string { public: string(); }; "
813                       "string GetStringByValue();"
814                       "void FunctionTakesString(string s);"
815                       "void run() { FunctionTakesString(GetStringByValue()); }",
816                       TempExpression));
817 }
818 
TEST_P(ASTMatchersTest,MaterializeTemporaryExpr_MatchesTemporary)819 TEST_P(ASTMatchersTest, MaterializeTemporaryExpr_MatchesTemporary) {
820   if (!GetParam().isCXX()) {
821     return;
822   }
823 
824   StringRef ClassString = "class string { public: string(); int length(); }; ";
825   StatementMatcher TempExpression =
826       traverse(TK_AsIs, materializeTemporaryExpr());
827 
828   EXPECT_TRUE(notMatches(ClassString +
829                              "string* GetStringPointer(); "
830                              "void FunctionTakesStringPtr(string* s);"
831                              "void run() {"
832                              "  string* s = GetStringPointer();"
833                              "  FunctionTakesStringPtr(GetStringPointer());"
834                              "  FunctionTakesStringPtr(s);"
835                              "}",
836                          TempExpression));
837 
838   EXPECT_TRUE(matches(ClassString +
839                           "string GetStringByValue();"
840                           "void run() { int k = GetStringByValue().length(); }",
841                       TempExpression));
842 
843   EXPECT_TRUE(notMatches(ClassString + "string GetStringByValue();"
844                                        "void run() { GetStringByValue(); }",
845                          TempExpression));
846 }
847 
TEST_P(ASTMatchersTest,Matcher_NewExpression)848 TEST_P(ASTMatchersTest, Matcher_NewExpression) {
849   if (!GetParam().isCXX()) {
850     return;
851   }
852 
853   StatementMatcher New = cxxNewExpr();
854 
855   EXPECT_TRUE(matches("class X { public: X(); }; void x() { new X; }", New));
856   EXPECT_TRUE(matches("class X { public: X(); }; void x() { new X(); }", New));
857   EXPECT_TRUE(
858       matches("class X { public: X(int); }; void x() { new X(0); }", New));
859   EXPECT_TRUE(matches("class X {}; void x(int) { new X; }", New));
860 }
861 
TEST_P(ASTMatchersTest,Matcher_DeleteExpression)862 TEST_P(ASTMatchersTest, Matcher_DeleteExpression) {
863   if (!GetParam().isCXX()) {
864     return;
865   }
866   EXPECT_TRUE(
867       matches("struct A {}; void f(A* a) { delete a; }", cxxDeleteExpr()));
868 }
869 
TEST_P(ASTMatchersTest,Matcher_NoexceptExpression)870 TEST_P(ASTMatchersTest, Matcher_NoexceptExpression) {
871   if (!GetParam().isCXX11OrLater()) {
872     return;
873   }
874   StatementMatcher NoExcept = cxxNoexceptExpr();
875   EXPECT_TRUE(matches("void foo(); bool bar = noexcept(foo());", NoExcept));
876   EXPECT_TRUE(
877       matches("void foo() noexcept; bool bar = noexcept(foo());", NoExcept));
878   EXPECT_TRUE(notMatches("void foo() noexcept;", NoExcept));
879   EXPECT_TRUE(notMatches("void foo() noexcept(0+1);", NoExcept));
880   EXPECT_TRUE(matches("void foo() noexcept(noexcept(1+1));", NoExcept));
881 }
882 
TEST_P(ASTMatchersTest,Matcher_DefaultArgument)883 TEST_P(ASTMatchersTest, Matcher_DefaultArgument) {
884   if (!GetParam().isCXX()) {
885     return;
886   }
887   StatementMatcher Arg = cxxDefaultArgExpr();
888   EXPECT_TRUE(matches("void x(int, int = 0) { int y; x(y); }", Arg));
889   EXPECT_TRUE(
890       matches("class X { void x(int, int = 0) { int y; x(y); } };", Arg));
891   EXPECT_TRUE(notMatches("void x(int, int = 0) { int y; x(y, 0); }", Arg));
892 }
893 
TEST_P(ASTMatchersTest,StringLiteral)894 TEST_P(ASTMatchersTest, StringLiteral) {
895   StatementMatcher Literal = stringLiteral();
896   EXPECT_TRUE(matches("const char *s = \"string\";", Literal));
897   // with escaped characters
898   EXPECT_TRUE(matches("const char *s = \"\x05five\";", Literal));
899   // no matching -- though the data type is the same, there is no string literal
900   EXPECT_TRUE(notMatches("const char s[1] = {'a'};", Literal));
901 }
902 
TEST_P(ASTMatchersTest,StringLiteral_CXX)903 TEST_P(ASTMatchersTest, StringLiteral_CXX) {
904   if (!GetParam().isCXX()) {
905     return;
906   }
907   EXPECT_TRUE(matches("const wchar_t *s = L\"string\";", stringLiteral()));
908 }
909 
TEST_P(ASTMatchersTest,CharacterLiteral)910 TEST_P(ASTMatchersTest, CharacterLiteral) {
911   EXPECT_TRUE(matches("const char c = 'c';", characterLiteral()));
912   EXPECT_TRUE(notMatches("const char c = 0x1;", characterLiteral()));
913 }
914 
TEST_P(ASTMatchersTest,CharacterLiteral_CXX)915 TEST_P(ASTMatchersTest, CharacterLiteral_CXX) {
916   if (!GetParam().isCXX()) {
917     return;
918   }
919   // wide character
920   EXPECT_TRUE(matches("const char c = L'c';", characterLiteral()));
921   // wide character, Hex encoded, NOT MATCHED!
922   EXPECT_TRUE(notMatches("const wchar_t c = 0x2126;", characterLiteral()));
923 }
924 
TEST_P(ASTMatchersTest,IntegerLiteral)925 TEST_P(ASTMatchersTest, IntegerLiteral) {
926   StatementMatcher HasIntLiteral = integerLiteral();
927   EXPECT_TRUE(matches("int i = 10;", HasIntLiteral));
928   EXPECT_TRUE(matches("int i = 0x1AB;", HasIntLiteral));
929   EXPECT_TRUE(matches("int i = 10L;", HasIntLiteral));
930   EXPECT_TRUE(matches("int i = 10U;", HasIntLiteral));
931 
932   // Non-matching cases (character literals, float and double)
933   EXPECT_TRUE(notMatches("int i = L'a';",
934                          HasIntLiteral)); // this is actually a character
935   // literal cast to int
936   EXPECT_TRUE(notMatches("int i = 'a';", HasIntLiteral));
937   EXPECT_TRUE(notMatches("int i = 1e10;", HasIntLiteral));
938   EXPECT_TRUE(notMatches("int i = 10.0;", HasIntLiteral));
939 
940   // Negative integers.
941   EXPECT_TRUE(
942       matches("int i = -10;",
943               unaryOperator(hasOperatorName("-"),
944                             hasUnaryOperand(integerLiteral(equals(10))))));
945 }
946 
TEST_P(ASTMatchersTest,FloatLiteral)947 TEST_P(ASTMatchersTest, FloatLiteral) {
948   StatementMatcher HasFloatLiteral = floatLiteral();
949   EXPECT_TRUE(matches("float i = 10.0;", HasFloatLiteral));
950   EXPECT_TRUE(matches("float i = 10.0f;", HasFloatLiteral));
951   EXPECT_TRUE(matches("double i = 10.0;", HasFloatLiteral));
952   EXPECT_TRUE(matches("double i = 10.0L;", HasFloatLiteral));
953   EXPECT_TRUE(matches("double i = 1e10;", HasFloatLiteral));
954   EXPECT_TRUE(matches("double i = 5.0;", floatLiteral(equals(5.0))));
955   EXPECT_TRUE(matches("double i = 5.0;", floatLiteral(equals(5.0f))));
956   EXPECT_TRUE(
957       matches("double i = 5.0;", floatLiteral(equals(llvm::APFloat(5.0)))));
958 
959   EXPECT_TRUE(notMatches("float i = 10;", HasFloatLiteral));
960   EXPECT_TRUE(notMatches("double i = 5.0;", floatLiteral(equals(6.0))));
961   EXPECT_TRUE(notMatches("double i = 5.0;", floatLiteral(equals(6.0f))));
962   EXPECT_TRUE(
963       notMatches("double i = 5.0;", floatLiteral(equals(llvm::APFloat(6.0)))));
964 }
965 
TEST_P(ASTMatchersTest,CXXNullPtrLiteralExpr)966 TEST_P(ASTMatchersTest, CXXNullPtrLiteralExpr) {
967   if (!GetParam().isCXX11OrLater()) {
968     return;
969   }
970   EXPECT_TRUE(matches("int* i = nullptr;", cxxNullPtrLiteralExpr()));
971 }
972 
TEST_P(ASTMatchersTest,ChooseExpr)973 TEST_P(ASTMatchersTest, ChooseExpr) {
974   EXPECT_TRUE(matches("void f() { (void)__builtin_choose_expr(1, 2, 3); }",
975                       chooseExpr()));
976 }
977 
TEST_P(ASTMatchersTest,GNUNullExpr)978 TEST_P(ASTMatchersTest, GNUNullExpr) {
979   if (!GetParam().isCXX()) {
980     return;
981   }
982   EXPECT_TRUE(matches("int* i = __null;", gnuNullExpr()));
983 }
984 
TEST_P(ASTMatchersTest,GenericSelectionExpr)985 TEST_P(ASTMatchersTest, GenericSelectionExpr) {
986   EXPECT_TRUE(matches("void f() { (void)_Generic(1, int: 1, float: 2.0); }",
987                       genericSelectionExpr()));
988 }
989 
TEST_P(ASTMatchersTest,AtomicExpr)990 TEST_P(ASTMatchersTest, AtomicExpr) {
991   EXPECT_TRUE(matches("void foo() { int *ptr; __atomic_load_n(ptr, 1); }",
992                       atomicExpr()));
993 }
994 
TEST_P(ASTMatchersTest,Initializers_C99)995 TEST_P(ASTMatchersTest, Initializers_C99) {
996   if (!GetParam().isC99OrLater()) {
997     return;
998   }
999   EXPECT_TRUE(matches(
1000       "void foo() { struct point { double x; double y; };"
1001       "  struct point ptarray[10] = "
1002       "      { [2].y = 1.0, [2].x = 2.0, [0].x = 1.0 }; }",
1003       initListExpr(hasSyntacticForm(initListExpr(
1004           has(designatedInitExpr(designatorCountIs(2),
1005                                  hasDescendant(floatLiteral(equals(1.0))),
1006                                  hasDescendant(integerLiteral(equals(2))))),
1007           has(designatedInitExpr(designatorCountIs(2),
1008                                  hasDescendant(floatLiteral(equals(2.0))),
1009                                  hasDescendant(integerLiteral(equals(2))))),
1010           has(designatedInitExpr(
1011               designatorCountIs(2), hasDescendant(floatLiteral(equals(1.0))),
1012               hasDescendant(integerLiteral(equals(0))))))))));
1013 }
1014 
TEST_P(ASTMatchersTest,Initializers_CXX)1015 TEST_P(ASTMatchersTest, Initializers_CXX) {
1016   if (GetParam().Language != Lang_CXX03) {
1017     // FIXME: Make this test pass with other C++ standard versions.
1018     return;
1019   }
1020   EXPECT_TRUE(matches(
1021       "void foo() { struct point { double x; double y; };"
1022       "  struct point ptarray[10] = "
1023       "      { [2].y = 1.0, [2].x = 2.0, [0].x = 1.0 }; }",
1024       initListExpr(
1025           has(cxxConstructExpr(requiresZeroInitialization())),
1026           has(initListExpr(
1027               hasType(asString("struct point")), has(floatLiteral(equals(1.0))),
1028               has(implicitValueInitExpr(hasType(asString("double")))))),
1029           has(initListExpr(hasType(asString("struct point")),
1030                            has(floatLiteral(equals(2.0))),
1031                            has(floatLiteral(equals(1.0))))))));
1032 }
1033 
TEST_P(ASTMatchersTest,ParenListExpr)1034 TEST_P(ASTMatchersTest, ParenListExpr) {
1035   if (!GetParam().isCXX()) {
1036     return;
1037   }
1038   EXPECT_TRUE(
1039       matches("template<typename T> class foo { void bar() { foo X(*this); } };"
1040               "template class foo<int>;",
1041               varDecl(hasInitializer(parenListExpr(has(unaryOperator()))))));
1042 }
1043 
TEST_P(ASTMatchersTest,StmtExpr)1044 TEST_P(ASTMatchersTest, StmtExpr) {
1045   EXPECT_TRUE(matches("void declToImport() { int C = ({int X=4; X;}); }",
1046                       varDecl(hasInitializer(stmtExpr()))));
1047 }
1048 
TEST_P(ASTMatchersTest,PredefinedExpr)1049 TEST_P(ASTMatchersTest, PredefinedExpr) {
1050   // __func__ expands as StringLiteral("foo")
1051   EXPECT_TRUE(matches("void foo() { __func__; }",
1052                       predefinedExpr(hasType(asString("const char [4]")),
1053                                      has(stringLiteral()))));
1054 }
1055 
TEST_P(ASTMatchersTest,AsmStatement)1056 TEST_P(ASTMatchersTest, AsmStatement) {
1057   EXPECT_TRUE(matches("void foo() { __asm(\"mov al, 2\"); }", asmStmt()));
1058 }
1059 
TEST_P(ASTMatchersTest,HasCondition)1060 TEST_P(ASTMatchersTest, HasCondition) {
1061   if (!GetParam().isCXX()) {
1062     // FIXME: Add a test for `hasCondition()` that does not depend on C++.
1063     return;
1064   }
1065 
1066   StatementMatcher Condition =
1067       ifStmt(hasCondition(cxxBoolLiteral(equals(true))));
1068 
1069   EXPECT_TRUE(matches("void x() { if (true) {} }", Condition));
1070   EXPECT_TRUE(notMatches("void x() { if (false) {} }", Condition));
1071   EXPECT_TRUE(notMatches("void x() { bool a = true; if (a) {} }", Condition));
1072   EXPECT_TRUE(notMatches("void x() { if (true || false) {} }", Condition));
1073   EXPECT_TRUE(notMatches("void x() { if (1) {} }", Condition));
1074 }
1075 
TEST_P(ASTMatchersTest,ConditionalOperator)1076 TEST_P(ASTMatchersTest, ConditionalOperator) {
1077   if (!GetParam().isCXX()) {
1078     // FIXME: Add a test for `conditionalOperator()` that does not depend on
1079     // C++.
1080     return;
1081   }
1082 
1083   StatementMatcher Conditional =
1084       conditionalOperator(hasCondition(cxxBoolLiteral(equals(true))),
1085                           hasTrueExpression(cxxBoolLiteral(equals(false))));
1086 
1087   EXPECT_TRUE(matches("void x() { true ? false : true; }", Conditional));
1088   EXPECT_TRUE(notMatches("void x() { false ? false : true; }", Conditional));
1089   EXPECT_TRUE(notMatches("void x() { true ? true : false; }", Conditional));
1090 
1091   StatementMatcher ConditionalFalse =
1092       conditionalOperator(hasFalseExpression(cxxBoolLiteral(equals(false))));
1093 
1094   EXPECT_TRUE(matches("void x() { true ? true : false; }", ConditionalFalse));
1095   EXPECT_TRUE(
1096       notMatches("void x() { true ? false : true; }", ConditionalFalse));
1097 
1098   EXPECT_TRUE(matches("void x() { true ? true : false; }", ConditionalFalse));
1099   EXPECT_TRUE(
1100       notMatches("void x() { true ? false : true; }", ConditionalFalse));
1101 }
1102 
TEST_P(ASTMatchersTest,BinaryConditionalOperator)1103 TEST_P(ASTMatchersTest, BinaryConditionalOperator) {
1104   if (!GetParam().isCXX()) {
1105     // FIXME: This test should work in non-C++ language modes.
1106     return;
1107   }
1108 
1109   StatementMatcher AlwaysOne = traverse(
1110       TK_AsIs, binaryConditionalOperator(
1111                    hasCondition(implicitCastExpr(has(opaqueValueExpr(
1112                        hasSourceExpression((integerLiteral(equals(1)))))))),
1113                    hasFalseExpression(integerLiteral(equals(0)))));
1114 
1115   EXPECT_TRUE(matches("void x() { 1 ?: 0; }", AlwaysOne));
1116 
1117   StatementMatcher FourNotFive = binaryConditionalOperator(
1118       hasTrueExpression(
1119           opaqueValueExpr(hasSourceExpression((integerLiteral(equals(4)))))),
1120       hasFalseExpression(integerLiteral(equals(5))));
1121 
1122   EXPECT_TRUE(matches("void x() { 4 ?: 5; }", FourNotFive));
1123 }
1124 
TEST_P(ASTMatchersTest,ArraySubscriptExpr)1125 TEST_P(ASTMatchersTest, ArraySubscriptExpr) {
1126   EXPECT_TRUE(
1127       matches("int i[2]; void f() { i[1] = 1; }", arraySubscriptExpr()));
1128   EXPECT_TRUE(notMatches("int i; void f() { i = 1; }", arraySubscriptExpr()));
1129 }
1130 
TEST_P(ASTMatchersTest,ForStmt)1131 TEST_P(ASTMatchersTest, ForStmt) {
1132   EXPECT_TRUE(matches("void f() { for(;;); }", forStmt()));
1133   EXPECT_TRUE(matches("void f() { if(1) for(;;); }", forStmt()));
1134 }
1135 
TEST_P(ASTMatchersTest,ForStmt_CXX11)1136 TEST_P(ASTMatchersTest, ForStmt_CXX11) {
1137   if (!GetParam().isCXX11OrLater()) {
1138     return;
1139   }
1140   EXPECT_TRUE(notMatches("int as[] = { 1, 2, 3 };"
1141                          "void f() { for (auto &a : as); }",
1142                          forStmt()));
1143 }
1144 
TEST_P(ASTMatchersTest,ForStmt_NoFalsePositives)1145 TEST_P(ASTMatchersTest, ForStmt_NoFalsePositives) {
1146   EXPECT_TRUE(notMatches("void f() { ; }", forStmt()));
1147   EXPECT_TRUE(notMatches("void f() { if(1); }", forStmt()));
1148 }
1149 
TEST_P(ASTMatchersTest,CompoundStatement)1150 TEST_P(ASTMatchersTest, CompoundStatement) {
1151   EXPECT_TRUE(notMatches("void f();", compoundStmt()));
1152   EXPECT_TRUE(matches("void f() {}", compoundStmt()));
1153   EXPECT_TRUE(matches("void f() {{}}", compoundStmt()));
1154 }
1155 
TEST_P(ASTMatchersTest,CompoundStatement_DoesNotMatchEmptyStruct)1156 TEST_P(ASTMatchersTest, CompoundStatement_DoesNotMatchEmptyStruct) {
1157   if (!GetParam().isCXX()) {
1158     // FIXME: Add a similar test that does not depend on C++.
1159     return;
1160   }
1161   // It's not a compound statement just because there's "{}" in the source
1162   // text. This is an AST search, not grep.
1163   EXPECT_TRUE(notMatches("namespace n { struct S {}; }", compoundStmt()));
1164   EXPECT_TRUE(
1165       matches("namespace n { struct S { void f() {{}} }; }", compoundStmt()));
1166 }
1167 
TEST_P(ASTMatchersTest,CastExpr_MatchesExplicitCasts)1168 TEST_P(ASTMatchersTest, CastExpr_MatchesExplicitCasts) {
1169   EXPECT_TRUE(matches("void *p = (void *)(&p);", castExpr()));
1170 }
1171 
TEST_P(ASTMatchersTest,CastExpr_MatchesExplicitCasts_CXX)1172 TEST_P(ASTMatchersTest, CastExpr_MatchesExplicitCasts_CXX) {
1173   if (!GetParam().isCXX()) {
1174     return;
1175   }
1176   EXPECT_TRUE(matches("char *p = reinterpret_cast<char *>(&p);", castExpr()));
1177   EXPECT_TRUE(matches("char q, *p = const_cast<char *>(&q);", castExpr()));
1178   EXPECT_TRUE(matches("char c = char(0);", castExpr()));
1179 }
1180 
TEST_P(ASTMatchersTest,CastExpression_MatchesImplicitCasts)1181 TEST_P(ASTMatchersTest, CastExpression_MatchesImplicitCasts) {
1182   // This test creates an implicit cast from int to char.
1183   EXPECT_TRUE(matches("char c = 0;", traverse(TK_AsIs, castExpr())));
1184   // This test creates an implicit cast from lvalue to rvalue.
1185   EXPECT_TRUE(matches("void f() { char c = 0, d = c; }",
1186                       traverse(TK_AsIs, castExpr())));
1187 }
1188 
TEST_P(ASTMatchersTest,CastExpr_DoesNotMatchNonCasts)1189 TEST_P(ASTMatchersTest, CastExpr_DoesNotMatchNonCasts) {
1190   if (GetParam().Language == Lang_C89 || GetParam().Language == Lang_C99) {
1191     // This does have a cast in C
1192     EXPECT_TRUE(matches("char c = '0';", implicitCastExpr()));
1193   } else {
1194     EXPECT_TRUE(notMatches("char c = '0';", castExpr()));
1195   }
1196   EXPECT_TRUE(notMatches("int i = (0);", castExpr()));
1197   EXPECT_TRUE(notMatches("int i = 0;", castExpr()));
1198 }
1199 
TEST_P(ASTMatchersTest,CastExpr_DoesNotMatchNonCasts_CXX)1200 TEST_P(ASTMatchersTest, CastExpr_DoesNotMatchNonCasts_CXX) {
1201   if (!GetParam().isCXX()) {
1202     return;
1203   }
1204   EXPECT_TRUE(notMatches("char c, &q = c;", castExpr()));
1205 }
1206 
TEST_P(ASTMatchersTest,CXXReinterpretCastExpr)1207 TEST_P(ASTMatchersTest, CXXReinterpretCastExpr) {
1208   if (!GetParam().isCXX()) {
1209     return;
1210   }
1211   EXPECT_TRUE(matches("char* p = reinterpret_cast<char*>(&p);",
1212                       cxxReinterpretCastExpr()));
1213 }
1214 
TEST_P(ASTMatchersTest,CXXReinterpretCastExpr_DoesNotMatchOtherCasts)1215 TEST_P(ASTMatchersTest, CXXReinterpretCastExpr_DoesNotMatchOtherCasts) {
1216   if (!GetParam().isCXX()) {
1217     return;
1218   }
1219   EXPECT_TRUE(notMatches("char* p = (char*)(&p);", cxxReinterpretCastExpr()));
1220   EXPECT_TRUE(notMatches("char q, *p = const_cast<char*>(&q);",
1221                          cxxReinterpretCastExpr()));
1222   EXPECT_TRUE(notMatches("void* p = static_cast<void*>(&p);",
1223                          cxxReinterpretCastExpr()));
1224   EXPECT_TRUE(notMatches("struct B { virtual ~B() {} }; struct D : B {};"
1225                          "B b;"
1226                          "D* p = dynamic_cast<D*>(&b);",
1227                          cxxReinterpretCastExpr()));
1228 }
1229 
TEST_P(ASTMatchersTest,CXXFunctionalCastExpr_MatchesSimpleCase)1230 TEST_P(ASTMatchersTest, CXXFunctionalCastExpr_MatchesSimpleCase) {
1231   if (!GetParam().isCXX()) {
1232     return;
1233   }
1234   StringRef foo_class = "class Foo { public: Foo(const char*); };";
1235   EXPECT_TRUE(matches(foo_class + "void r() { Foo f = Foo(\"hello world\"); }",
1236                       cxxFunctionalCastExpr()));
1237 }
1238 
TEST_P(ASTMatchersTest,CXXFunctionalCastExpr_DoesNotMatchOtherCasts)1239 TEST_P(ASTMatchersTest, CXXFunctionalCastExpr_DoesNotMatchOtherCasts) {
1240   if (!GetParam().isCXX()) {
1241     return;
1242   }
1243   StringRef FooClass = "class Foo { public: Foo(const char*); };";
1244   EXPECT_TRUE(
1245       notMatches(FooClass + "void r() { Foo f = (Foo) \"hello world\"; }",
1246                  cxxFunctionalCastExpr()));
1247   EXPECT_TRUE(notMatches(FooClass + "void r() { Foo f = \"hello world\"; }",
1248                          cxxFunctionalCastExpr()));
1249 }
1250 
TEST_P(ASTMatchersTest,CXXDynamicCastExpr)1251 TEST_P(ASTMatchersTest, CXXDynamicCastExpr) {
1252   if (!GetParam().isCXX()) {
1253     return;
1254   }
1255   EXPECT_TRUE(matches("struct B { virtual ~B() {} }; struct D : B {};"
1256                       "B b;"
1257                       "D* p = dynamic_cast<D*>(&b);",
1258                       cxxDynamicCastExpr()));
1259 }
1260 
TEST_P(ASTMatchersTest,CXXStaticCastExpr_MatchesSimpleCase)1261 TEST_P(ASTMatchersTest, CXXStaticCastExpr_MatchesSimpleCase) {
1262   if (!GetParam().isCXX()) {
1263     return;
1264   }
1265   EXPECT_TRUE(matches("void* p(static_cast<void*>(&p));", cxxStaticCastExpr()));
1266 }
1267 
TEST_P(ASTMatchersTest,CXXStaticCastExpr_DoesNotMatchOtherCasts)1268 TEST_P(ASTMatchersTest, CXXStaticCastExpr_DoesNotMatchOtherCasts) {
1269   if (!GetParam().isCXX()) {
1270     return;
1271   }
1272   EXPECT_TRUE(notMatches("char* p = (char*)(&p);", cxxStaticCastExpr()));
1273   EXPECT_TRUE(
1274       notMatches("char q, *p = const_cast<char*>(&q);", cxxStaticCastExpr()));
1275   EXPECT_TRUE(notMatches("void* p = reinterpret_cast<char*>(&p);",
1276                          cxxStaticCastExpr()));
1277   EXPECT_TRUE(notMatches("struct B { virtual ~B() {} }; struct D : B {};"
1278                          "B b;"
1279                          "D* p = dynamic_cast<D*>(&b);",
1280                          cxxStaticCastExpr()));
1281 }
1282 
TEST_P(ASTMatchersTest,CStyleCastExpr_MatchesSimpleCase)1283 TEST_P(ASTMatchersTest, CStyleCastExpr_MatchesSimpleCase) {
1284   EXPECT_TRUE(matches("int i = (int) 2.2f;", cStyleCastExpr()));
1285 }
1286 
TEST_P(ASTMatchersTest,CStyleCastExpr_DoesNotMatchOtherCasts)1287 TEST_P(ASTMatchersTest, CStyleCastExpr_DoesNotMatchOtherCasts) {
1288   if (!GetParam().isCXX()) {
1289     return;
1290   }
1291   EXPECT_TRUE(notMatches("char* p = static_cast<char*>(0);"
1292                          "char q, *r = const_cast<char*>(&q);"
1293                          "void* s = reinterpret_cast<char*>(&s);"
1294                          "struct B { virtual ~B() {} }; struct D : B {};"
1295                          "B b;"
1296                          "D* t = dynamic_cast<D*>(&b);",
1297                          cStyleCastExpr()));
1298 }
1299 
TEST_P(ASTMatchersTest,ImplicitCastExpr_MatchesSimpleCase)1300 TEST_P(ASTMatchersTest, ImplicitCastExpr_MatchesSimpleCase) {
1301   // This test creates an implicit const cast.
1302   EXPECT_TRUE(
1303       matches("void f() { int x = 0; const int y = x; }",
1304               traverse(TK_AsIs, varDecl(hasInitializer(implicitCastExpr())))));
1305   // This test creates an implicit cast from int to char.
1306   EXPECT_TRUE(
1307       matches("char c = 0;",
1308               traverse(TK_AsIs, varDecl(hasInitializer(implicitCastExpr())))));
1309   // This test creates an implicit array-to-pointer cast.
1310   EXPECT_TRUE(
1311       matches("int arr[6]; int *p = arr;",
1312               traverse(TK_AsIs, varDecl(hasInitializer(implicitCastExpr())))));
1313 }
1314 
TEST_P(ASTMatchersTest,ImplicitCastExpr_DoesNotMatchIncorrectly)1315 TEST_P(ASTMatchersTest, ImplicitCastExpr_DoesNotMatchIncorrectly) {
1316   // This test verifies that implicitCastExpr() matches exactly when implicit
1317   // casts are present, and that it ignores explicit and paren casts.
1318 
1319   // These two test cases have no casts.
1320   EXPECT_TRUE(
1321       notMatches("int x = 0;", varDecl(hasInitializer(implicitCastExpr()))));
1322   EXPECT_TRUE(
1323       notMatches("int x = (0);", varDecl(hasInitializer(implicitCastExpr()))));
1324   EXPECT_TRUE(notMatches("void f() { int x = 0; double d = (double) x; }",
1325                          varDecl(hasInitializer(implicitCastExpr()))));
1326 }
1327 
TEST_P(ASTMatchersTest,ImplicitCastExpr_DoesNotMatchIncorrectly_CXX)1328 TEST_P(ASTMatchersTest, ImplicitCastExpr_DoesNotMatchIncorrectly_CXX) {
1329   if (!GetParam().isCXX()) {
1330     return;
1331   }
1332   EXPECT_TRUE(notMatches("int x = 0, &y = x;",
1333                          varDecl(hasInitializer(implicitCastExpr()))));
1334   EXPECT_TRUE(notMatches("const int *p; int *q = const_cast<int *>(p);",
1335                          varDecl(hasInitializer(implicitCastExpr()))));
1336 }
1337 
TEST_P(ASTMatchersTest,Stmt_DoesNotMatchDeclarations)1338 TEST_P(ASTMatchersTest, Stmt_DoesNotMatchDeclarations) {
1339   EXPECT_TRUE(notMatches("struct X {};", stmt()));
1340 }
1341 
TEST_P(ASTMatchersTest,Stmt_MatchesCompoundStatments)1342 TEST_P(ASTMatchersTest, Stmt_MatchesCompoundStatments) {
1343   EXPECT_TRUE(matches("void x() {}", stmt()));
1344 }
1345 
TEST_P(ASTMatchersTest,DeclStmt_DoesNotMatchCompoundStatements)1346 TEST_P(ASTMatchersTest, DeclStmt_DoesNotMatchCompoundStatements) {
1347   EXPECT_TRUE(notMatches("void x() {}", declStmt()));
1348 }
1349 
TEST_P(ASTMatchersTest,DeclStmt_MatchesVariableDeclarationStatements)1350 TEST_P(ASTMatchersTest, DeclStmt_MatchesVariableDeclarationStatements) {
1351   EXPECT_TRUE(matches("void x() { int a; }", declStmt()));
1352 }
1353 
TEST_P(ASTMatchersTest,ExprWithCleanups_MatchesExprWithCleanups)1354 TEST_P(ASTMatchersTest, ExprWithCleanups_MatchesExprWithCleanups) {
1355   if (!GetParam().isCXX()) {
1356     return;
1357   }
1358   EXPECT_TRUE(
1359       matches("struct Foo { ~Foo(); };"
1360               "const Foo f = Foo();",
1361               traverse(TK_AsIs, varDecl(hasInitializer(exprWithCleanups())))));
1362   EXPECT_FALSE(
1363       matches("struct Foo { }; Foo a;"
1364               "const Foo f = a;",
1365               traverse(TK_AsIs, varDecl(hasInitializer(exprWithCleanups())))));
1366 }
1367 
TEST_P(ASTMatchersTest,InitListExpr)1368 TEST_P(ASTMatchersTest, InitListExpr) {
1369   EXPECT_TRUE(matches("int a[] = { 1, 2 };",
1370                       initListExpr(hasType(asString("int [2]")))));
1371   EXPECT_TRUE(matches("struct B { int x, y; }; struct B b = { 5, 6 };",
1372                       initListExpr(hasType(recordDecl(hasName("B"))))));
1373   EXPECT_TRUE(
1374       matches("int i[1] = {42, [0] = 43};", integerLiteral(equals(42))));
1375 }
1376 
TEST_P(ASTMatchersTest,InitListExpr_CXX)1377 TEST_P(ASTMatchersTest, InitListExpr_CXX) {
1378   if (!GetParam().isCXX()) {
1379     return;
1380   }
1381   EXPECT_TRUE(matches("struct S { S(void (*a)()); };"
1382                       "void f();"
1383                       "S s[1] = { &f };",
1384                       declRefExpr(to(functionDecl(hasName("f"))))));
1385 }
1386 
TEST_P(ASTMatchersTest,CXXStdInitializerListExpression_MatchesCXXStdInitializerListExpression)1387 TEST_P(ASTMatchersTest,
1388        CXXStdInitializerListExpression_MatchesCXXStdInitializerListExpression) {
1389   if (!GetParam().isCXX11OrLater()) {
1390     return;
1391   }
1392   StringRef code = "namespace std {"
1393                    "template <typename> class initializer_list {"
1394                    "  public: initializer_list() noexcept {}"
1395                    "};"
1396                    "}"
1397                    "struct A {"
1398                    "  A(std::initializer_list<int>) {}"
1399                    "};";
1400   EXPECT_TRUE(matches(
1401       code + "A a{0};",
1402       traverse(TK_AsIs, cxxConstructExpr(has(cxxStdInitializerListExpr()),
1403                                          hasDeclaration(cxxConstructorDecl(
1404                                              ofClass(hasName("A"))))))));
1405   EXPECT_TRUE(matches(
1406       code + "A a = {0};",
1407       traverse(TK_AsIs, cxxConstructExpr(has(cxxStdInitializerListExpr()),
1408                                          hasDeclaration(cxxConstructorDecl(
1409                                              ofClass(hasName("A"))))))));
1410 
1411   EXPECT_TRUE(notMatches("int a[] = { 1, 2 };", cxxStdInitializerListExpr()));
1412   EXPECT_TRUE(notMatches("struct B { int x, y; }; B b = { 5, 6 };",
1413                          cxxStdInitializerListExpr()));
1414 }
1415 
TEST_P(ASTMatchersTest,UsingDecl_MatchesUsingDeclarations)1416 TEST_P(ASTMatchersTest, UsingDecl_MatchesUsingDeclarations) {
1417   if (!GetParam().isCXX()) {
1418     return;
1419   }
1420   EXPECT_TRUE(matches("namespace X { int x; } using X::x;", usingDecl()));
1421 }
1422 
TEST_P(ASTMatchersTest,UsingDecl_MatchesShadowUsingDelcarations)1423 TEST_P(ASTMatchersTest, UsingDecl_MatchesShadowUsingDelcarations) {
1424   if (!GetParam().isCXX()) {
1425     return;
1426   }
1427   EXPECT_TRUE(matches("namespace f { int a; } using f::a;",
1428                       usingDecl(hasAnyUsingShadowDecl(hasName("a")))));
1429 }
1430 
TEST_P(ASTMatchersTest,UsingEnumDecl_MatchesUsingEnumDeclarations)1431 TEST_P(ASTMatchersTest, UsingEnumDecl_MatchesUsingEnumDeclarations) {
1432   if (!GetParam().isCXX20OrLater()) {
1433     return;
1434   }
1435   EXPECT_TRUE(
1436       matches("namespace X { enum x {}; } using enum X::x;", usingEnumDecl()));
1437 }
1438 
TEST_P(ASTMatchersTest,UsingEnumDecl_MatchesShadowUsingDeclarations)1439 TEST_P(ASTMatchersTest, UsingEnumDecl_MatchesShadowUsingDeclarations) {
1440   if (!GetParam().isCXX20OrLater()) {
1441     return;
1442   }
1443   EXPECT_TRUE(matches("namespace f { enum a {b}; } using enum f::a;",
1444                       usingEnumDecl(hasAnyUsingShadowDecl(hasName("b")))));
1445 }
1446 
TEST_P(ASTMatchersTest,UsingDirectiveDecl_MatchesUsingNamespace)1447 TEST_P(ASTMatchersTest, UsingDirectiveDecl_MatchesUsingNamespace) {
1448   if (!GetParam().isCXX()) {
1449     return;
1450   }
1451   EXPECT_TRUE(matches("namespace X { int x; } using namespace X;",
1452                       usingDirectiveDecl()));
1453   EXPECT_FALSE(
1454       matches("namespace X { int x; } using X::x;", usingDirectiveDecl()));
1455 }
1456 
TEST_P(ASTMatchersTest,WhileStmt)1457 TEST_P(ASTMatchersTest, WhileStmt) {
1458   EXPECT_TRUE(notMatches("void x() {}", whileStmt()));
1459   EXPECT_TRUE(matches("void x() { while(1); }", whileStmt()));
1460   EXPECT_TRUE(notMatches("void x() { do {} while(1); }", whileStmt()));
1461 }
1462 
TEST_P(ASTMatchersTest,DoStmt_MatchesDoLoops)1463 TEST_P(ASTMatchersTest, DoStmt_MatchesDoLoops) {
1464   EXPECT_TRUE(matches("void x() { do {} while(1); }", doStmt()));
1465   EXPECT_TRUE(matches("void x() { do ; while(0); }", doStmt()));
1466 }
1467 
TEST_P(ASTMatchersTest,DoStmt_DoesNotMatchWhileLoops)1468 TEST_P(ASTMatchersTest, DoStmt_DoesNotMatchWhileLoops) {
1469   EXPECT_TRUE(notMatches("void x() { while(1) {} }", doStmt()));
1470 }
1471 
TEST_P(ASTMatchersTest,SwitchCase_MatchesCase)1472 TEST_P(ASTMatchersTest, SwitchCase_MatchesCase) {
1473   EXPECT_TRUE(matches("void x() { switch(42) { case 42:; } }", switchCase()));
1474   EXPECT_TRUE(matches("void x() { switch(42) { default:; } }", switchCase()));
1475   EXPECT_TRUE(matches("void x() { switch(42) default:; }", switchCase()));
1476   EXPECT_TRUE(notMatches("void x() { switch(42) {} }", switchCase()));
1477 }
1478 
TEST_P(ASTMatchersTest,SwitchCase_MatchesSwitch)1479 TEST_P(ASTMatchersTest, SwitchCase_MatchesSwitch) {
1480   EXPECT_TRUE(matches("void x() { switch(42) { case 42:; } }", switchStmt()));
1481   EXPECT_TRUE(matches("void x() { switch(42) { default:; } }", switchStmt()));
1482   EXPECT_TRUE(matches("void x() { switch(42) default:; }", switchStmt()));
1483   EXPECT_TRUE(notMatches("void x() {}", switchStmt()));
1484 }
1485 
TEST_P(ASTMatchersTest,CxxExceptionHandling_SimpleCases)1486 TEST_P(ASTMatchersTest, CxxExceptionHandling_SimpleCases) {
1487   if (!GetParam().isCXX()) {
1488     return;
1489   }
1490   EXPECT_TRUE(matches("void foo() try { } catch(int X) { }", cxxCatchStmt()));
1491   EXPECT_TRUE(matches("void foo() try { } catch(int X) { }", cxxTryStmt()));
1492   EXPECT_TRUE(
1493       notMatches("void foo() try { } catch(int X) { }", cxxThrowExpr()));
1494   EXPECT_TRUE(
1495       matches("void foo() try { throw; } catch(int X) { }", cxxThrowExpr()));
1496   EXPECT_TRUE(
1497       matches("void foo() try { throw 5;} catch(int X) { }", cxxThrowExpr()));
1498   EXPECT_TRUE(matches("void foo() try { throw; } catch(...) { }",
1499                       cxxCatchStmt(isCatchAll())));
1500   EXPECT_TRUE(notMatches("void foo() try { throw; } catch(int) { }",
1501                          cxxCatchStmt(isCatchAll())));
1502   EXPECT_TRUE(matches("void foo() try {} catch(int X) { }",
1503                       varDecl(isExceptionVariable())));
1504   EXPECT_TRUE(notMatches("void foo() try { int X; } catch (...) { }",
1505                          varDecl(isExceptionVariable())));
1506 }
1507 
TEST_P(ASTMatchersTest,ParenExpr_SimpleCases)1508 TEST_P(ASTMatchersTest, ParenExpr_SimpleCases) {
1509   EXPECT_TRUE(matches("int i = (3);", traverse(TK_AsIs, parenExpr())));
1510   EXPECT_TRUE(matches("int i = (3 + 7);", traverse(TK_AsIs, parenExpr())));
1511   EXPECT_TRUE(notMatches("int i = 3;", traverse(TK_AsIs, parenExpr())));
1512   EXPECT_TRUE(notMatches("int f() { return 1; }; void g() { int a = f(); }",
1513                          traverse(TK_AsIs, parenExpr())));
1514 }
1515 
TEST_P(ASTMatchersTest,IgnoringParens)1516 TEST_P(ASTMatchersTest, IgnoringParens) {
1517   EXPECT_FALSE(matches("const char* str = (\"my-string\");",
1518                        traverse(TK_AsIs, implicitCastExpr(hasSourceExpression(
1519                                              stringLiteral())))));
1520   EXPECT_TRUE(
1521       matches("const char* str = (\"my-string\");",
1522               traverse(TK_AsIs, implicitCastExpr(hasSourceExpression(
1523                                     ignoringParens(stringLiteral()))))));
1524 }
1525 
TEST_P(ASTMatchersTest,QualType)1526 TEST_P(ASTMatchersTest, QualType) {
1527   EXPECT_TRUE(matches("struct S {};", qualType().bind("loc")));
1528 }
1529 
TEST_P(ASTMatchersTest,ConstantArrayType)1530 TEST_P(ASTMatchersTest, ConstantArrayType) {
1531   EXPECT_TRUE(matches("int a[2];", constantArrayType()));
1532   EXPECT_TRUE(notMatches("void f() { int a[] = { 2, 3 }; int b[a[0]]; }",
1533                          constantArrayType(hasElementType(builtinType()))));
1534 
1535   EXPECT_TRUE(matches("int a[42];", constantArrayType(hasSize(42))));
1536   EXPECT_TRUE(matches("int b[2*21];", constantArrayType(hasSize(42))));
1537   EXPECT_TRUE(notMatches("int c[41], d[43];", constantArrayType(hasSize(42))));
1538 }
1539 
TEST_P(ASTMatchersTest,DependentSizedArrayType)1540 TEST_P(ASTMatchersTest, DependentSizedArrayType) {
1541   if (!GetParam().isCXX()) {
1542     return;
1543   }
1544   EXPECT_TRUE(
1545       matches("template <typename T, int Size> class array { T data[Size]; };",
1546               dependentSizedArrayType()));
1547   EXPECT_TRUE(
1548       notMatches("int a[42]; int b[] = { 2, 3 }; void f() { int c[b[0]]; }",
1549                  dependentSizedArrayType()));
1550 }
1551 
TEST_P(ASTMatchersTest,IncompleteArrayType)1552 TEST_P(ASTMatchersTest, IncompleteArrayType) {
1553   EXPECT_TRUE(matches("int a[] = { 2, 3 };", incompleteArrayType()));
1554   EXPECT_TRUE(matches("void f(int a[]) {}", incompleteArrayType()));
1555 
1556   EXPECT_TRUE(notMatches("int a[42]; void f() { int b[a[0]]; }",
1557                          incompleteArrayType()));
1558 }
1559 
TEST_P(ASTMatchersTest,VariableArrayType)1560 TEST_P(ASTMatchersTest, VariableArrayType) {
1561   EXPECT_TRUE(matches("void f(int b) { int a[b]; }", variableArrayType()));
1562   EXPECT_TRUE(notMatches("int a[] = {2, 3}; int b[42];", variableArrayType()));
1563 
1564   EXPECT_TRUE(matches("void f(int b) { int a[b]; }",
1565                       variableArrayType(hasSizeExpr(ignoringImpCasts(
1566                           declRefExpr(to(varDecl(hasName("b")))))))));
1567 }
1568 
TEST_P(ASTMatchersTest,AtomicType)1569 TEST_P(ASTMatchersTest, AtomicType) {
1570   if (llvm::Triple(llvm::sys::getDefaultTargetTriple()).getOS() !=
1571       llvm::Triple::Win32) {
1572     // FIXME: Make this work for MSVC.
1573     EXPECT_TRUE(matches("_Atomic(int) i;", atomicType()));
1574 
1575     EXPECT_TRUE(
1576         matches("_Atomic(int) i;", atomicType(hasValueType(isInteger()))));
1577     EXPECT_TRUE(
1578         notMatches("_Atomic(float) f;", atomicType(hasValueType(isInteger()))));
1579   }
1580 }
1581 
TEST_P(ASTMatchersTest,AutoType)1582 TEST_P(ASTMatchersTest, AutoType) {
1583   if (!GetParam().isCXX11OrLater()) {
1584     return;
1585   }
1586   EXPECT_TRUE(matches("auto i = 2;", autoType()));
1587   EXPECT_TRUE(matches("int v[] = { 2, 3 }; void f() { for (int i : v) {} }",
1588                       autoType()));
1589 
1590   EXPECT_TRUE(matches("auto i = 2;", varDecl(hasType(isInteger()))));
1591   EXPECT_TRUE(matches("struct X{}; auto x = X{};",
1592                       varDecl(hasType(recordDecl(hasName("X"))))));
1593 
1594   // FIXME: Matching against the type-as-written can't work here, because the
1595   //        type as written was not deduced.
1596   // EXPECT_TRUE(matches("auto a = 1;",
1597   //                    autoType(hasDeducedType(isInteger()))));
1598   // EXPECT_TRUE(notMatches("auto b = 2.0;",
1599   //                       autoType(hasDeducedType(isInteger()))));
1600 }
1601 
TEST_P(ASTMatchersTest,DecltypeType)1602 TEST_P(ASTMatchersTest, DecltypeType) {
1603   if (!GetParam().isCXX11OrLater()) {
1604     return;
1605   }
1606   EXPECT_TRUE(matches("decltype(1 + 1) sum = 1 + 1;", decltypeType()));
1607   EXPECT_TRUE(matches("decltype(1 + 1) sum = 1 + 1;",
1608                       decltypeType(hasUnderlyingType(isInteger()))));
1609 }
1610 
TEST_P(ASTMatchersTest,FunctionType)1611 TEST_P(ASTMatchersTest, FunctionType) {
1612   EXPECT_TRUE(matches("int (*f)(int);", functionType()));
1613   EXPECT_TRUE(matches("void f(int i) {}", functionType()));
1614 }
1615 
TEST_P(ASTMatchersTest,IgnoringParens_Type)1616 TEST_P(ASTMatchersTest, IgnoringParens_Type) {
1617   EXPECT_TRUE(
1618       notMatches("void (*fp)(void);", pointerType(pointee(functionType()))));
1619   EXPECT_TRUE(matches("void (*fp)(void);",
1620                       pointerType(pointee(ignoringParens(functionType())))));
1621 }
1622 
TEST_P(ASTMatchersTest,FunctionProtoType)1623 TEST_P(ASTMatchersTest, FunctionProtoType) {
1624   EXPECT_TRUE(matches("int (*f)(int);", functionProtoType()));
1625   EXPECT_TRUE(matches("void f(int i);", functionProtoType()));
1626   EXPECT_TRUE(matches("void f(void);", functionProtoType(parameterCountIs(0))));
1627 }
1628 
TEST_P(ASTMatchersTest,FunctionProtoType_C)1629 TEST_P(ASTMatchersTest, FunctionProtoType_C) {
1630   if (!GetParam().isC()) {
1631     return;
1632   }
1633   EXPECT_TRUE(notMatches("void f();", functionProtoType()));
1634 }
1635 
TEST_P(ASTMatchersTest,FunctionProtoType_CXX)1636 TEST_P(ASTMatchersTest, FunctionProtoType_CXX) {
1637   if (!GetParam().isCXX()) {
1638     return;
1639   }
1640   EXPECT_TRUE(matches("void f();", functionProtoType(parameterCountIs(0))));
1641 }
1642 
TEST_P(ASTMatchersTest,ParenType)1643 TEST_P(ASTMatchersTest, ParenType) {
1644   EXPECT_TRUE(
1645       matches("int (*array)[4];", varDecl(hasType(pointsTo(parenType())))));
1646   EXPECT_TRUE(notMatches("int *array[4];", varDecl(hasType(parenType()))));
1647 
1648   EXPECT_TRUE(matches(
1649       "int (*ptr_to_func)(int);",
1650       varDecl(hasType(pointsTo(parenType(innerType(functionType())))))));
1651   EXPECT_TRUE(notMatches(
1652       "int (*ptr_to_array)[4];",
1653       varDecl(hasType(pointsTo(parenType(innerType(functionType())))))));
1654 }
1655 
TEST_P(ASTMatchersTest,PointerType)1656 TEST_P(ASTMatchersTest, PointerType) {
1657   // FIXME: Reactive when these tests can be more specific (not matching
1658   // implicit code on certain platforms), likely when we have hasDescendant for
1659   // Types/TypeLocs.
1660   // EXPECT_TRUE(matchAndVerifyResultTrue(
1661   //    "int* a;",
1662   //    pointerTypeLoc(pointeeLoc(typeLoc().bind("loc"))),
1663   //    std::make_unique<VerifyIdIsBoundTo<TypeLoc>>("loc", 1)));
1664   // EXPECT_TRUE(matchAndVerifyResultTrue(
1665   //    "int* a;",
1666   //    pointerTypeLoc().bind("loc"),
1667   //    std::make_unique<VerifyIdIsBoundTo<TypeLoc>>("loc", 1)));
1668   EXPECT_TRUE(matches("int** a;", loc(pointerType(pointee(qualType())))));
1669   EXPECT_TRUE(matches("int** a;", loc(pointerType(pointee(pointerType())))));
1670   EXPECT_TRUE(matches("int* b; int* * const a = &b;",
1671                       loc(qualType(isConstQualified(), pointerType()))));
1672 
1673   StringRef Fragment = "int *ptr;";
1674   EXPECT_TRUE(notMatches(Fragment,
1675                          varDecl(hasName("ptr"), hasType(blockPointerType()))));
1676   EXPECT_TRUE(notMatches(
1677       Fragment, varDecl(hasName("ptr"), hasType(memberPointerType()))));
1678   EXPECT_TRUE(
1679       matches(Fragment, varDecl(hasName("ptr"), hasType(pointerType()))));
1680   EXPECT_TRUE(
1681       notMatches(Fragment, varDecl(hasName("ptr"), hasType(referenceType()))));
1682 }
1683 
TEST_P(ASTMatchersTest,PointerType_CXX)1684 TEST_P(ASTMatchersTest, PointerType_CXX) {
1685   if (!GetParam().isCXX()) {
1686     return;
1687   }
1688   StringRef Fragment = "struct A { int i; }; int A::* ptr = &A::i;";
1689   EXPECT_TRUE(notMatches(Fragment,
1690                          varDecl(hasName("ptr"), hasType(blockPointerType()))));
1691   EXPECT_TRUE(
1692       matches(Fragment, varDecl(hasName("ptr"), hasType(memberPointerType()))));
1693   EXPECT_TRUE(
1694       notMatches(Fragment, varDecl(hasName("ptr"), hasType(pointerType()))));
1695   EXPECT_TRUE(
1696       notMatches(Fragment, varDecl(hasName("ptr"), hasType(referenceType()))));
1697   EXPECT_TRUE(notMatches(
1698       Fragment, varDecl(hasName("ptr"), hasType(lValueReferenceType()))));
1699   EXPECT_TRUE(notMatches(
1700       Fragment, varDecl(hasName("ptr"), hasType(rValueReferenceType()))));
1701 
1702   Fragment = "int a; int &ref = a;";
1703   EXPECT_TRUE(notMatches(Fragment,
1704                          varDecl(hasName("ref"), hasType(blockPointerType()))));
1705   EXPECT_TRUE(notMatches(
1706       Fragment, varDecl(hasName("ref"), hasType(memberPointerType()))));
1707   EXPECT_TRUE(
1708       notMatches(Fragment, varDecl(hasName("ref"), hasType(pointerType()))));
1709   EXPECT_TRUE(
1710       matches(Fragment, varDecl(hasName("ref"), hasType(referenceType()))));
1711   EXPECT_TRUE(matches(Fragment,
1712                       varDecl(hasName("ref"), hasType(lValueReferenceType()))));
1713   EXPECT_TRUE(notMatches(
1714       Fragment, varDecl(hasName("ref"), hasType(rValueReferenceType()))));
1715 }
1716 
TEST_P(ASTMatchersTest,PointerType_CXX11)1717 TEST_P(ASTMatchersTest, PointerType_CXX11) {
1718   if (!GetParam().isCXX11OrLater()) {
1719     return;
1720   }
1721   StringRef Fragment = "int &&ref = 2;";
1722   EXPECT_TRUE(notMatches(Fragment,
1723                          varDecl(hasName("ref"), hasType(blockPointerType()))));
1724   EXPECT_TRUE(notMatches(
1725       Fragment, varDecl(hasName("ref"), hasType(memberPointerType()))));
1726   EXPECT_TRUE(
1727       notMatches(Fragment, varDecl(hasName("ref"), hasType(pointerType()))));
1728   EXPECT_TRUE(
1729       matches(Fragment, varDecl(hasName("ref"), hasType(referenceType()))));
1730   EXPECT_TRUE(notMatches(
1731       Fragment, varDecl(hasName("ref"), hasType(lValueReferenceType()))));
1732   EXPECT_TRUE(matches(Fragment,
1733                       varDecl(hasName("ref"), hasType(rValueReferenceType()))));
1734 }
1735 
TEST_P(ASTMatchersTest,AutoRefTypes)1736 TEST_P(ASTMatchersTest, AutoRefTypes) {
1737   if (!GetParam().isCXX11OrLater()) {
1738     return;
1739   }
1740 
1741   StringRef Fragment = "auto a = 1;"
1742                        "auto b = a;"
1743                        "auto &c = a;"
1744                        "auto &&d = c;"
1745                        "auto &&e = 2;";
1746   EXPECT_TRUE(
1747       notMatches(Fragment, varDecl(hasName("a"), hasType(referenceType()))));
1748   EXPECT_TRUE(
1749       notMatches(Fragment, varDecl(hasName("b"), hasType(referenceType()))));
1750   EXPECT_TRUE(
1751       matches(Fragment, varDecl(hasName("c"), hasType(referenceType()))));
1752   EXPECT_TRUE(
1753       matches(Fragment, varDecl(hasName("c"), hasType(lValueReferenceType()))));
1754   EXPECT_TRUE(notMatches(
1755       Fragment, varDecl(hasName("c"), hasType(rValueReferenceType()))));
1756   EXPECT_TRUE(
1757       matches(Fragment, varDecl(hasName("d"), hasType(referenceType()))));
1758   EXPECT_TRUE(
1759       matches(Fragment, varDecl(hasName("d"), hasType(lValueReferenceType()))));
1760   EXPECT_TRUE(notMatches(
1761       Fragment, varDecl(hasName("d"), hasType(rValueReferenceType()))));
1762   EXPECT_TRUE(
1763       matches(Fragment, varDecl(hasName("e"), hasType(referenceType()))));
1764   EXPECT_TRUE(notMatches(
1765       Fragment, varDecl(hasName("e"), hasType(lValueReferenceType()))));
1766   EXPECT_TRUE(
1767       matches(Fragment, varDecl(hasName("e"), hasType(rValueReferenceType()))));
1768 }
1769 
TEST_P(ASTMatchersTest,EnumType)1770 TEST_P(ASTMatchersTest, EnumType) {
1771   EXPECT_TRUE(
1772       matches("enum Color { Green }; enum Color color;", loc(enumType())));
1773 }
1774 
TEST_P(ASTMatchersTest,EnumType_CXX)1775 TEST_P(ASTMatchersTest, EnumType_CXX) {
1776   if (!GetParam().isCXX()) {
1777     return;
1778   }
1779   EXPECT_TRUE(matches("enum Color { Green }; Color color;", loc(enumType())));
1780 }
1781 
TEST_P(ASTMatchersTest,EnumType_CXX11)1782 TEST_P(ASTMatchersTest, EnumType_CXX11) {
1783   if (!GetParam().isCXX11OrLater()) {
1784     return;
1785   }
1786   EXPECT_TRUE(
1787       matches("enum class Color { Green }; Color color;", loc(enumType())));
1788 }
1789 
TEST_P(ASTMatchersTest,PointerType_MatchesPointersToConstTypes)1790 TEST_P(ASTMatchersTest, PointerType_MatchesPointersToConstTypes) {
1791   EXPECT_TRUE(matches("int b; int * const a = &b;", loc(pointerType())));
1792   EXPECT_TRUE(matches("int b; int * const a = &b;", loc(pointerType())));
1793   EXPECT_TRUE(matches("int b; const int * a = &b;",
1794                       loc(pointerType(pointee(builtinType())))));
1795   EXPECT_TRUE(matches("int b; const int * a = &b;",
1796                       pointerType(pointee(builtinType()))));
1797 }
1798 
TEST_P(ASTMatchersTest,TypedefType)1799 TEST_P(ASTMatchersTest, TypedefType) {
1800   EXPECT_TRUE(matches("typedef int X; X a;",
1801                       varDecl(hasName("a"), hasType(typedefType()))));
1802 }
1803 
TEST_P(ASTMatchersTest,TemplateSpecializationType)1804 TEST_P(ASTMatchersTest, TemplateSpecializationType) {
1805   if (!GetParam().isCXX()) {
1806     return;
1807   }
1808   EXPECT_TRUE(matches("template <typename T> class A{}; A<int> a;",
1809                       templateSpecializationType()));
1810 }
1811 
TEST_P(ASTMatchersTest,DeducedTemplateSpecializationType)1812 TEST_P(ASTMatchersTest, DeducedTemplateSpecializationType) {
1813   if (!GetParam().isCXX17OrLater()) {
1814     return;
1815   }
1816   EXPECT_TRUE(
1817       matches("template <typename T> class A{ public: A(T) {} }; A a(1);",
1818               deducedTemplateSpecializationType()));
1819 }
1820 
TEST_P(ASTMatchersTest,RecordType)1821 TEST_P(ASTMatchersTest, RecordType) {
1822   EXPECT_TRUE(matches("struct S {}; struct S s;",
1823                       recordType(hasDeclaration(recordDecl(hasName("S"))))));
1824   EXPECT_TRUE(notMatches("int i;",
1825                          recordType(hasDeclaration(recordDecl(hasName("S"))))));
1826 }
1827 
TEST_P(ASTMatchersTest,RecordType_CXX)1828 TEST_P(ASTMatchersTest, RecordType_CXX) {
1829   if (!GetParam().isCXX()) {
1830     return;
1831   }
1832   EXPECT_TRUE(matches("class C {}; C c;", recordType()));
1833   EXPECT_TRUE(matches("struct S {}; S s;",
1834                       recordType(hasDeclaration(recordDecl(hasName("S"))))));
1835 }
1836 
TEST_P(ASTMatchersTest,ElaboratedType)1837 TEST_P(ASTMatchersTest, ElaboratedType) {
1838   if (!GetParam().isCXX()) {
1839     // FIXME: Add a test for `elaboratedType()` that does not depend on C++.
1840     return;
1841   }
1842   EXPECT_TRUE(matches("namespace N {"
1843                       "  namespace M {"
1844                       "    class D {};"
1845                       "  }"
1846                       "}"
1847                       "N::M::D d;",
1848                       elaboratedType()));
1849   EXPECT_TRUE(matches("class C {} c;", elaboratedType()));
1850   EXPECT_TRUE(notMatches("class C {}; C c;", elaboratedType()));
1851 }
1852 
TEST_P(ASTMatchersTest,SubstTemplateTypeParmType)1853 TEST_P(ASTMatchersTest, SubstTemplateTypeParmType) {
1854   if (!GetParam().isCXX()) {
1855     return;
1856   }
1857   StringRef code = "template <typename T>"
1858                    "int F() {"
1859                    "  return 1 + T();"
1860                    "}"
1861                    "int i = F<int>();";
1862   EXPECT_FALSE(matches(code, binaryOperator(hasLHS(
1863                                  expr(hasType(substTemplateTypeParmType()))))));
1864   EXPECT_TRUE(matches(code, binaryOperator(hasRHS(
1865                                 expr(hasType(substTemplateTypeParmType()))))));
1866 }
1867 
TEST_P(ASTMatchersTest,NestedNameSpecifier)1868 TEST_P(ASTMatchersTest, NestedNameSpecifier) {
1869   if (!GetParam().isCXX()) {
1870     return;
1871   }
1872   EXPECT_TRUE(
1873       matches("namespace ns { struct A {}; } ns::A a;", nestedNameSpecifier()));
1874   EXPECT_TRUE(matches("template <typename T> class A { typename T::B b; };",
1875                       nestedNameSpecifier()));
1876   EXPECT_TRUE(
1877       matches("struct A { void f(); }; void A::f() {}", nestedNameSpecifier()));
1878   EXPECT_TRUE(matches("namespace a { namespace b {} } namespace ab = a::b;",
1879                       nestedNameSpecifier()));
1880 
1881   EXPECT_TRUE(matches("struct A { static void f() {} }; void g() { A::f(); }",
1882                       nestedNameSpecifier()));
1883   EXPECT_TRUE(
1884       notMatches("struct A { static void f() {} }; void g(A* a) { a->f(); }",
1885                  nestedNameSpecifier()));
1886 }
1887 
TEST_P(ASTMatchersTest,Attr)1888 TEST_P(ASTMatchersTest, Attr) {
1889   // Windows adds some implicit attributes.
1890   bool AutomaticAttributes = StringRef(GetParam().Target).contains("win32");
1891   if (GetParam().isCXX11OrLater()) {
1892     EXPECT_TRUE(matches("struct [[clang::warn_unused_result]] F{};", attr()));
1893 
1894     // Unknown attributes are not parsed into an AST node.
1895     if (!AutomaticAttributes) {
1896       EXPECT_TRUE(notMatches("int x [[unknownattr]];", attr()));
1897     }
1898   }
1899   if (GetParam().isCXX17OrLater()) {
1900     EXPECT_TRUE(matches("struct [[nodiscard]] F{};", attr()));
1901   }
1902   EXPECT_TRUE(matches("int x(int * __attribute__((nonnull)) );", attr()));
1903   if (!AutomaticAttributes) {
1904     EXPECT_TRUE(notMatches("struct F{}; int x(int *);", attr()));
1905     // Some known attributes are not parsed into an AST node.
1906     EXPECT_TRUE(notMatches("typedef int x __attribute__((ext_vector_type(1)));",
1907                            attr()));
1908   }
1909 }
1910 
TEST_P(ASTMatchersTest,NullStmt)1911 TEST_P(ASTMatchersTest, NullStmt) {
1912   EXPECT_TRUE(matches("void f() {int i;;}", nullStmt()));
1913   EXPECT_TRUE(notMatches("void f() {int i;}", nullStmt()));
1914 }
1915 
TEST_P(ASTMatchersTest,NamespaceAliasDecl)1916 TEST_P(ASTMatchersTest, NamespaceAliasDecl) {
1917   if (!GetParam().isCXX()) {
1918     return;
1919   }
1920   EXPECT_TRUE(matches("namespace test {} namespace alias = ::test;",
1921                       namespaceAliasDecl(hasName("alias"))));
1922 }
1923 
TEST_P(ASTMatchersTest,NestedNameSpecifier_MatchesTypes)1924 TEST_P(ASTMatchersTest, NestedNameSpecifier_MatchesTypes) {
1925   if (!GetParam().isCXX()) {
1926     return;
1927   }
1928   NestedNameSpecifierMatcher Matcher = nestedNameSpecifier(
1929       specifiesType(hasDeclaration(recordDecl(hasName("A")))));
1930   EXPECT_TRUE(matches("struct A { struct B {}; }; A::B b;", Matcher));
1931   EXPECT_TRUE(
1932       matches("struct A { struct B { struct C {}; }; }; A::B::C c;", Matcher));
1933   EXPECT_TRUE(notMatches("namespace A { struct B {}; } A::B b;", Matcher));
1934 }
1935 
TEST_P(ASTMatchersTest,NestedNameSpecifier_MatchesNamespaceDecls)1936 TEST_P(ASTMatchersTest, NestedNameSpecifier_MatchesNamespaceDecls) {
1937   if (!GetParam().isCXX()) {
1938     return;
1939   }
1940   NestedNameSpecifierMatcher Matcher =
1941       nestedNameSpecifier(specifiesNamespace(hasName("ns")));
1942   EXPECT_TRUE(matches("namespace ns { struct A {}; } ns::A a;", Matcher));
1943   EXPECT_TRUE(notMatches("namespace xx { struct A {}; } xx::A a;", Matcher));
1944   EXPECT_TRUE(notMatches("struct ns { struct A {}; }; ns::A a;", Matcher));
1945 }
1946 
TEST_P(ASTMatchersTest,NestedNameSpecifier_MatchesNestedNameSpecifierPrefixes)1947 TEST_P(ASTMatchersTest,
1948        NestedNameSpecifier_MatchesNestedNameSpecifierPrefixes) {
1949   if (!GetParam().isCXX()) {
1950     return;
1951   }
1952   EXPECT_TRUE(matches(
1953       "struct A { struct B { struct C {}; }; }; A::B::C c;",
1954       nestedNameSpecifier(hasPrefix(specifiesType(asString("struct A"))))));
1955   EXPECT_TRUE(matches("struct A { struct B { struct C {}; }; }; A::B::C c;",
1956                       nestedNameSpecifierLoc(hasPrefix(specifiesTypeLoc(
1957                           loc(qualType(asString("struct A"))))))));
1958   EXPECT_TRUE(matches(
1959       "namespace N { struct A { struct B { struct C {}; }; }; } N::A::B::C c;",
1960       nestedNameSpecifierLoc(hasPrefix(
1961           specifiesTypeLoc(loc(qualType(asString("struct N::A"))))))));
1962 }
1963 
1964 template <typename T>
1965 class VerifyAncestorHasChildIsEqual : public BoundNodesCallback {
1966 public:
run(const BoundNodes * Nodes)1967   bool run(const BoundNodes *Nodes) override { return false; }
1968 
run(const BoundNodes * Nodes,ASTContext * Context)1969   bool run(const BoundNodes *Nodes, ASTContext *Context) override {
1970     const T *Node = Nodes->getNodeAs<T>("");
1971     return verify(*Nodes, *Context, Node);
1972   }
1973 
verify(const BoundNodes & Nodes,ASTContext & Context,const Stmt * Node)1974   bool verify(const BoundNodes &Nodes, ASTContext &Context, const Stmt *Node) {
1975     // Use the original typed pointer to verify we can pass pointers to subtypes
1976     // to equalsNode.
1977     const T *TypedNode = cast<T>(Node);
1978     return selectFirst<T>(
1979                "", match(stmt(hasParent(
1980                              stmt(has(stmt(equalsNode(TypedNode)))).bind(""))),
1981                          *Node, Context)) != nullptr;
1982   }
verify(const BoundNodes & Nodes,ASTContext & Context,const Decl * Node)1983   bool verify(const BoundNodes &Nodes, ASTContext &Context, const Decl *Node) {
1984     // Use the original typed pointer to verify we can pass pointers to subtypes
1985     // to equalsNode.
1986     const T *TypedNode = cast<T>(Node);
1987     return selectFirst<T>(
1988                "", match(decl(hasParent(
1989                              decl(has(decl(equalsNode(TypedNode)))).bind(""))),
1990                          *Node, Context)) != nullptr;
1991   }
verify(const BoundNodes & Nodes,ASTContext & Context,const Type * Node)1992   bool verify(const BoundNodes &Nodes, ASTContext &Context, const Type *Node) {
1993     // Use the original typed pointer to verify we can pass pointers to subtypes
1994     // to equalsNode.
1995     const T *TypedNode = cast<T>(Node);
1996     const auto *Dec = Nodes.getNodeAs<FieldDecl>("decl");
1997     return selectFirst<T>(
1998                "", match(fieldDecl(hasParent(decl(has(fieldDecl(
1999                              hasType(type(equalsNode(TypedNode)).bind(""))))))),
2000                          *Dec, Context)) != nullptr;
2001   }
2002 };
2003 
TEST_P(ASTMatchersTest,IsEqualTo_MatchesNodesByIdentity)2004 TEST_P(ASTMatchersTest, IsEqualTo_MatchesNodesByIdentity) {
2005   EXPECT_TRUE(matchAndVerifyResultTrue(
2006       "void f() { if (1) if(1) {} }", ifStmt().bind(""),
2007       std::make_unique<VerifyAncestorHasChildIsEqual<IfStmt>>()));
2008 }
2009 
TEST_P(ASTMatchersTest,IsEqualTo_MatchesNodesByIdentity_Cxx)2010 TEST_P(ASTMatchersTest, IsEqualTo_MatchesNodesByIdentity_Cxx) {
2011   if (!GetParam().isCXX()) {
2012     return;
2013   }
2014   EXPECT_TRUE(matchAndVerifyResultTrue(
2015       "class X { class Y {}; };", recordDecl(hasName("::X::Y")).bind(""),
2016       std::make_unique<VerifyAncestorHasChildIsEqual<CXXRecordDecl>>()));
2017   EXPECT_TRUE(matchAndVerifyResultTrue(
2018       "class X { class Y {} y; };",
2019       fieldDecl(hasName("y"), hasType(type().bind(""))).bind("decl"),
2020       std::make_unique<VerifyAncestorHasChildIsEqual<Type>>()));
2021 }
2022 
TEST_P(ASTMatchersTest,TypedefDecl)2023 TEST_P(ASTMatchersTest, TypedefDecl) {
2024   EXPECT_TRUE(matches("typedef int typedefDeclTest;",
2025                       typedefDecl(hasName("typedefDeclTest"))));
2026 }
2027 
TEST_P(ASTMatchersTest,TypedefDecl_Cxx)2028 TEST_P(ASTMatchersTest, TypedefDecl_Cxx) {
2029   if (!GetParam().isCXX11OrLater()) {
2030     return;
2031   }
2032   EXPECT_TRUE(notMatches("using typedefDeclTest = int;",
2033                          typedefDecl(hasName("typedefDeclTest"))));
2034 }
2035 
TEST_P(ASTMatchersTest,TypeAliasDecl)2036 TEST_P(ASTMatchersTest, TypeAliasDecl) {
2037   EXPECT_TRUE(notMatches("typedef int typeAliasTest;",
2038                          typeAliasDecl(hasName("typeAliasTest"))));
2039 }
2040 
TEST_P(ASTMatchersTest,TypeAliasDecl_CXX)2041 TEST_P(ASTMatchersTest, TypeAliasDecl_CXX) {
2042   if (!GetParam().isCXX11OrLater()) {
2043     return;
2044   }
2045   EXPECT_TRUE(matches("using typeAliasTest = int;",
2046                       typeAliasDecl(hasName("typeAliasTest"))));
2047 }
2048 
TEST_P(ASTMatchersTest,TypedefNameDecl)2049 TEST_P(ASTMatchersTest, TypedefNameDecl) {
2050   EXPECT_TRUE(matches("typedef int typedefNameDeclTest1;",
2051                       typedefNameDecl(hasName("typedefNameDeclTest1"))));
2052 }
2053 
TEST_P(ASTMatchersTest,TypedefNameDecl_CXX)2054 TEST_P(ASTMatchersTest, TypedefNameDecl_CXX) {
2055   if (!GetParam().isCXX11OrLater()) {
2056     return;
2057   }
2058   EXPECT_TRUE(matches("using typedefNameDeclTest = int;",
2059                       typedefNameDecl(hasName("typedefNameDeclTest"))));
2060 }
2061 
TEST_P(ASTMatchersTest,TypeAliasTemplateDecl)2062 TEST_P(ASTMatchersTest, TypeAliasTemplateDecl) {
2063   if (!GetParam().isCXX11OrLater()) {
2064     return;
2065   }
2066   StringRef Code = R"(
2067     template <typename T>
2068     class X { T t; };
2069 
2070     template <typename T>
2071     using typeAliasTemplateDecl = X<T>;
2072 
2073     using typeAliasDecl = X<int>;
2074   )";
2075   EXPECT_TRUE(
2076       matches(Code, typeAliasTemplateDecl(hasName("typeAliasTemplateDecl"))));
2077   EXPECT_TRUE(
2078       notMatches(Code, typeAliasTemplateDecl(hasName("typeAliasDecl"))));
2079 }
2080 
TEST_P(ASTMatchersTest,QualifiedTypeLocTest_BindsToConstIntVarDecl)2081 TEST_P(ASTMatchersTest, QualifiedTypeLocTest_BindsToConstIntVarDecl) {
2082   EXPECT_TRUE(matches("const int x = 0;",
2083                       qualifiedTypeLoc(loc(asString("const int")))));
2084 }
2085 
TEST_P(ASTMatchersTest,QualifiedTypeLocTest_BindsToConstIntFunctionDecl)2086 TEST_P(ASTMatchersTest, QualifiedTypeLocTest_BindsToConstIntFunctionDecl) {
2087   EXPECT_TRUE(matches("const int f() { return 5; }",
2088                       qualifiedTypeLoc(loc(asString("const int")))));
2089 }
2090 
TEST_P(ASTMatchersTest,QualifiedTypeLocTest_DoesNotBindToUnqualifiedVarDecl)2091 TEST_P(ASTMatchersTest, QualifiedTypeLocTest_DoesNotBindToUnqualifiedVarDecl) {
2092   EXPECT_TRUE(notMatches("int x = 0;", qualifiedTypeLoc(loc(asString("int")))));
2093 }
2094 
TEST_P(ASTMatchersTest,QualifiedTypeLocTest_IntDoesNotBindToConstIntDecl)2095 TEST_P(ASTMatchersTest, QualifiedTypeLocTest_IntDoesNotBindToConstIntDecl) {
2096   EXPECT_TRUE(
2097       notMatches("const int x = 0;", qualifiedTypeLoc(loc(asString("int")))));
2098 }
2099 
TEST_P(ASTMatchersTest,QualifiedTypeLocTest_IntDoesNotBindToConstFloatDecl)2100 TEST_P(ASTMatchersTest, QualifiedTypeLocTest_IntDoesNotBindToConstFloatDecl) {
2101   EXPECT_TRUE(
2102       notMatches("const float x = 0;", qualifiedTypeLoc(loc(asString("int")))));
2103 }
2104 
TEST_P(ASTMatchersTest,PointerTypeLocTest_BindsToAnyPointerTypeLoc)2105 TEST_P(ASTMatchersTest, PointerTypeLocTest_BindsToAnyPointerTypeLoc) {
2106   auto matcher = varDecl(hasName("x"), hasTypeLoc(pointerTypeLoc()));
2107   EXPECT_TRUE(matches("int* x;", matcher));
2108   EXPECT_TRUE(matches("float* x;", matcher));
2109   EXPECT_TRUE(matches("char* x;", matcher));
2110   EXPECT_TRUE(matches("void* x;", matcher));
2111 }
2112 
TEST_P(ASTMatchersTest,PointerTypeLocTest_DoesNotBindToNonPointerTypeLoc)2113 TEST_P(ASTMatchersTest, PointerTypeLocTest_DoesNotBindToNonPointerTypeLoc) {
2114   auto matcher = varDecl(hasName("x"), hasTypeLoc(pointerTypeLoc()));
2115   EXPECT_TRUE(notMatches("int x;", matcher));
2116   EXPECT_TRUE(notMatches("float x;", matcher));
2117   EXPECT_TRUE(notMatches("char x;", matcher));
2118 }
2119 
TEST_P(ASTMatchersTest,ReferenceTypeLocTest_BindsToAnyReferenceTypeLoc)2120 TEST_P(ASTMatchersTest, ReferenceTypeLocTest_BindsToAnyReferenceTypeLoc) {
2121   if (!GetParam().isCXX()) {
2122     return;
2123   }
2124   auto matcher = varDecl(hasName("r"), hasTypeLoc(referenceTypeLoc()));
2125   EXPECT_TRUE(matches("int rr = 3; int& r = rr;", matcher));
2126   EXPECT_TRUE(matches("int rr = 3; auto& r = rr;", matcher));
2127   EXPECT_TRUE(matches("int rr = 3; const int& r = rr;", matcher));
2128   EXPECT_TRUE(matches("float rr = 3.0; float& r = rr;", matcher));
2129   EXPECT_TRUE(matches("char rr = 'a'; char& r = rr;", matcher));
2130 }
2131 
TEST_P(ASTMatchersTest,ReferenceTypeLocTest_DoesNotBindToNonReferenceTypeLoc)2132 TEST_P(ASTMatchersTest, ReferenceTypeLocTest_DoesNotBindToNonReferenceTypeLoc) {
2133   auto matcher = varDecl(hasName("r"), hasTypeLoc(referenceTypeLoc()));
2134   EXPECT_TRUE(notMatches("int r;", matcher));
2135   EXPECT_TRUE(notMatches("int r = 3;", matcher));
2136   EXPECT_TRUE(notMatches("const int r = 3;", matcher));
2137   EXPECT_TRUE(notMatches("int* r;", matcher));
2138   EXPECT_TRUE(notMatches("float r;", matcher));
2139   EXPECT_TRUE(notMatches("char r;", matcher));
2140 }
2141 
TEST_P(ASTMatchersTest,ReferenceTypeLocTest_BindsToAnyRvalueReferenceTypeLoc)2142 TEST_P(ASTMatchersTest, ReferenceTypeLocTest_BindsToAnyRvalueReferenceTypeLoc) {
2143   if (!GetParam().isCXX()) {
2144     return;
2145   }
2146   auto matcher = varDecl(hasName("r"), hasTypeLoc(referenceTypeLoc()));
2147   EXPECT_TRUE(matches("int&& r = 3;", matcher));
2148   EXPECT_TRUE(matches("auto&& r = 3;", matcher));
2149   EXPECT_TRUE(matches("float&& r = 3.0;", matcher));
2150 }
2151 
TEST_P(ASTMatchersTest,TemplateSpecializationTypeLocTest_BindsToTemplateSpecializationExplicitInstantiation)2152 TEST_P(
2153     ASTMatchersTest,
2154     TemplateSpecializationTypeLocTest_BindsToTemplateSpecializationExplicitInstantiation) {
2155   if (!GetParam().isCXX()) {
2156     return;
2157   }
2158   EXPECT_TRUE(
2159       matches("template <typename T> class C {}; template class C<int>;",
2160               classTemplateSpecializationDecl(
2161                   hasName("C"), hasTypeLoc(templateSpecializationTypeLoc()))));
2162 }
2163 
TEST_P(ASTMatchersTest,TemplateSpecializationTypeLocTest_BindsToVarDeclTemplateSpecialization)2164 TEST_P(ASTMatchersTest,
2165        TemplateSpecializationTypeLocTest_BindsToVarDeclTemplateSpecialization) {
2166   if (!GetParam().isCXX()) {
2167     return;
2168   }
2169   EXPECT_TRUE(matches(
2170       "template <typename T> class C {}; C<char> var;",
2171       varDecl(hasName("var"), hasTypeLoc(templateSpecializationTypeLoc()))));
2172 }
2173 
TEST_P(ASTMatchersTest,TemplateSpecializationTypeLocTest_DoesNotBindToNonTemplateSpecialization)2174 TEST_P(
2175     ASTMatchersTest,
2176     TemplateSpecializationTypeLocTest_DoesNotBindToNonTemplateSpecialization) {
2177   if (!GetParam().isCXX()) {
2178     return;
2179   }
2180   EXPECT_TRUE(notMatches(
2181       "class C {}; C var;",
2182       varDecl(hasName("var"), hasTypeLoc(templateSpecializationTypeLoc()))));
2183 }
2184 
TEST_P(ASTMatchersTest,ElaboratedTypeLocTest_BindsToElaboratedObjectDeclaration)2185 TEST_P(ASTMatchersTest,
2186        ElaboratedTypeLocTest_BindsToElaboratedObjectDeclaration) {
2187   if (!GetParam().isCXX()) {
2188     return;
2189   }
2190   EXPECT_TRUE(matches("class C {}; class C c;",
2191                       varDecl(hasName("c"), hasTypeLoc(elaboratedTypeLoc()))));
2192 }
2193 
TEST_P(ASTMatchersTest,ElaboratedTypeLocTest_BindsToNamespaceElaboratedObjectDeclaration)2194 TEST_P(ASTMatchersTest,
2195        ElaboratedTypeLocTest_BindsToNamespaceElaboratedObjectDeclaration) {
2196   if (!GetParam().isCXX()) {
2197     return;
2198   }
2199   EXPECT_TRUE(matches("namespace N { class D {}; } N::D d;",
2200                       varDecl(hasName("d"), hasTypeLoc(elaboratedTypeLoc()))));
2201 }
2202 
TEST_P(ASTMatchersTest,ElaboratedTypeLocTest_BindsToElaboratedStructDeclaration)2203 TEST_P(ASTMatchersTest,
2204        ElaboratedTypeLocTest_BindsToElaboratedStructDeclaration) {
2205   EXPECT_TRUE(matches("struct s {}; struct s ss;",
2206                       varDecl(hasName("ss"), hasTypeLoc(elaboratedTypeLoc()))));
2207 }
2208 
TEST_P(ASTMatchersTest,ElaboratedTypeLocTest_DoesNotBindToNonElaboratedObjectDeclaration)2209 TEST_P(ASTMatchersTest,
2210        ElaboratedTypeLocTest_DoesNotBindToNonElaboratedObjectDeclaration) {
2211   if (!GetParam().isCXX()) {
2212     return;
2213   }
2214   EXPECT_TRUE(
2215       notMatches("class C {}; C c;",
2216                  varDecl(hasName("c"), hasTypeLoc(elaboratedTypeLoc()))));
2217 }
2218 
TEST_P(ASTMatchersTest,ElaboratedTypeLocTest_DoesNotBindToNamespaceNonElaboratedObjectDeclaration)2219 TEST_P(
2220     ASTMatchersTest,
2221     ElaboratedTypeLocTest_DoesNotBindToNamespaceNonElaboratedObjectDeclaration) {
2222   if (!GetParam().isCXX()) {
2223     return;
2224   }
2225   EXPECT_TRUE(
2226       notMatches("namespace N { class D {}; } using N::D; D d;",
2227                  varDecl(hasName("d"), hasTypeLoc(elaboratedTypeLoc()))));
2228 }
2229 
TEST_P(ASTMatchersTest,ElaboratedTypeLocTest_DoesNotBindToNonElaboratedStructDeclaration)2230 TEST_P(ASTMatchersTest,
2231        ElaboratedTypeLocTest_DoesNotBindToNonElaboratedStructDeclaration) {
2232   if (!GetParam().isCXX()) {
2233     return;
2234   }
2235   EXPECT_TRUE(
2236       notMatches("struct s {}; s ss;",
2237                  varDecl(hasName("ss"), hasTypeLoc(elaboratedTypeLoc()))));
2238 }
2239 
TEST(ASTMatchersTestObjC,ObjCMessageExpr)2240 TEST(ASTMatchersTestObjC, ObjCMessageExpr) {
2241   // Don't find ObjCMessageExpr where none are present.
2242   EXPECT_TRUE(notMatchesObjC("", objcMessageExpr(anything())));
2243 
2244   StringRef Objc1String = "@interface Str "
2245                           " - (Str *)uppercaseString;"
2246                           "@end "
2247                           "@interface foo "
2248                           "- (void)contents;"
2249                           "- (void)meth:(Str *)text;"
2250                           "@end "
2251                           " "
2252                           "@implementation foo "
2253                           "- (void) meth:(Str *)text { "
2254                           "  [self contents];"
2255                           "  Str *up = [text uppercaseString];"
2256                           "} "
2257                           "@end ";
2258   EXPECT_TRUE(matchesObjC(Objc1String, objcMessageExpr(anything())));
2259   EXPECT_TRUE(matchesObjC(Objc1String,
2260                           objcMessageExpr(hasAnySelector({"contents", "meth:"}))
2261 
2262                               ));
2263   EXPECT_TRUE(
2264       matchesObjC(Objc1String, objcMessageExpr(hasSelector("contents"))));
2265   EXPECT_TRUE(matchesObjC(
2266       Objc1String, objcMessageExpr(hasAnySelector("contents", "contentsA"))));
2267   EXPECT_FALSE(matchesObjC(
2268       Objc1String, objcMessageExpr(hasAnySelector("contentsB", "contentsC"))));
2269   EXPECT_TRUE(
2270       matchesObjC(Objc1String, objcMessageExpr(matchesSelector("cont*"))));
2271   EXPECT_FALSE(
2272       matchesObjC(Objc1String, objcMessageExpr(matchesSelector("?cont*"))));
2273   EXPECT_TRUE(
2274       notMatchesObjC(Objc1String, objcMessageExpr(hasSelector("contents"),
2275                                                   hasNullSelector())));
2276   EXPECT_TRUE(matchesObjC(Objc1String, objcMessageExpr(hasSelector("contents"),
2277                                                        hasUnarySelector())));
2278   EXPECT_TRUE(matchesObjC(Objc1String, objcMessageExpr(hasSelector("contents"),
2279                                                        numSelectorArgs(0))));
2280   EXPECT_TRUE(
2281       matchesObjC(Objc1String, objcMessageExpr(matchesSelector("uppercase*"),
2282                                                argumentCountIs(0))));
2283 }
2284 
TEST(ASTMatchersTestObjC,ObjCDecls)2285 TEST(ASTMatchersTestObjC, ObjCDecls) {
2286   StringRef ObjCString = "@protocol Proto "
2287                          "- (void)protoDidThing; "
2288                          "@end "
2289                          "@interface Thing "
2290                          "@property int enabled; "
2291                          "@end "
2292                          "@interface Thing (ABC) "
2293                          "- (void)abc_doThing; "
2294                          "@end "
2295                          "@implementation Thing "
2296                          "{ id _ivar; } "
2297                          "- (void)anything {} "
2298                          "@end "
2299                          "@implementation Thing (ABC) "
2300                          "- (void)abc_doThing {} "
2301                          "@end ";
2302 
2303   EXPECT_TRUE(matchesObjC(ObjCString, objcProtocolDecl(hasName("Proto"))));
2304   EXPECT_TRUE(
2305       matchesObjC(ObjCString, objcImplementationDecl(hasName("Thing"))));
2306   EXPECT_TRUE(matchesObjC(ObjCString, objcCategoryDecl(hasName("ABC"))));
2307   EXPECT_TRUE(matchesObjC(ObjCString, objcCategoryImplDecl(hasName("ABC"))));
2308   EXPECT_TRUE(
2309       matchesObjC(ObjCString, objcMethodDecl(hasName("protoDidThing"))));
2310   EXPECT_TRUE(matchesObjC(ObjCString, objcMethodDecl(hasName("abc_doThing"))));
2311   EXPECT_TRUE(matchesObjC(ObjCString, objcMethodDecl(hasName("anything"))));
2312   EXPECT_TRUE(matchesObjC(ObjCString, objcIvarDecl(hasName("_ivar"))));
2313   EXPECT_TRUE(matchesObjC(ObjCString, objcPropertyDecl(hasName("enabled"))));
2314 }
2315 
TEST(ASTMatchersTestObjC,ObjCExceptionStmts)2316 TEST(ASTMatchersTestObjC, ObjCExceptionStmts) {
2317   StringRef ObjCString = "void f(id obj) {"
2318                          "  @try {"
2319                          "    @throw obj;"
2320                          "  } @catch (...) {"
2321                          "  } @finally {}"
2322                          "}";
2323 
2324   EXPECT_TRUE(matchesObjC(ObjCString, objcTryStmt()));
2325   EXPECT_TRUE(matchesObjC(ObjCString, objcThrowStmt()));
2326   EXPECT_TRUE(matchesObjC(ObjCString, objcCatchStmt()));
2327   EXPECT_TRUE(matchesObjC(ObjCString, objcFinallyStmt()));
2328 }
2329 
TEST(ASTMatchersTest,DecompositionDecl)2330 TEST(ASTMatchersTest, DecompositionDecl) {
2331   StringRef Code = R"cpp(
2332 void foo()
2333 {
2334     int arr[3];
2335     auto &[f, s, t] = arr;
2336 
2337     f = 42;
2338 }
2339   )cpp";
2340   EXPECT_TRUE(matchesConditionally(
2341       Code, decompositionDecl(hasBinding(0, bindingDecl(hasName("f")))), true,
2342       {"-std=c++17"}));
2343   EXPECT_FALSE(matchesConditionally(
2344       Code, decompositionDecl(hasBinding(42, bindingDecl(hasName("f")))), true,
2345       {"-std=c++17"}));
2346   EXPECT_FALSE(matchesConditionally(
2347       Code, decompositionDecl(hasBinding(0, bindingDecl(hasName("s")))), true,
2348       {"-std=c++17"}));
2349   EXPECT_TRUE(matchesConditionally(
2350       Code, decompositionDecl(hasBinding(1, bindingDecl(hasName("s")))), true,
2351       {"-std=c++17"}));
2352 
2353   EXPECT_TRUE(matchesConditionally(
2354       Code,
2355       bindingDecl(decl().bind("self"), hasName("f"),
2356                   forDecomposition(decompositionDecl(
2357                       hasAnyBinding(bindingDecl(equalsBoundNode("self")))))),
2358       true, {"-std=c++17"}));
2359 }
2360 
TEST(ASTMatchersTestObjC,ObjCAutoreleasePoolStmt)2361 TEST(ASTMatchersTestObjC, ObjCAutoreleasePoolStmt) {
2362   StringRef ObjCString = "void f() {"
2363                          "@autoreleasepool {"
2364                          "  int x = 1;"
2365                          "}"
2366                          "}";
2367   EXPECT_TRUE(matchesObjC(ObjCString, autoreleasePoolStmt()));
2368   StringRef ObjCStringNoPool = "void f() { int x = 1; }";
2369   EXPECT_FALSE(matchesObjC(ObjCStringNoPool, autoreleasePoolStmt()));
2370 }
2371 
TEST(ASTMatchersTestOpenMP,OMPExecutableDirective)2372 TEST(ASTMatchersTestOpenMP, OMPExecutableDirective) {
2373   auto Matcher = stmt(ompExecutableDirective());
2374 
2375   StringRef Source0 = R"(
2376 void x() {
2377 #pragma omp parallel
2378 ;
2379 })";
2380   EXPECT_TRUE(matchesWithOpenMP(Source0, Matcher));
2381 
2382   StringRef Source1 = R"(
2383 void x() {
2384 #pragma omp taskyield
2385 ;
2386 })";
2387   EXPECT_TRUE(matchesWithOpenMP(Source1, Matcher));
2388 
2389   StringRef Source2 = R"(
2390 void x() {
2391 ;
2392 })";
2393   EXPECT_TRUE(notMatchesWithOpenMP(Source2, Matcher));
2394 }
2395 
TEST(ASTMatchersTestOpenMP,OMPDefaultClause)2396 TEST(ASTMatchersTestOpenMP, OMPDefaultClause) {
2397   auto Matcher = ompExecutableDirective(hasAnyClause(ompDefaultClause()));
2398 
2399   StringRef Source0 = R"(
2400 void x() {
2401 ;
2402 })";
2403   EXPECT_TRUE(notMatchesWithOpenMP(Source0, Matcher));
2404 
2405   StringRef Source1 = R"(
2406 void x() {
2407 #pragma omp parallel
2408 ;
2409 })";
2410   EXPECT_TRUE(notMatchesWithOpenMP(Source1, Matcher));
2411 
2412   StringRef Source2 = R"(
2413 void x() {
2414 #pragma omp parallel default(none)
2415 ;
2416 })";
2417   EXPECT_TRUE(matchesWithOpenMP(Source2, Matcher));
2418 
2419   StringRef Source3 = R"(
2420 void x() {
2421 #pragma omp parallel default(shared)
2422 ;
2423 })";
2424   EXPECT_TRUE(matchesWithOpenMP(Source3, Matcher));
2425 
2426   StringRef Source4 = R"(
2427 void x() {
2428 #pragma omp parallel default(firstprivate)
2429 ;
2430 })";
2431   EXPECT_TRUE(matchesWithOpenMP51(Source4, Matcher));
2432 
2433   StringRef Source5 = R"(
2434 void x(int x) {
2435 #pragma omp parallel num_threads(x)
2436 ;
2437 })";
2438   EXPECT_TRUE(notMatchesWithOpenMP(Source5, Matcher));
2439 }
2440 
TEST(ASTMatchersTest,Finder_DynamicOnlyAcceptsSomeMatchers)2441 TEST(ASTMatchersTest, Finder_DynamicOnlyAcceptsSomeMatchers) {
2442   MatchFinder Finder;
2443   EXPECT_TRUE(Finder.addDynamicMatcher(decl(), nullptr));
2444   EXPECT_TRUE(Finder.addDynamicMatcher(callExpr(), nullptr));
2445   EXPECT_TRUE(
2446       Finder.addDynamicMatcher(constantArrayType(hasSize(42)), nullptr));
2447 
2448   // Do not accept non-toplevel matchers.
2449   EXPECT_FALSE(Finder.addDynamicMatcher(isMain(), nullptr));
2450   EXPECT_FALSE(Finder.addDynamicMatcher(hasName("x"), nullptr));
2451 }
2452 
TEST(MatchFinderAPI,MatchesDynamic)2453 TEST(MatchFinderAPI, MatchesDynamic) {
2454   StringRef SourceCode = "struct A { void f() {} };";
2455   auto Matcher = functionDecl(isDefinition()).bind("method");
2456 
2457   auto astUnit = tooling::buildASTFromCode(SourceCode);
2458 
2459   auto GlobalBoundNodes = matchDynamic(Matcher, astUnit->getASTContext());
2460 
2461   EXPECT_EQ(GlobalBoundNodes.size(), 1u);
2462   EXPECT_EQ(GlobalBoundNodes[0].getMap().size(), 1u);
2463 
2464   auto GlobalMethodNode = GlobalBoundNodes[0].getNodeAs<FunctionDecl>("method");
2465   EXPECT_TRUE(GlobalMethodNode != nullptr);
2466 
2467   auto MethodBoundNodes =
2468       matchDynamic(Matcher, *GlobalMethodNode, astUnit->getASTContext());
2469   EXPECT_EQ(MethodBoundNodes.size(), 1u);
2470   EXPECT_EQ(MethodBoundNodes[0].getMap().size(), 1u);
2471 
2472   auto MethodNode = MethodBoundNodes[0].getNodeAs<FunctionDecl>("method");
2473   EXPECT_EQ(MethodNode, GlobalMethodNode);
2474 }
2475 
allTestClangConfigs()2476 static std::vector<TestClangConfig> allTestClangConfigs() {
2477   std::vector<TestClangConfig> all_configs;
2478   for (TestLanguage lang : {Lang_C89, Lang_C99, Lang_CXX03, Lang_CXX11,
2479                             Lang_CXX14, Lang_CXX17, Lang_CXX20}) {
2480     TestClangConfig config;
2481     config.Language = lang;
2482 
2483     // Use an unknown-unknown triple so we don't instantiate the full system
2484     // toolchain.  On Linux, instantiating the toolchain involves stat'ing
2485     // large portions of /usr/lib, and this slows down not only this test, but
2486     // all other tests, via contention in the kernel.
2487     //
2488     // FIXME: This is a hack to work around the fact that there's no way to do
2489     // the equivalent of runToolOnCodeWithArgs without instantiating a full
2490     // Driver.  We should consider having a function, at least for tests, that
2491     // invokes cc1.
2492     config.Target = "i386-unknown-unknown";
2493     all_configs.push_back(config);
2494 
2495     // Windows target is interesting to test because it enables
2496     // `-fdelayed-template-parsing`.
2497     config.Target = "x86_64-pc-win32-msvc";
2498     all_configs.push_back(config);
2499   }
2500   return all_configs;
2501 }
2502 
2503 INSTANTIATE_TEST_SUITE_P(ASTMatchersTests, ASTMatchersTest,
2504                          testing::ValuesIn(allTestClangConfigs()));
2505 
2506 } // namespace ast_matchers
2507 } // namespace clang
2508