| 1 | //===- unittests/Serialization/NoComments.cpp - CI tests -----===// |
| 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/ASTMatchers/ASTMatchFinder.h" |
| 10 | #include "clang/ASTMatchers/ASTMatchers.h" |
| 11 | #include "clang/Basic/FileManager.h" |
| 12 | #include "clang/Frontend/CompilerInstance.h" |
| 13 | #include "clang/Frontend/CompilerInvocation.h" |
| 14 | #include "clang/Frontend/FrontendActions.h" |
| 15 | #include "clang/Frontend/Utils.h" |
| 16 | #include "clang/Lex/HeaderSearch.h" |
| 17 | #include "clang/Tooling/Tooling.h" |
| 18 | #include "llvm/ADT/SmallString.h" |
| 19 | #include "llvm/Support/FileSystem.h" |
| 20 | #include "llvm/Support/raw_ostream.h" |
| 21 | |
| 22 | #include "gtest/gtest.h" |
| 23 | |
| 24 | using namespace llvm; |
| 25 | using namespace clang; |
| 26 | |
| 27 | namespace { |
| 28 | |
| 29 | class : public ::testing::Test { |
| 30 | void () override { |
| 31 | ASSERT_FALSE( |
| 32 | sys::fs::createUniqueDirectory("modules-no-comments-test" , TestDir)); |
| 33 | } |
| 34 | |
| 35 | void () override { sys::fs::remove_directories(path: TestDir); } |
| 36 | |
| 37 | public: |
| 38 | SmallString<256> ; |
| 39 | |
| 40 | void (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 | |
| 56 | TEST_F(NoComments, NonModulesTest) { |
| 57 | std::unique_ptr<ASTUnit> AST = tooling::buildASTFromCodeWithArgs( |
| 58 | Code: R"cpp( |
| 59 | /// Any comments |
| 60 | void foo() {} |
| 61 | )cpp" , |
| 62 | /*Args=*/{"-std=c++20" }); |
| 63 | EXPECT_TRUE(AST); |
| 64 | |
| 65 | ASTContext &Ctx = AST->getASTContext(); |
| 66 | |
| 67 | using namespace clang::ast_matchers; |
| 68 | auto *foo = selectFirst<FunctionDecl>( |
| 69 | BoundTo: "foo" , Results: match(Matcher: functionDecl(hasName(Name: "foo" )).bind(ID: "foo" ), Context&: Ctx)); |
| 70 | EXPECT_TRUE(foo); |
| 71 | |
| 72 | const RawComment *RC = getCompletionComment(Ctx, foo); |
| 73 | EXPECT_TRUE(RC); |
| 74 | EXPECT_TRUE(RC->getRawText(Ctx.getSourceManager()).trim() == |
| 75 | "/// Any comments" ); |
| 76 | } |
| 77 | |
| 78 | TEST_F(NoComments, ModulesTest) { |
| 79 | addFile(Path: "Comments.cppm" , Contents: R"cpp( |
| 80 | export module Comments; |
| 81 | |
| 82 | /// Any comments |
| 83 | void foo() {} |
| 84 | )cpp" ); |
| 85 | |
| 86 | CreateInvocationOptions CIOpts; |
| 87 | CIOpts.VFS = llvm::vfs::createPhysicalFileSystem(); |
| 88 | DiagnosticOptions DiagOpts; |
| 89 | IntrusiveRefCntPtr<DiagnosticsEngine> Diags = |
| 90 | CompilerInstance::createDiagnostics(VFS&: *CIOpts.VFS, Opts&: DiagOpts); |
| 91 | CIOpts.Diags = Diags; |
| 92 | |
| 93 | std::string CacheBMIPath = llvm::Twine(TestDir + "/Comments.pcm" ).str(); |
| 94 | const char *Args[] = {"clang++" , "-std=c++20" , |
| 95 | "--precompile" , "-working-directory" , |
| 96 | TestDir.c_str(), "Comments.cppm" }; |
| 97 | std::shared_ptr<CompilerInvocation> Invocation = |
| 98 | createInvocation(Args, Opts: CIOpts); |
| 99 | ASSERT_TRUE(Invocation); |
| 100 | |
| 101 | CompilerInstance Instance(std::move(Invocation)); |
| 102 | Instance.setDiagnostics(Diags.get()); |
| 103 | Instance.getFrontendOpts().OutputFile = CacheBMIPath; |
| 104 | GenerateReducedModuleInterfaceAction Action; |
| 105 | ASSERT_TRUE(Instance.ExecuteAction(Action)); |
| 106 | ASSERT_FALSE(Diags->hasErrorOccurred()); |
| 107 | |
| 108 | std::string DepArg = |
| 109 | llvm::Twine("-fmodule-file=Comments=" + CacheBMIPath).str(); |
| 110 | std::unique_ptr<ASTUnit> AST = tooling::buildASTFromCodeWithArgs( |
| 111 | Code: R"cpp( |
| 112 | import Comments; |
| 113 | )cpp" , |
| 114 | /*Args=*/{"-std=c++20" , DepArg.c_str()}); |
| 115 | EXPECT_TRUE(AST); |
| 116 | |
| 117 | ASTContext &Ctx = AST->getASTContext(); |
| 118 | |
| 119 | using namespace clang::ast_matchers; |
| 120 | auto *foo = selectFirst<FunctionDecl>( |
| 121 | BoundTo: "foo" , Results: match(Matcher: functionDecl(hasName(Name: "foo" )).bind(ID: "foo" ), Context&: Ctx)); |
| 122 | EXPECT_TRUE(foo); |
| 123 | |
| 124 | const RawComment *RC = getCompletionComment(Ctx, foo); |
| 125 | EXPECT_FALSE(RC); |
| 126 | } |
| 127 | |
| 128 | } // anonymous namespace |
| 129 | |