1 //===- unittest/Tooling/RecursiveASTVisitorTests/Attr.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
11 using namespace clang;
12
13 namespace {
14
15 // Check to ensure that attributes and expressions within them are being
16 // visited.
17 class AttrVisitor : public ExpectedLocationVisitor<AttrVisitor> {
18 public:
VisitMemberExpr(MemberExpr * ME)19 bool VisitMemberExpr(MemberExpr *ME) {
20 Match(ME->getMemberDecl()->getNameAsString(), ME->getBeginLoc());
21 return true;
22 }
VisitAttr(Attr * A)23 bool VisitAttr(Attr *A) {
24 Match("Attr", A->getLocation());
25 return true;
26 }
VisitGuardedByAttr(GuardedByAttr * A)27 bool VisitGuardedByAttr(GuardedByAttr *A) {
28 Match("guarded_by", A->getLocation());
29 return true;
30 }
31 };
32
33
TEST(RecursiveASTVisitor,AttributesAreVisited)34 TEST(RecursiveASTVisitor, AttributesAreVisited) {
35 AttrVisitor Visitor;
36 Visitor.ExpectMatch("Attr", 4, 24);
37 Visitor.ExpectMatch("guarded_by", 4, 24);
38 Visitor.ExpectMatch("mu1", 4, 35);
39 Visitor.ExpectMatch("Attr", 5, 29);
40 Visitor.ExpectMatch("mu1", 5, 54);
41 Visitor.ExpectMatch("mu2", 5, 59);
42 EXPECT_TRUE(Visitor.runOver(
43 "class Foo {\n"
44 " int mu1;\n"
45 " int mu2;\n"
46 " int a __attribute__((guarded_by(mu1)));\n"
47 " void bar() __attribute__((exclusive_locks_required(mu1, mu2)));\n"
48 "};\n"));
49 }
50
51 } // end anonymous namespace
52