MCJIT: add ProcessAllSections smoke test

[labath]

Summary

None of the existing MCJIT tests set ProcessAllSections to true or use
modules with debug info, so add a test to exercise these code paths.

This would have caught the bug fixed by r303239.

unittests/ExecutionEngine/MCJIT/MCJITTest.cpp

 //===- MCJITTest.cpp - Unit tests for the MCJIT -----------------*- C++ -*-===//
 //
 //                     The LLVM Compiler Infrastructure
 //
 // This file is distributed under the University of Illinois Open Source
 // License. See LICENSE.TXT for details.
 //
 //===----------------------------------------------------------------------===//
 //
 // This test suite verifies basic MCJIT functionality such as making function
 // calls, using global variables, and compiling multpile modules.
 //
 //===----------------------------------------------------------------------===//
 
+#include "MCJITTestBase.h"
 #include "llvm/ExecutionEngine/MCJIT.h"
+#include "llvm/IR/DIBuilder.h"
 #include "llvm/Support/DynamicLibrary.h"
-#include "MCJITTestBase.h"
 #include "gtest/gtest.h"
 
 using namespace llvm;
 
 namespace {
 
 class MCJITTest : public testing::Test, public MCJITTestBase {
 protected:
TEST_F(MCJITTest, lazy_function_creator_lambda) {
   
   // Verify that our handler was called for each unresolved function.
   auto I = UnresolvedExternals.begin(), E = UnresolvedExternals.end();
   EXPECT_EQ(UnresolvedExternals.size(), 2u);
   EXPECT_FALSE(std::find(I, E, "Foo1") == E);
   EXPECT_FALSE(std::find(I, E, "Foo2") == E);
 }
 
+static Function *insertMainWithDI(LLVMContext &Context, Module &M) {
+  DIBuilder DIB(M);
+  DIFile *File = DIB.createFile("-", ".");
+  DIB.createCompileUnit(dwarf::DW_LANG_C99, File, "producer", false, "", 0);
+  DISubroutineType *MainType = DIB.createSubroutineType({});
+  DISubprogram *DIMain = DIB.createFunction(File, "main", "main", File, 1,
+                                            MainType, false, true, 1);
+  DILocation *Loc = DILocation::get(Context, 1, 1, DIMain, nullptr);
+  Function *Main = Function::Create(TypeBuilder<int(void), false>::get(Context),
+                                    GlobalValue::ExternalLinkage, "main", &M);
+  Main->setSubprogram(DIMain);
+  BasicBlock *Entry = BasicBlock::Create(Context, "entry", Main);
+  ReturnInst *Ret =
+      IRBuilder<>(Entry).CreateRet(ConstantInt::get(Context, APInt(32, 47)));
+  Ret->setDebugLoc(Loc);
+  DIB.finalize();
+
+  return Main;
+}
+
+TEST_F(MCJITTest, ProcessAllSections) {
+  Function *Main = insertMainWithDI(Context, *M);
+
+  createJIT(std::move(M));
+  TheJIT->setProcessAllSections(true);
+  TheJIT->finalizeObject();
+
+  uint64_t ptr = TheJIT->getFunctionAddress(Main->getName().str());
+  ASSERT_NE(0u, ptr) << "Unable to get pointer to main() from JIT";
+
+  int (*FuncPtr)() = (int (*)())ptr;
+  int returnCode = FuncPtr();
+  EXPECT_EQ(returnCode, 47);
+}
+
 } // end anonymous namespace