1 //===- unittest/Tooling/RecursiveASTVisitorTests/TraversalScope.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 class Visitor : public ExpectedLocationVisitor<Visitor, clang::TestVisitor> {
16 public:
Visitor(ASTContext * Context)17 Visitor(ASTContext *Context) { this->Context = Context; }
18
VisitNamedDecl(NamedDecl * D)19 bool VisitNamedDecl(NamedDecl *D) {
20 if (!D->isImplicit())
21 Match(D->getName(), D->getLocation());
22 return true;
23 }
24 };
25
TEST(RecursiveASTVisitor,RespectsTraversalScope)26 TEST(RecursiveASTVisitor, RespectsTraversalScope) {
27 auto AST = tooling::buildASTFromCode(
28 R"cpp(
29 struct foo {
30 struct bar {
31 struct baz {};
32 };
33 };
34 )cpp",
35 "foo.cpp", std::make_shared<PCHContainerOperations>());
36 auto &Ctx = AST->getASTContext();
37 auto &TU = *Ctx.getTranslationUnitDecl();
38 auto &Foo = *TU.lookup(&Ctx.Idents.get("foo")).front();
39 auto &Bar = *cast<DeclContext>(Foo).lookup(&Ctx.Idents.get("bar")).front();
40
41 Ctx.setTraversalScope({&Bar});
42
43 Visitor V(&Ctx);
44 V.DisallowMatch("foo", 2, 8);
45 V.ExpectMatch("bar", 3, 10);
46 V.ExpectMatch("baz", 4, 12);
47 V.TraverseAST(Ctx);
48 }
49
50 } // end anonymous namespace
51