1 //=------ unittest/Tooling/RecursiveASTVisitorTests/CXXMethodDecl.cpp ------=//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "TestVisitor.h"
10 #include "clang/AST/Expr.h"
11 
12 using namespace clang;
13 
14 namespace {
15 
16 class CXXMethodDeclVisitor
17     : public ExpectedLocationVisitor<CXXMethodDeclVisitor> {
18 public:
19   CXXMethodDeclVisitor(bool VisitImplicitCode)
20       : VisitImplicitCode(VisitImplicitCode) {}
21 
22   bool shouldVisitImplicitCode() const { return VisitImplicitCode; }
23 
24   bool VisitDeclRefExpr(DeclRefExpr *D) {
25     Match("declref", D->getLocation());
26     return true;
27   }
28   bool VisitParmVarDecl(ParmVarDecl *P) {
29     Match("parm", P->getLocation());
30     return true;
31   }
32 
33 private:
34   bool VisitImplicitCode;
35 };
36 
37 TEST(RecursiveASTVisitor, CXXMethodDeclNoDefaultBodyVisited) {
38   for (bool VisitImplCode : {false, true}) {
39     CXXMethodDeclVisitor Visitor(VisitImplCode);
40     if (VisitImplCode)
41       Visitor.ExpectMatch("declref", 8, 28);
42     else
43       Visitor.DisallowMatch("declref", 8, 28);
44 
45     Visitor.ExpectMatch("parm", 8, 27);
46     llvm::StringRef Code = R"cpp(
47       struct B {};
48       struct A {
49         B BB;
50         A &operator=(A &&O);
51       };
52 
53       A &A::operator=(A &&O) = default;
54     )cpp";
55     EXPECT_TRUE(Visitor.runOver(Code, CXXMethodDeclVisitor::Lang_CXX11));
56   }
57 }
58 
59 TEST(RecursiveASTVisitor, FunctionDeclNoDefaultBodyVisited) {
60   for (bool VisitImplCode : {false, true}) {
61     CXXMethodDeclVisitor Visitor(VisitImplCode);
62     if (VisitImplCode)
63       Visitor.ExpectMatch("declref", 4, 58, /*Times=*/2);
64     else
65       Visitor.DisallowMatch("declref", 4, 58);
66     llvm::StringRef Code = R"cpp(
67       struct s {
68         int x;
69         friend auto operator==(s a, s b) -> bool = default;
70       };
71       bool k = s() == s(); // make sure clang generates the "==" definition.
72     )cpp";
73     EXPECT_TRUE(Visitor.runOver(Code, CXXMethodDeclVisitor::Lang_CXX2a));
74   }
75 }
76 } // end anonymous namespace
77