1 //===- unittests/Frontend/UtilsTest.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/Frontend/Utils.h" 10 #include "clang/Basic/Diagnostic.h" 11 #include "clang/Basic/TargetOptions.h" 12 #include "clang/Frontend/CompilerInstance.h" 13 #include "clang/Frontend/CompilerInvocation.h" 14 #include "clang/Lex/PreprocessorOptions.h" 15 #include "llvm/ADT/IntrusiveRefCntPtr.h" 16 #include "llvm/Support/VirtualFileSystem.h" 17 #include "gmock/gmock.h" 18 #include "gtest/gtest.h" 19 20 namespace clang { 21 namespace { 22 using testing::ElementsAre; 23 24 TEST(BuildCompilerInvocationTest, RecoverMultipleJobs) { 25 // This generates multiple jobs and we recover by using the first. 26 std::vector<const char *> Args = {"clang", "--target=macho", "-arch", "i386", 27 "-arch", "x86_64", "foo.cpp"}; 28 clang::IgnoringDiagConsumer D; 29 CreateInvocationOptions Opts; 30 Opts.RecoverOnError = true; 31 Opts.VFS = new llvm::vfs::InMemoryFileSystem(); 32 Opts.Diags = clang::CompilerInstance::createDiagnostics( 33 *Opts.VFS, new DiagnosticOptions, &D, false); 34 std::unique_ptr<CompilerInvocation> CI = createInvocation(Args, Opts); 35 ASSERT_TRUE(CI); 36 EXPECT_THAT(CI->TargetOpts->Triple, testing::StartsWith("i386-")); 37 } 38 39 // buildInvocationFromCommandLine should not translate -include to -include-pch, 40 // even if the PCH file exists. 41 TEST(BuildCompilerInvocationTest, ProbePrecompiled) { 42 std::vector<const char *> Args = {"clang", "-include", "foo.h", "foo.cpp"}; 43 auto FS = llvm::makeIntrusiveRefCnt<llvm::vfs::InMemoryFileSystem>(); 44 FS->addFile("foo.h", 0, llvm::MemoryBuffer::getMemBuffer("")); 45 FS->addFile("foo.h.pch", 0, llvm::MemoryBuffer::getMemBuffer("")); 46 47 clang::IgnoringDiagConsumer D; 48 llvm::IntrusiveRefCntPtr<DiagnosticsEngine> CommandLineDiagsEngine = 49 clang::CompilerInstance::createDiagnostics(*FS, new DiagnosticOptions, &D, 50 false); 51 // Default: ProbePrecompiled=false 52 CreateInvocationOptions CIOpts; 53 CIOpts.Diags = CommandLineDiagsEngine; 54 CIOpts.VFS = FS; 55 std::unique_ptr<CompilerInvocation> CI = createInvocation(Args, CIOpts); 56 ASSERT_TRUE(CI); 57 EXPECT_THAT(CI->getPreprocessorOpts().Includes, ElementsAre("foo.h")); 58 EXPECT_EQ(CI->getPreprocessorOpts().ImplicitPCHInclude, ""); 59 60 CIOpts.ProbePrecompiled = true; 61 CI = createInvocation(Args, CIOpts); 62 ASSERT_TRUE(CI); 63 EXPECT_THAT(CI->getPreprocessorOpts().Includes, ElementsAre()); 64 EXPECT_EQ(CI->getPreprocessorOpts().ImplicitPCHInclude, "foo.h.pch"); 65 } 66 67 } // namespace 68 } // namespace clang 69