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 IntrusiveRefCntPtr<DiagnosticsEngine> Diags =
61 CompilerInstance::createDiagnostics(Opts: new DiagnosticOptions());
62 CreateInvocationOptions CIOpts;
63 CIOpts.Diags = Diags;
64 CIOpts.VFS = llvm::vfs::createPhysicalFileSystem();
65
66 std::string CacheBMIPath =
67 llvm::Twine(TestDir + "/" + ModuleName + " .pcm").str();
68 std::string PrebuiltModulePath =
69 "-fprebuilt-module-path=" + TestDir.str().str();
70 const char *Args[] = {"clang++",
71 "-std=c++20",
72 "--precompile",
73 PrebuiltModulePath.c_str(),
74 "-working-directory",
75 TestDir.c_str(),
76 "-I",
77 TestDir.c_str(),
78 FileName.c_str(),
79 "-o",
80 CacheBMIPath.c_str()};
81 std::shared_ptr<CompilerInvocation> Invocation =
82 createInvocation(Args, Opts: CIOpts);
83 EXPECT_TRUE(Invocation);
84
85 CompilerInstance Instance;
86 Instance.setDiagnostics(Diags.get());
87 Instance.setInvocation(Invocation);
88 GenerateModuleInterfaceAction Action;
89 EXPECT_TRUE(Instance.ExecuteAction(Action));
90 EXPECT_FALSE(Diags->hasErrorOccurred());
91
92 return CacheBMIPath;
93 }
94};
95
96struct TrivialVisibleDeclConsumer : public VisibleDeclConsumer {
97 TrivialVisibleDeclConsumer() {}
98 void EnteredContext(DeclContext *Ctx) override {}
99 void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
100 bool InBaseClass) override {
101 FoundNum++;
102 }
103
104 int FoundNum = 0;
105};
106
107class NoloadLookupConsumer : public SemaConsumer {
108public:
109 void InitializeSema(Sema &S) override { SemaPtr = &S; }
110
111 bool HandleTopLevelDecl(DeclGroupRef D) override {
112 if (!D.isSingleDecl())
113 return true;
114
115 Decl *TD = D.getSingleDecl();
116
117 auto *ID = dyn_cast<ImportDecl>(Val: TD);
118 if (!ID)
119 return true;
120
121 clang::Module *M = ID->getImportedModule();
122 assert(M);
123 if (M->Name != "R")
124 return true;
125
126 auto *Std = SemaPtr->getStdNamespace();
127 EXPECT_TRUE(Std);
128 TrivialVisibleDeclConsumer Consumer;
129 SemaPtr->LookupVisibleDecls(Std, Sema::LookupNameKind::LookupOrdinaryName,
130 Consumer,
131 /*IncludeGlobalScope=*/true,
132 /*IncludeDependentBases=*/false,
133 /*LoadExternal=*/false);
134 EXPECT_EQ(Consumer.FoundNum, 1);
135 return true;
136 }
137
138private:
139 Sema *SemaPtr = nullptr;
140};
141
142class NoloadLookupAction : public ASTFrontendAction {
143 std::unique_ptr<ASTConsumer>
144 CreateASTConsumer(CompilerInstance &CI, StringRef /*Unused*/) override {
145 return std::make_unique<NoloadLookupConsumer>();
146 }
147};
148
149TEST_F(NoloadLookupTest, NonModulesTest) {
150 GenerateModuleInterface(ModuleName: "M", Contents: R"cpp(
151module;
152namespace std {
153 int What();
154
155 void bar(int x = What()) {
156 }
157}
158export module M;
159export using std::bar;
160 )cpp");
161
162 GenerateModuleInterface(ModuleName: "R", Contents: R"cpp(
163module;
164namespace std {
165 class Another;
166 int What(Another);
167 int What();
168}
169export module R;
170 )cpp");
171
172 const char *test_file_contents = R"cpp(
173import M;
174namespace std {
175 void use() {
176 bar();
177 }
178}
179import R;
180 )cpp";
181 std::string DepArg = "-fprebuilt-module-path=" + TestDir.str().str();
182 EXPECT_TRUE(runToolOnCodeWithArgs(std::make_unique<NoloadLookupAction>(),
183 test_file_contents,
184 {
185 "-std=c++20",
186 DepArg.c_str(),
187 "-I",
188 TestDir.c_str(),
189 },
190 "test.cpp"));
191}
192
193} // namespace
194

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