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 "llvm/ExecutionEngine/Orc/CompileOnDemandLayer.h"
  11 #include "OrcTestCommon.h"
  12 #include "gtest/gtest.h"
  13
  14 using namespace llvm;
  15 using namespace llvm::orc;
  16
  17 namespace {
  18
  19 class DummyTrampolinePool : public orc::TrampolinePool {
  20 public:
  21   Expected<JITTargetAddress> getTrampoline() {
  22     llvm_unreachable("Unimplemented");
  23   }
  24 };
  25
  26 class DummyCallbackManager : public JITCompileCallbackManager {
  27 public:
  28   DummyCallbackManager(ExecutionSession &ES)
  29       : JITCompileCallbackManager(llvm::make_unique<DummyTrampolinePool>(), ES,
  30                                   0) {}
  31 };
  32
  33 class DummyStubsManager : public orc::IndirectStubsManager {
  34 public:
  35   Error createStub(StringRef StubName, JITTargetAddress InitAddr,
  36                    JITSymbolFlags Flags) override {
  37     llvm_unreachable("Not implemented");
  38   }
  39
  40   Error createStubs(const StubInitsMap &StubInits) override {
  41     llvm_unreachable("Not implemented");
  42   }
  43
  44   JITEvaluatedSymbol findStub(StringRef Name, bool ExportedStubsOnly) override {
  45     llvm_unreachable("Not implemented");
  46   }
  47
  48   JITEvaluatedSymbol findPointer(StringRef Name) override {
  49     llvm_unreachable("Not implemented");
  50   }
  51
  52   Error updatePointer(StringRef Name, JITTargetAddress NewAddr) override {
  53     llvm_unreachable("Not implemented");
  54   }
  55 };
  56
  57 TEST(CompileOnDemandLayerTest, FindSymbol) {
  58   MockBaseLayer<int, std::shared_ptr<Module>> TestBaseLayer;
  59   TestBaseLayer.findSymbolImpl =
  60     [](const std::string &Name, bool) {
  61       if (Name == "foo")
  62         return JITSymbol(1, JITSymbolFlags::Exported);
  63       return JITSymbol(nullptr);
  64     };
  65
  66
  67   ExecutionSession ES(std::make_shared<SymbolStringPool>());
  68   DummyCallbackManager CallbackMgr(ES);
  69
  70   auto GetResolver =
  71       [](orc::VModuleKey) -> std::shared_ptr<llvm::orc::SymbolResolver> {
  72     llvm_unreachable("Should never be called");
  73   };
  74
  75   auto SetResolver = [](orc::VModuleKey, std::shared_ptr<orc::SymbolResolver>) {
  76     llvm_unreachable("Should never be called");
  77   };
  78
  79   llvm::orc::CompileOnDemandLayer<decltype(TestBaseLayer)> COD(
  80       ES, TestBaseLayer, GetResolver, SetResolver,
  81       [](Function &F) { return std::set<Function *>{&F}; }, CallbackMgr,
  82       [] { return llvm::make_unique<DummyStubsManager>(); }, true);
  83
  84   auto Sym = COD.findSymbol("foo", true);
  85
  86   EXPECT_TRUE(!!Sym) << "CompileOnDemand::findSymbol should call findSymbol in "
  87                         "the base layer.";
  88 }
  89 }