1 //===- unittests/Lex/PPConditionalDirectiveRecordTest.cpp-PP directive 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/Lex/PPConditionalDirectiveRecord.h"
10 #include "clang/Basic/Diagnostic.h"
11 #include "clang/Basic/DiagnosticOptions.h"
12 #include "clang/Basic/FileManager.h"
13 #include "clang/Basic/LangOptions.h"
14 #include "clang/Basic/SourceManager.h"
15 #include "clang/Basic/TargetInfo.h"
16 #include "clang/Basic/TargetOptions.h"
17 #include "clang/Lex/HeaderSearch.h"
18 #include "clang/Lex/HeaderSearchOptions.h"
19 #include "clang/Lex/ModuleLoader.h"
20 #include "clang/Lex/Preprocessor.h"
21 #include "clang/Lex/PreprocessorOptions.h"
22 #include "gtest/gtest.h"
23
24 using namespace clang;
25
26 namespace {
27
28 // The test fixture.
29 class PPConditionalDirectiveRecordTest : public ::testing::Test {
30 protected:
PPConditionalDirectiveRecordTest()31 PPConditionalDirectiveRecordTest()
32 : FileMgr(FileMgrOpts),
33 DiagID(new DiagnosticIDs()),
34 Diags(DiagID, new DiagnosticOptions, new IgnoringDiagConsumer()),
35 SourceMgr(Diags, FileMgr),
36 TargetOpts(new TargetOptions)
37 {
38 TargetOpts->Triple = "x86_64-apple-darwin11.1.0";
39 Target = TargetInfo::CreateTargetInfo(Diags, TargetOpts);
40 }
41
42 FileSystemOptions FileMgrOpts;
43 FileManager FileMgr;
44 IntrusiveRefCntPtr<DiagnosticIDs> DiagID;
45 DiagnosticsEngine Diags;
46 SourceManager SourceMgr;
47 LangOptions LangOpts;
48 std::shared_ptr<TargetOptions> TargetOpts;
49 IntrusiveRefCntPtr<TargetInfo> Target;
50 };
51
TEST_F(PPConditionalDirectiveRecordTest,PPRecAPI)52 TEST_F(PPConditionalDirectiveRecordTest, PPRecAPI) {
53 const char *source =
54 "0 1\n"
55 "#if 1\n"
56 "2\n"
57 "#ifndef BB\n"
58 "3 4\n"
59 "#else\n"
60 "#endif\n"
61 "5\n"
62 "#endif\n"
63 "6\n"
64 "#if 1\n"
65 "7\n"
66 "#if 1\n"
67 "#endif\n"
68 "8\n"
69 "#endif\n"
70 "9\n";
71
72 std::unique_ptr<llvm::MemoryBuffer> Buf =
73 llvm::MemoryBuffer::getMemBuffer(source);
74 SourceMgr.setMainFileID(SourceMgr.createFileID(std::move(Buf)));
75
76 TrivialModuleLoader ModLoader;
77 HeaderSearch HeaderInfo(std::make_shared<HeaderSearchOptions>(), SourceMgr,
78 Diags, LangOpts, Target.get());
79 Preprocessor PP(std::make_shared<PreprocessorOptions>(), Diags, LangOpts,
80 SourceMgr, HeaderInfo, ModLoader,
81 /*IILookup =*/nullptr,
82 /*OwnsHeaderSearch =*/false);
83 PP.Initialize(*Target);
84 PPConditionalDirectiveRecord *
85 PPRec = new PPConditionalDirectiveRecord(SourceMgr);
86 PP.addPPCallbacks(std::unique_ptr<PPCallbacks>(PPRec));
87 PP.EnterMainSourceFile();
88
89 std::vector<Token> toks;
90 while (1) {
91 Token tok;
92 PP.Lex(tok);
93 if (tok.is(tok::eof))
94 break;
95 toks.push_back(tok);
96 }
97
98 // Make sure we got the tokens that we expected.
99 ASSERT_EQ(10U, toks.size());
100
101 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
102 SourceRange(toks[0].getLocation(), toks[1].getLocation())));
103 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
104 SourceRange(toks[0].getLocation(), toks[2].getLocation())));
105 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
106 SourceRange(toks[3].getLocation(), toks[4].getLocation())));
107 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
108 SourceRange(toks[1].getLocation(), toks[5].getLocation())));
109 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
110 SourceRange(toks[2].getLocation(), toks[6].getLocation())));
111 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
112 SourceRange(toks[2].getLocation(), toks[5].getLocation())));
113 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
114 SourceRange(toks[0].getLocation(), toks[6].getLocation())));
115 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
116 SourceRange(toks[2].getLocation(), toks[8].getLocation())));
117 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
118 SourceRange(toks[0].getLocation(), toks[9].getLocation())));
119
120 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
121 toks[0].getLocation(), toks[2].getLocation()));
122 EXPECT_FALSE(PPRec->areInDifferentConditionalDirectiveRegion(
123 toks[3].getLocation(), toks[4].getLocation()));
124 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
125 toks[1].getLocation(), toks[5].getLocation()));
126 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
127 toks[2].getLocation(), toks[0].getLocation()));
128 EXPECT_FALSE(PPRec->areInDifferentConditionalDirectiveRegion(
129 toks[4].getLocation(), toks[3].getLocation()));
130 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
131 toks[5].getLocation(), toks[1].getLocation()));
132 }
133
134 } // anonymous namespace
135