diff options
| author | 2016-09-03 22:46:54 +0000 | |
|---|---|---|
| committer | 2016-09-03 22:46:54 +0000 | |
| commit | b5500b9ca0102f1ccaf32f0e77e96d0739aded9b (patch) | |
| tree | e1b7ebb5a0231f9e6d8d3f6f719582cebd64dc98 /gnu/llvm/tools/clang/unittests/Tooling/RecursiveASTVisitorTest.cpp | |
| parent | clarify purpose of src/gnu/ directory. (diff) | |
| download | wireguard-openbsd-b5500b9ca0102f1ccaf32f0e77e96d0739aded9b.tar.xz wireguard-openbsd-b5500b9ca0102f1ccaf32f0e77e96d0739aded9b.zip | |
Use the space freed up by sparc and zaurus to import LLVM.
ok hackroom@
Diffstat (limited to 'gnu/llvm/tools/clang/unittests/Tooling/RecursiveASTVisitorTest.cpp')
| -rw-r--r-- | gnu/llvm/tools/clang/unittests/Tooling/RecursiveASTVisitorTest.cpp | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/gnu/llvm/tools/clang/unittests/Tooling/RecursiveASTVisitorTest.cpp b/gnu/llvm/tools/clang/unittests/Tooling/RecursiveASTVisitorTest.cpp new file mode 100644 index 00000000000..c28704532a6 --- /dev/null +++ b/gnu/llvm/tools/clang/unittests/Tooling/RecursiveASTVisitorTest.cpp @@ -0,0 +1,137 @@ +//===- unittest/Tooling/RecursiveASTVisitorTest.cpp -----------------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#include "TestVisitor.h" +#include <stack> + +using namespace clang; + +namespace { + +class LambdaExprVisitor : public ExpectedLocationVisitor<LambdaExprVisitor> { +public: + bool VisitLambdaExpr(LambdaExpr *Lambda) { + PendingBodies.push(Lambda); + Match("", Lambda->getIntroducerRange().getBegin()); + return true; + } + /// For each call to VisitLambdaExpr, we expect a subsequent call (with + /// proper nesting) to TraverseLambdaBody. + bool TraverseLambdaBody(LambdaExpr *Lambda) { + EXPECT_FALSE(PendingBodies.empty()); + EXPECT_EQ(PendingBodies.top(), Lambda); + PendingBodies.pop(); + return TraverseStmt(Lambda->getBody()); + } + /// Determine whether TraverseLambdaBody has been called for every call to + /// VisitLambdaExpr. + bool allBodiesHaveBeenTraversed() const { + return PendingBodies.empty(); + } +private: + std::stack<LambdaExpr *> PendingBodies; +}; + +TEST(RecursiveASTVisitor, VisitsLambdaExpr) { + LambdaExprVisitor Visitor; + Visitor.ExpectMatch("", 1, 12); + EXPECT_TRUE(Visitor.runOver("void f() { []{ return; }(); }", + LambdaExprVisitor::Lang_CXX11)); +} + +TEST(RecursiveASTVisitor, TraverseLambdaBodyCanBeOverridden) { + LambdaExprVisitor Visitor; + EXPECT_TRUE(Visitor.runOver("void f() { []{ return; }(); }", + LambdaExprVisitor::Lang_CXX11)); + EXPECT_TRUE(Visitor.allBodiesHaveBeenTraversed()); +} + +// Matches the (optional) capture-default of a lambda-introducer. +class LambdaDefaultCaptureVisitor + : public ExpectedLocationVisitor<LambdaDefaultCaptureVisitor> { +public: + bool VisitLambdaExpr(LambdaExpr *Lambda) { + if (Lambda->getCaptureDefault() != LCD_None) { + Match("", Lambda->getCaptureDefaultLoc()); + } + return true; + } +}; + +TEST(RecursiveASTVisitor, HasCaptureDefaultLoc) { + LambdaDefaultCaptureVisitor Visitor; + Visitor.ExpectMatch("", 1, 20); + EXPECT_TRUE(Visitor.runOver("void f() { int a; [=]{a;}; }", + LambdaDefaultCaptureVisitor::Lang_CXX11)); +} + +// Checks for lambda classes that are not marked as implicitly-generated. +// (There should be none.) +class ClassVisitor : public ExpectedLocationVisitor<ClassVisitor> { +public: + ClassVisitor() : SawNonImplicitLambdaClass(false) {} + bool VisitCXXRecordDecl(CXXRecordDecl* record) { + if (record->isLambda() && !record->isImplicit()) + SawNonImplicitLambdaClass = true; + return true; + } + + bool sawOnlyImplicitLambdaClasses() const { + return !SawNonImplicitLambdaClass; + } + +private: + bool SawNonImplicitLambdaClass; +}; + +TEST(RecursiveASTVisitor, LambdaClosureTypesAreImplicit) { + ClassVisitor Visitor; + EXPECT_TRUE(Visitor.runOver("auto lambda = []{};", + ClassVisitor::Lang_CXX11)); + EXPECT_TRUE(Visitor.sawOnlyImplicitLambdaClasses()); +} + + +// Check to ensure that attributes and expressions within them are being +// visited. +class AttrVisitor : public ExpectedLocationVisitor<AttrVisitor> { +public: + bool VisitMemberExpr(MemberExpr *ME) { + Match(ME->getMemberDecl()->getNameAsString(), ME->getLocStart()); + return true; + } + bool VisitAttr(Attr *A) { + Match("Attr", A->getLocation()); + return true; + } + bool VisitGuardedByAttr(GuardedByAttr *A) { + Match("guarded_by", A->getLocation()); + return true; + } +}; + + +TEST(RecursiveASTVisitor, AttributesAreVisited) { + AttrVisitor Visitor; + Visitor.ExpectMatch("Attr", 4, 24); + Visitor.ExpectMatch("guarded_by", 4, 24); + Visitor.ExpectMatch("mu1", 4, 35); + Visitor.ExpectMatch("Attr", 5, 29); + Visitor.ExpectMatch("mu1", 5, 54); + Visitor.ExpectMatch("mu2", 5, 59); + EXPECT_TRUE(Visitor.runOver( + "class Foo {\n" + " int mu1;\n" + " int mu2;\n" + " int a __attribute__((guarded_by(mu1)));\n" + " void bar() __attribute__((exclusive_locks_required(mu1, mu2)));\n" + "};\n")); +} + +} // end anonymous namespace |
