blob: f7b6f717a1db9c7e58c5e00c967ab878c14461e3 [file] [log] [blame]
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +00001//===- unittests/Lex/PPConditionalDirectiveRecordTest.cpp-PP directive tests =//
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
Chandler Carruth320d9662012-12-04 09:45:34 +000010#include "clang/Lex/PPConditionalDirectiveRecord.h"
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000011#include "clang/Basic/Diagnostic.h"
Douglas Gregor3f7d5482012-10-23 22:43:37 +000012#include "clang/Basic/DiagnosticOptions.h"
Chandler Carruth320d9662012-12-04 09:45:34 +000013#include "clang/Basic/FileManager.h"
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000014#include "clang/Basic/LangOptions.h"
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +000015#include "clang/Basic/MemoryBufferCache.h"
Chandler Carruthfa0b3bb2012-12-04 09:53:37 +000016#include "clang/Basic/SourceManager.h"
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000017#include "clang/Basic/TargetInfo.h"
Chandler Carruth320d9662012-12-04 09:45:34 +000018#include "clang/Basic/TargetOptions.h"
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000019#include "clang/Lex/HeaderSearch.h"
Douglas Gregor40ba1a02012-10-24 16:24:38 +000020#include "clang/Lex/HeaderSearchOptions.h"
Chandler Carruth320d9662012-12-04 09:45:34 +000021#include "clang/Lex/ModuleLoader.h"
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000022#include "clang/Lex/Preprocessor.h"
Douglas Gregor1452ff12012-10-24 17:46:57 +000023#include "clang/Lex/PreprocessorOptions.h"
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000024#include "gtest/gtest.h"
25
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000026using namespace clang;
27
28namespace {
29
30// The test fixture.
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +000031class PPConditionalDirectiveRecordTest : public ::testing::Test {
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000032protected:
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +000033 PPConditionalDirectiveRecordTest()
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000034 : FileMgr(FileMgrOpts),
35 DiagID(new DiagnosticIDs()),
Douglas Gregor3f7d5482012-10-23 22:43:37 +000036 Diags(DiagID, new DiagnosticOptions, new IgnoringDiagConsumer()),
Douglas Gregor44d63612012-10-17 00:11:35 +000037 SourceMgr(Diags, FileMgr),
38 TargetOpts(new TargetOptions)
39 {
40 TargetOpts->Triple = "x86_64-apple-darwin11.1.0";
Alp Toker80758082014-07-06 05:26:44 +000041 Target = TargetInfo::CreateTargetInfo(Diags, TargetOpts);
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000042 }
43
44 FileSystemOptions FileMgrOpts;
45 FileManager FileMgr;
46 IntrusiveRefCntPtr<DiagnosticIDs> DiagID;
47 DiagnosticsEngine Diags;
48 SourceManager SourceMgr;
49 LangOptions LangOpts;
Alp Toker80758082014-07-06 05:26:44 +000050 std::shared_ptr<TargetOptions> TargetOpts;
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000051 IntrusiveRefCntPtr<TargetInfo> Target;
52};
53
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +000054TEST_F(PPConditionalDirectiveRecordTest, PPRecAPI) {
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000055 const char *source =
56 "0 1\n"
57 "#if 1\n"
58 "2\n"
59 "#ifndef BB\n"
60 "3 4\n"
61 "#else\n"
62 "#endif\n"
63 "5\n"
64 "#endif\n"
65 "6\n"
66 "#if 1\n"
67 "7\n"
68 "#if 1\n"
69 "#endif\n"
70 "8\n"
71 "#endif\n"
72 "9\n";
73
James Y Knightb214cbc2016-03-04 19:00:41 +000074 std::unique_ptr<llvm::MemoryBuffer> Buf =
75 llvm::MemoryBuffer::getMemBuffer(source);
David Blaikie50a5f972014-08-29 07:59:55 +000076 SourceMgr.setMainFileID(SourceMgr.createFileID(std::move(Buf)));
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000077
Richard Smith5d2ed482017-06-09 19:22:32 +000078 TrivialModuleLoader ModLoader;
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +000079 MemoryBufferCache PCMCache;
David Blaikie9c28cb32017-01-06 01:04:46 +000080 HeaderSearch HeaderInfo(std::make_shared<HeaderSearchOptions>(), SourceMgr,
81 Diags, LangOpts, Target.get());
David Blaikiee3041682017-01-05 19:11:36 +000082 Preprocessor PP(std::make_shared<PreprocessorOptions>(), Diags, LangOpts,
Duncan P. N. Exon Smith030d7d62017-03-20 17:58:26 +000083 SourceMgr, PCMCache, HeaderInfo, ModLoader,
Craig Topper416fa342014-06-08 08:38:12 +000084 /*IILookup =*/nullptr,
Alp Toker1ae02f62014-05-02 03:43:30 +000085 /*OwnsHeaderSearch =*/false);
86 PP.Initialize(*Target);
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +000087 PPConditionalDirectiveRecord *
88 PPRec = new PPConditionalDirectiveRecord(SourceMgr);
Craig Topperb8a70532014-09-10 04:53:53 +000089 PP.addPPCallbacks(std::unique_ptr<PPCallbacks>(PPRec));
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +000090 PP.EnterMainSourceFile();
91
92 std::vector<Token> toks;
93 while (1) {
94 Token tok;
95 PP.Lex(tok);
96 if (tok.is(tok::eof))
97 break;
98 toks.push_back(tok);
99 }
100
101 // Make sure we got the tokens that we expected.
102 ASSERT_EQ(10U, toks.size());
103
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000104 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000105 SourceRange(toks[0].getLocation(), toks[1].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000106 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000107 SourceRange(toks[0].getLocation(), toks[2].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000108 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000109 SourceRange(toks[3].getLocation(), toks[4].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000110 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000111 SourceRange(toks[1].getLocation(), toks[5].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000112 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000113 SourceRange(toks[2].getLocation(), toks[6].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000114 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000115 SourceRange(toks[2].getLocation(), toks[5].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000116 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000117 SourceRange(toks[0].getLocation(), toks[6].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000118 EXPECT_TRUE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000119 SourceRange(toks[2].getLocation(), toks[8].getLocation())));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000120 EXPECT_FALSE(PPRec->rangeIntersectsConditionalDirective(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000121 SourceRange(toks[0].getLocation(), toks[9].getLocation())));
122
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000123 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000124 toks[0].getLocation(), toks[2].getLocation()));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000125 EXPECT_FALSE(PPRec->areInDifferentConditionalDirectiveRegion(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000126 toks[3].getLocation(), toks[4].getLocation()));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000127 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000128 toks[1].getLocation(), toks[5].getLocation()));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000129 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000130 toks[2].getLocation(), toks[0].getLocation()));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000131 EXPECT_FALSE(PPRec->areInDifferentConditionalDirectiveRegion(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000132 toks[4].getLocation(), toks[3].getLocation()));
Argyrios Kyrtzidisf3d587e2012-12-04 07:27:05 +0000133 EXPECT_TRUE(PPRec->areInDifferentConditionalDirectiveRegion(
Argyrios Kyrtzidis647dcd82012-03-05 05:48:17 +0000134 toks[5].getLocation(), toks[1].getLocation()));
135}
136
137} // anonymous namespace