Home | History | Annotate | Download | only in CodeGen
      1 //===- unittests/CodeGen/BufferSourceTest.cpp - MemoryBuffer source 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/AST/ASTConsumer.h"
     11 #include "clang/AST/ASTContext.h"
     12 #include "clang/AST/RecursiveASTVisitor.h"
     13 #include "clang/Basic/TargetInfo.h"
     14 #include "clang/CodeGen/ModuleBuilder.h"
     15 #include "clang/Frontend/CompilerInstance.h"
     16 #include "clang/Lex/Preprocessor.h"
     17 #include "clang/Parse/ParseAST.h"
     18 #include "clang/Sema/Sema.h"
     19 #include "llvm/ADT/Triple.h"
     20 #include "llvm/IR/LLVMContext.h"
     21 #include "llvm/Support/Host.h"
     22 #include "llvm/Support/MemoryBuffer.h"
     23 #include "gtest/gtest.h"
     24 
     25 using namespace llvm;
     26 using namespace clang;
     27 
     28 namespace {
     29 
     30 // Emitting constructors for global objects involves looking
     31 // at the source file name. This makes sure that we don't crash
     32 // if the source file is a memory buffer.
     33 const char TestProgram[] =
     34     "class EmitCXXGlobalInitFunc    "
     35     "{                              "
     36     "public:                        "
     37     "   EmitCXXGlobalInitFunc() {}  "
     38     "};                             "
     39     "EmitCXXGlobalInitFunc test;    ";
     40 
     41 TEST(BufferSourceTest, EmitCXXGlobalInitFunc) {
     42     LLVMContext Context;
     43     CompilerInstance compiler;
     44 
     45     compiler.createDiagnostics();
     46     compiler.getLangOpts().CPlusPlus = 1;
     47     compiler.getLangOpts().CPlusPlus11 = 1;
     48 
     49     compiler.getTargetOpts().Triple = llvm::Triple::normalize(
     50         llvm::sys::getProcessTriple());
     51     compiler.setTarget(clang::TargetInfo::CreateTargetInfo(
     52       compiler.getDiagnostics(),
     53       std::make_shared<clang::TargetOptions>(
     54         compiler.getTargetOpts())));
     55 
     56     compiler.createFileManager();
     57     compiler.createSourceManager(compiler.getFileManager());
     58     compiler.createPreprocessor(clang::TU_Prefix);
     59 
     60     compiler.createASTContext();
     61 
     62     compiler.setASTConsumer(std::unique_ptr<ASTConsumer>(
     63         CreateLLVMCodeGen(
     64             compiler.getDiagnostics(),
     65             "EmitCXXGlobalInitFuncTest",
     66             compiler.getHeaderSearchOpts(),
     67             compiler.getPreprocessorOpts(),
     68             compiler.getCodeGenOpts(),
     69             Context)));
     70 
     71     compiler.createSema(clang::TU_Prefix, nullptr);
     72 
     73     clang::SourceManager &sm = compiler.getSourceManager();
     74     sm.setMainFileID(sm.createFileID(
     75         llvm::MemoryBuffer::getMemBuffer(TestProgram), clang::SrcMgr::C_User));
     76 
     77     clang::ParseAST(compiler.getSema(), false, false);
     78 }
     79 
     80 } // end anonymous namespace
     81