1//== unittests/Sema/SemaNoloadLookupTest.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 "clang/AST/DeclLookups.h"
10#include "clang/AST/DeclarationName.h"
11#include "clang/ASTMatchers/ASTMatchFinder.h"
12#include "clang/ASTMatchers/ASTMatchers.h"
13#include "clang/Frontend/CompilerInstance.h"
14#include "clang/Frontend/FrontendAction.h"
15#include "clang/Frontend/FrontendActions.h"
16#include "clang/Parse/ParseAST.h"
17#include "clang/Sema/Lookup.h"
18#include "clang/Sema/Sema.h"
19#include "clang/Sema/SemaConsumer.h"
20#include "clang/Tooling/Tooling.h"
21#include "gtest/gtest.h"
22
23using namespace llvm;
24using namespace clang;
25using namespace clang::tooling;
26
27namespace {
28
29class NoloadLookupTest : public ::testing::Test {
30 void SetUp() override {
31 ASSERT_FALSE(
32 sys::fs::createUniqueDirectory("modules-no-comments-test", TestDir));
33 }
34
35 void TearDown() override { sys::fs::remove_directories(path: TestDir); }
36
37public:
38 SmallString<256> TestDir;
39
40 void addFile(StringRef Path, StringRef Contents) {
41 ASSERT_FALSE(sys::path::is_absolute(Path));
42
43 SmallString<256> AbsPath(TestDir);
44 sys::path::append(path&: AbsPath, a: Path);
45
46 ASSERT_FALSE(
47 sys::fs::create_directories(llvm::sys::path::parent_path(AbsPath)));
48
49 std::error_code EC;
50 llvm::raw_fd_ostream OS(AbsPath, EC);
51 ASSERT_FALSE(EC);
52 OS << Contents;
53 }
54
55 std::string GenerateModuleInterface(StringRef ModuleName,
56 StringRef Contents) {
57 std::string FileName = llvm::Twine(ModuleName + ".cppm").str();
58 addFile(Path: FileName, Contents);
59
60 CreateInvocationOptions CIOpts;
61 CIOpts.VFS = llvm::vfs::createPhysicalFileSystem();
62 DiagnosticOptions DiagOpts;
63 IntrusiveRefCntPtr<DiagnosticsEngine> Diags =
64 CompilerInstance::createDiagnostics(VFS&: *CIOpts.VFS, Opts&: DiagOpts);
65 CIOpts.Diags = Diags;
66
67 std::string CacheBMIPath =
68 llvm::Twine(TestDir + "/" + ModuleName + ".pcm").str();
69 std::string PrebuiltModulePath =
70 "-fprebuilt-module-path=" + TestDir.str().str();
71 const char *Args[] = {"clang++",
72 "-std=c++20",
73 "--precompile",
74 PrebuiltModulePath.c_str(),
75 "-working-directory",
76 TestDir.c_str(),
77 "-I",
78 TestDir.c_str(),
79 FileName.c_str()};
80 std::shared_ptr<CompilerInvocation> Invocation =
81 createInvocation(Args, Opts: CIOpts);
82 EXPECT_TRUE(Invocation);
83
84 CompilerInstance Instance(std::move(Invocation));
85 Instance.setDiagnostics(Diags.get());
86 Instance.getFrontendOpts().OutputFile = CacheBMIPath;
87 GenerateReducedModuleInterfaceAction Action;
88 EXPECT_TRUE(Instance.ExecuteAction(Action));
89 EXPECT_FALSE(Diags->hasErrorOccurred());
90
91 return CacheBMIPath;
92 }
93};
94
95struct TrivialVisibleDeclConsumer : public VisibleDeclConsumer {
96 TrivialVisibleDeclConsumer() {}
97 void EnteredContext(DeclContext *Ctx) override {}
98 void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
99 bool InBaseClass) override {
100 FoundNum++;
101 }
102
103 int FoundNum = 0;
104};
105
106class NoloadLookupConsumer : public SemaConsumer {
107public:
108 void InitializeSema(Sema &S) override { SemaPtr = &S; }
109
110 bool HandleTopLevelDecl(DeclGroupRef D) override {
111 if (!D.isSingleDecl())
112 return true;
113
114 Decl *TD = D.getSingleDecl();
115
116 auto *ID = dyn_cast<ImportDecl>(Val: TD);
117 if (!ID)
118 return true;
119
120 clang::Module *M = ID->getImportedModule();
121 assert(M);
122 if (M->Name != "R")
123 return true;
124
125 auto *Std = SemaPtr->getStdNamespace();
126 EXPECT_TRUE(Std);
127 TrivialVisibleDeclConsumer Consumer;
128 SemaPtr->LookupVisibleDecls(Std, Sema::LookupNameKind::LookupOrdinaryName,
129 Consumer,
130 /*IncludeGlobalScope=*/true,
131 /*IncludeDependentBases=*/false,
132 /*LoadExternal=*/false);
133 EXPECT_EQ(Consumer.FoundNum, 1);
134 return true;
135 }
136
137private:
138 Sema *SemaPtr = nullptr;
139};
140
141class NoloadLookupAction : public ASTFrontendAction {
142 std::unique_ptr<ASTConsumer>
143 CreateASTConsumer(CompilerInstance &CI, StringRef /*Unused*/) override {
144 return std::make_unique<NoloadLookupConsumer>();
145 }
146};
147
148TEST_F(NoloadLookupTest, NonModulesTest) {
149 GenerateModuleInterface(ModuleName: "M", Contents: R"cpp(
150module;
151namespace std {
152 int What();
153
154 void bar(int x = What()) {
155 }
156}
157export module M;
158export using std::bar;
159 )cpp");
160
161 GenerateModuleInterface(ModuleName: "R", Contents: R"cpp(
162module;
163namespace std {
164 class Another;
165 int What(Another);
166 int What();
167}
168export module R;
169 )cpp");
170
171 const char *test_file_contents = R"cpp(
172import M;
173namespace std {
174 void use() {
175 bar();
176 }
177}
178import R;
179 )cpp";
180 std::string DepArg = "-fprebuilt-module-path=" + TestDir.str().str();
181 EXPECT_TRUE(runToolOnCodeWithArgs(std::make_unique<NoloadLookupAction>(),
182 test_file_contents,
183 {
184 "-std=c++20",
185 DepArg.c_str(),
186 "-I",
187 TestDir.c_str(),
188 },
189 "test.cpp"));
190}
191
192} // namespace
193

source code of clang/unittests/Sema/SemaNoloadLookupTest.cpp