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 createStubs(const StubInitsMap &StubInits) override {
  37     llvm_unreachable("Not implemented");
  38   }
  39
  40   JITSymbol findStub(StringRef Name, bool ExportedStubsOnly) override {
  41     llvm_unreachable("Not implemented");
  42   }
  43
  44   JITSymbol findPointer(StringRef Name) override {
  45     llvm_unreachable("Not implemented");
  46   }
  47
  48   std::error_code updatePointer(StringRef Name,
  49                                 TargetAddress NewAddr) override {
  50     llvm_unreachable("Not implemented");
  51   }
  52 };
  53
  54 TEST(CompileOnDemandLayerTest, FindSymbol) {
  55   auto MockBaseLayer =
  56     createMockBaseLayer<int>(DoNothingAndReturn<int>(0),
  57                              DoNothingAndReturn<void>(),
  58                              [](const std::string &Name, bool) {
  59                                if (Name == "foo")
  60                                  return JITSymbol(1, JITSymbolFlags::Exported);
  61                                return JITSymbol(nullptr);
  62                              },
  63                              DoNothingAndReturn<JITSymbol>(nullptr));
  64
  65   typedef decltype(MockBaseLayer) MockBaseLayerT;
  66   DummyCallbackManager CallbackMgr;
  67   auto StubsMgrBuilder =
  68     []() {
  69       return llvm::make_unique<DummyStubsManager>();
  70     };
  71
  72   llvm::orc::CompileOnDemandLayer<MockBaseLayerT>
  73     COD(MockBaseLayer,
  74         [](Function &F) { std::set<Function*> S; S.insert(&F); return S; },
  75         CallbackMgr, StubsMgrBuilder, true);
  76   auto Sym = COD.findSymbol("foo", true);
  77
  78   EXPECT_TRUE(!!Sym)
  79     << "CompileOnDemand::findSymbol should call findSymbol in the base layer.";
  80 }
  81
  82 }