1 //===- unittest/AST/ExternalASTSourceTest.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 // This file contains tests for Clang's ExternalASTSource.
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include "clang/AST/ASTConsumer.h"
14 #include "clang/AST/ASTContext.h"
15 #include "clang/AST/ExternalASTSource.h"
16 #include "clang/Frontend/CompilerInstance.h"
17 #include "clang/Frontend/CompilerInvocation.h"
18 #include "clang/Frontend/FrontendActions.h"
19 #include "clang/Lex/PreprocessorOptions.h"
20 #include "gtest/gtest.h"
21
22 using namespace clang;
23 using namespace llvm;
24
25
26 class TestFrontendAction : public ASTFrontendAction {
27 public:
TestFrontendAction(ExternalASTSource * Source)28 TestFrontendAction(ExternalASTSource *Source) : Source(Source) {}
29
30 private:
ExecuteAction()31 void ExecuteAction() override {
32 getCompilerInstance().getASTContext().setExternalSource(Source);
33 getCompilerInstance().getASTContext().getTranslationUnitDecl()
34 ->setHasExternalVisibleStorage();
35 return ASTFrontendAction::ExecuteAction();
36 }
37
CreateASTConsumer(CompilerInstance & CI,StringRef InFile)38 std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI,
39 StringRef InFile) override {
40 return llvm::make_unique<ASTConsumer>();
41 }
42
43 IntrusiveRefCntPtr<ExternalASTSource> Source;
44 };
45
testExternalASTSource(ExternalASTSource * Source,StringRef FileContents)46 bool testExternalASTSource(ExternalASTSource *Source,
47 StringRef FileContents) {
48 CompilerInstance Compiler;
49 Compiler.createDiagnostics();
50
51 auto Invocation = std::make_shared<CompilerInvocation>();
52 Invocation->getPreprocessorOpts().addRemappedFile(
53 "test.cc", MemoryBuffer::getMemBuffer(FileContents).release());
54 const char *Args[] = { "test.cc" };
55 CompilerInvocation::CreateFromArgs(*Invocation, Args,
56 Args + array_lengthof(Args),
57 Compiler.getDiagnostics());
58 Compiler.setInvocation(std::move(Invocation));
59
60 TestFrontendAction Action(Source);
61 return Compiler.ExecuteAction(Action);
62 }
63
64
65 // Ensure that a failed name lookup into an external source only occurs once.
TEST(ExternalASTSourceTest,FailedLookupOccursOnce)66 TEST(ExternalASTSourceTest, FailedLookupOccursOnce) {
67 struct TestSource : ExternalASTSource {
68 TestSource(unsigned &Calls) : Calls(Calls) {}
69
70 bool FindExternalVisibleDeclsByName(const DeclContext *,
71 DeclarationName Name) override {
72 if (Name.getAsString() == "j")
73 ++Calls;
74 return false;
75 }
76
77 unsigned &Calls;
78 };
79
80 unsigned Calls = 0;
81 ASSERT_TRUE(testExternalASTSource(new TestSource(Calls), "int j, k = j;"));
82 EXPECT_EQ(1u, Calls);
83 }
84