clang/unittests/Lex/PPConditionalDirectiveRecordTest.cpp

   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:
  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
  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