unittests/ExecutionEngine/Orc/CompileOnDemandLayerTest.cpp

   1 //===----- CompileOnDemandLayerTest.cpp - Unit tests for the COD layer ----===//
   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 "OrcTestCommon.h"
  11 #include "llvm/ExecutionEngine/Orc/CompileOnDemandLayer.h"
  12 #include "gtest/gtest.h"
  13
  14 using namespace llvm;
  15 using namespace llvm::orc;
  16
  17 namespace {
  18
  19 class DummyCallbackManager : public orc::JITCompileCallbackManagerBase {
  20 public:
  21   DummyCallbackManager()
  22       : JITCompileCallbackManagerBase(0, 0), NextStubAddress(0),
  23         UniversalCompile([]() { return 0; }) {
  24   }
  25
  26   CompileCallbackInfo getCompileCallback(LLVMContext &Context) override {
  27     return CompileCallbackInfo(++NextStubAddress, UniversalCompile);
  28   }
  29 public:
  30   TargetAddress NextStubAddress;
  31   CompileFtor UniversalCompile;
  32 };
  33
  34 class DummyStubsManager : public orc::IndirectStubsManagerBase {
  35 public:
  36   std::error_code createStub(StringRef StubName, TargetAddress InitAddr,
  37                              JITSymbolFlags Flags) override {
  38     llvm_unreachable("Not implemented");
  39   }
  40
  41   std::error_code createStubs(const StubInitsMap &StubInits) override {
  42     llvm_unreachable("Not implemented");
  43   }
  44
  45   JITSymbol findStub(StringRef Name, bool ExportedStubsOnly) override {
  46     llvm_unreachable("Not implemented");
  47   }
  48
  49   JITSymbol findPointer(StringRef Name) override {
  50     llvm_unreachable("Not implemented");
  51   }
  52
  53   std::error_code updatePointer(StringRef Name,
  54                                 TargetAddress NewAddr) override {
  55     llvm_unreachable("Not implemented");
  56   }
  57 };
  58
  59 TEST(CompileOnDemandLayerTest, FindSymbol) {
  60   auto MockBaseLayer =
  61     createMockBaseLayer<int>(DoNothingAndReturn<int>(0),
  62                              DoNothingAndReturn<void>(),
  63                              [](const std::string &Name, bool) {
  64                                if (Name == "foo")
  65                                  return JITSymbol(1, JITSymbolFlags::Exported);
  66                                return JITSymbol(nullptr);
  67                              },
  68                              DoNothingAndReturn<JITSymbol>(nullptr));
  69
  70   typedef decltype(MockBaseLayer) MockBaseLayerT;
  71   DummyCallbackManager CallbackMgr;
  72   auto StubsMgrBuilder =
  73     []() {
  74       return llvm::make_unique<DummyStubsManager>();
  75     };
  76
  77   llvm::orc::CompileOnDemandLayer<MockBaseLayerT>
  78     COD(MockBaseLayer,
  79         [](Function &F) { std::set<Function*> S; S.insert(&F); return S; },
  80         CallbackMgr, StubsMgrBuilder, true);
  81   auto Sym = COD.findSymbol("foo", true);
  82
  83   EXPECT_TRUE(!!Sym)
  84     << "CompileOnDemand::findSymbol should call findSymbol in the base layer.";
  85 }
  86
  87 }