135 lines
4.9 KiB
C++
135 lines
4.9 KiB
C++
|
//===- unittests/Lex/PPConditionalDirectiveRecordTest.cpp-PP directive 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/Lex/PPConditionalDirectiveRecord.h"
|
||
|
|
#include "clang/Basic/Diagnostic.h"
|
||
|
|
#include "clang/Basic/DiagnosticOptions.h"
|
||
|
|
#include "clang/Basic/FileManager.h"
|
||
|
|
#include "clang/Basic/LangOptions.h"
|
||
|
|
#include "clang/Basic/SourceManager.h"
|
||
|
|
#include "clang/Basic/TargetInfo.h"
|
||
|
|
#include "clang/Basic/TargetOptions.h"
|
||
|
|
#include "clang/Lex/HeaderSearch.h"
|
||
|
|
#include "clang/Lex/HeaderSearchOptions.h"
|
||
|
|
#include "clang/Lex/ModuleLoader.h"
|
||
|
|
#include "clang/Lex/Preprocessor.h"
|
||
|
|
#include "clang/Lex/PreprocessorOptions.h"
|
||
|
|
#include "gtest/gtest.h"
|
||
|
|
|
||
|
|
using namespace clang;
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
// The test fixture.
|
||
|
|
class PPConditionalDirectiveRecordTest : public ::testing::Test {
|
||
|
|
protected:
|
||
|
|
PPConditionalDirectiveRecordTest()
|
||
|
|
: FileMgr(FileMgrOpts),
|
||
|
|
DiagID(new DiagnosticIDs()),
|
||
|
|
Diags(DiagID, new DiagnosticOptions, new IgnoringDiagConsumer()),
|
||
|
|
SourceMgr(Diags, FileMgr),
|
||
|
|
TargetOpts(new TargetOptions)
|
||
|
|
{
|
||
|
|
TargetOpts->Triple = "x86_64-apple-darwin11.1.0";
|
||
|
|
Target = TargetInfo::CreateTargetInfo(Diags, TargetOpts);
|
||
|
|
}
|
||
|
|
|
||
|
|
FileSystemOptions FileMgrOpts;
|
||
|
|
FileManager FileMgr;
|
||
|
|
IntrusiveRefCntPtr<DiagnosticIDs> DiagID;
|
||
|
|
DiagnosticsEngine Diags;
|
||
|
|
SourceManager SourceMgr;
|
||
|
|
LangOptions LangOpts;
|
||
|
|
std::shared_ptr<TargetOptions> TargetOpts;
|
||
|
|
IntrusiveRefCntPtr<TargetInfo> Target;
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_F(PPConditionalDirectiveRecordTest, PPRecAPI) {
|
||
|
|
const char *source =
|
||
|
|
"0 1\n"
|
||
|
|
"#if 1\n"
|
||
|
|
"2\n"
|
||
|
|
"#ifndef BB\n"
|
||
|
|
"3 4\n"
|
||
|
|
"#else\n"
|
||
|
|
"#endif\n"
|
||
|
|
"5\n"
|
||
|
|
"#endif\n"
|
||
|
|
"6\n"
|
||
|
|
"#if 1\n"
|
||
|
|
"7\n"
|
||
|
|
"#if 1\n"
|
||
|
|
"#endif\n"
|
||
|
|
"8\n"
|
||
|
|
"#endif\n"
|
||
|
|
"9\n";
|
||
|
|
|
||
|
|
std::unique_ptr<llvm::MemoryBuffer> Buf =
|
||
|
|
llvm::MemoryBuffer::getMemBuffer(source);
|
||
|
|
SourceMgr.setMainFileID(SourceMgr.createFileID(std::move(Buf)));
|
||
|
|
|
||
|
|
TrivialModuleLoader ModLoader;
|
||
|
|
HeaderSearch HeaderInfo(std::make_shared<HeaderSearchOptions>(), SourceMgr,
|
||
|
|
Diags, LangOpts, Target.get());
|
||
|
|
Preprocessor PP(std::make_shared<PreprocessorOptions>(), Diags, LangOpts,
|
||
|
|
SourceMgr, HeaderInfo, ModLoader,
|
||
|
|
/*IILookup =*/nullptr,
|
||
|
|
/*OwnsHeaderSearch =*/false);
|
||
|
|
PP.Initialize(*Target);
|
||
|
|
PPConditionalDirectiveRecord *
|
||
|
|
PPRec = new PPConditionalDirectiveRecord(SourceMgr);
|
||
|
|
PP.addPPCallbacks(std::unique_ptr<PPCallbacks>(PPRec));
|
||
|
|
PP.EnterMainSourceFile();
|
||
|
|
|
||
|
|
std::vector<Token> toks;
|
||
|
|
while (1) {
|
||
|
|
Token tok;
|
||
|
|
PP.Lex(tok);
|
||
|
|
if (tok.is(tok::eof))
|
||
|
|
break;
|
||
|
|
toks.push_back(tok);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure we got the tokens that we expected.
|
||
|
|
ASSERT_EQ(10U, toks.size());
|
||
|
|
|
||
|
|
EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[0].getLocation(), toks[1].getLocation())));
|
||
|
|
EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[0].getLocation(), toks[2].getLocation())));
|
||
|
|
EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[3].getLocation(), toks[4].getLocation())));
|
||
|
|
EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[1].getLocation(), toks[5].getLocation())));
|
||
|
|
EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[2].getLocation(), toks[6].getLocation())));
|
||
|
|
EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[2].getLocation(), toks[5].getLocation())));
|
||
|
|
EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[0].getLocation(), toks[6].getLocation())));
|
||
|
|
EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[2].getLocation(), toks[8].getLocation())));
|
||
|
|
EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
|
||
|
|
SourceRange(toks[0].getLocation(), toks[9].getLocation())));
|
||
|
|
|
||
|
|
EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
|
||
|
|
toks[0].getLocation(), toks[2].getLocation()));
|
||
|
|
EXPECT_FALSE(PPRec->areInDifferentConditionalDirectiveRegion(
|
||
|
|
toks[3].getLocation(), toks[4].getLocation()));
|
||
|
|
EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
|
||
|
|
toks[1].getLocation(), toks[5].getLocation()));
|
||
|
|
EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
|
||
|
|
toks[2].getLocation(), toks[0].getLocation()));
|
||
|
|
EXPECT_FALSE(PPRec->areInDifferentConditionalDirectiveRegion(
|
||
|
|
toks[4].getLocation(), toks[3].getLocation()));
|
||
|
|
EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
|
||
|
|
toks[5].getLocation(), toks[1].getLocation()));
|
||
|
|
}
|
||
|
|
|
||
|
|
} // anonymous namespace
|