bdd10d9d24
Starting with 41e3919ded78d8870f7c95e9181c7f7e29aa3cc4 DiagnosticsEngine creation might perform IO. It was implicitly defaulting to getRealFileSystem. This patch makes it explicit by pushing the decision making to callers. It uses ambient VFS if one is available, and keeps using `getRealFileSystem` if there aren't any VFS.
116 lines
4.4 KiB
C++
116 lines
4.4 KiB
C++
//===- unittests/Frontend/OutputStreamTest.cpp --- FrontendAction tests --===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Basic/LangStandard.h"
|
|
#include "clang/CodeGen/BackendUtil.h"
|
|
#include "clang/CodeGen/CodeGenAction.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
#include "clang/FrontendTool/Utils.h"
|
|
#include "clang/Lex/PreprocessorOptions.h"
|
|
#include "llvm/Support/VirtualFileSystem.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
using namespace clang;
|
|
using namespace clang::frontend;
|
|
|
|
namespace {
|
|
|
|
TEST(FrontendOutputTests, TestOutputStream) {
|
|
auto Invocation = std::make_shared<CompilerInvocation>();
|
|
Invocation->getPreprocessorOpts().addRemappedFile(
|
|
"test.cc", MemoryBuffer::getMemBuffer("").release());
|
|
Invocation->getFrontendOpts().Inputs.push_back(
|
|
FrontendInputFile("test.cc", Language::CXX));
|
|
Invocation->getFrontendOpts().ProgramAction = EmitBC;
|
|
Invocation->getTargetOpts().Triple = "i386-unknown-linux-gnu";
|
|
CompilerInstance Compiler;
|
|
|
|
SmallVector<char, 256> IRBuffer;
|
|
std::unique_ptr<raw_pwrite_stream> IRStream(
|
|
new raw_svector_ostream(IRBuffer));
|
|
|
|
Compiler.setOutputStream(std::move(IRStream));
|
|
Compiler.setInvocation(std::move(Invocation));
|
|
Compiler.createDiagnostics(*llvm::vfs::getRealFileSystem());
|
|
|
|
bool Success = ExecuteCompilerInvocation(&Compiler);
|
|
EXPECT_TRUE(Success);
|
|
EXPECT_TRUE(!IRBuffer.empty());
|
|
EXPECT_TRUE(StringRef(IRBuffer.data()).starts_with("BC"));
|
|
}
|
|
|
|
TEST(FrontendOutputTests, TestVerboseOutputStreamShared) {
|
|
auto Invocation = std::make_shared<CompilerInvocation>();
|
|
Invocation->getPreprocessorOpts().addRemappedFile(
|
|
"test.cc", MemoryBuffer::getMemBuffer("invalid").release());
|
|
Invocation->getFrontendOpts().Inputs.push_back(
|
|
FrontendInputFile("test.cc", Language::CXX));
|
|
Invocation->getFrontendOpts().ProgramAction = EmitBC;
|
|
Invocation->getTargetOpts().Triple = "i386-unknown-linux-gnu";
|
|
CompilerInstance Compiler;
|
|
|
|
std::string VerboseBuffer;
|
|
raw_string_ostream VerboseStream(VerboseBuffer);
|
|
|
|
Compiler.setOutputStream(std::make_unique<raw_null_ostream>());
|
|
Compiler.setInvocation(std::move(Invocation));
|
|
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
Compiler.createDiagnostics(
|
|
*llvm::vfs::getRealFileSystem(),
|
|
new TextDiagnosticPrinter(llvm::nulls(), &*DiagOpts), true);
|
|
Compiler.setVerboseOutputStream(VerboseStream);
|
|
|
|
bool Success = ExecuteCompilerInvocation(&Compiler);
|
|
EXPECT_FALSE(Success);
|
|
EXPECT_TRUE(!VerboseBuffer.empty());
|
|
EXPECT_TRUE(StringRef(VerboseBuffer.data()).contains("errors generated"));
|
|
}
|
|
|
|
TEST(FrontendOutputTests, TestVerboseOutputStreamOwned) {
|
|
std::string VerboseBuffer;
|
|
bool Success;
|
|
{
|
|
auto Invocation = std::make_shared<CompilerInvocation>();
|
|
Invocation->getPreprocessorOpts().addRemappedFile(
|
|
"test.cc", MemoryBuffer::getMemBuffer("invalid").release());
|
|
Invocation->getFrontendOpts().Inputs.push_back(
|
|
FrontendInputFile("test.cc", Language::CXX));
|
|
Invocation->getFrontendOpts().ProgramAction = EmitBC;
|
|
Invocation->getTargetOpts().Triple = "i386-unknown-linux-gnu";
|
|
CompilerInstance Compiler;
|
|
|
|
std::unique_ptr<raw_ostream> VerboseStream =
|
|
std::make_unique<raw_string_ostream>(VerboseBuffer);
|
|
|
|
Compiler.setOutputStream(std::make_unique<raw_null_ostream>());
|
|
Compiler.setInvocation(std::move(Invocation));
|
|
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
Compiler.createDiagnostics(
|
|
*llvm::vfs::getRealFileSystem(),
|
|
new TextDiagnosticPrinter(llvm::nulls(), &*DiagOpts), true);
|
|
Compiler.setVerboseOutputStream(std::move(VerboseStream));
|
|
|
|
Success = ExecuteCompilerInvocation(&Compiler);
|
|
}
|
|
EXPECT_FALSE(Success);
|
|
EXPECT_TRUE(!VerboseBuffer.empty());
|
|
EXPECT_TRUE(StringRef(VerboseBuffer.data()).contains("errors generated"));
|
|
}
|
|
|
|
TEST(FrontendOutputTests, TestVerboseOutputStreamOwnedNotLeaked) {
|
|
CompilerInstance Compiler;
|
|
Compiler.setVerboseOutputStream(std::make_unique<raw_null_ostream>());
|
|
|
|
// Trust leak sanitizer bots to catch a leak here.
|
|
Compiler.setVerboseOutputStream(llvm::nulls());
|
|
}
|
|
|
|
} // anonymous namespace
|