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