unittests/ExecutionEngine/Orc/GlobalMappingLayerTest.cpp

   1 //===--- GlobalMappingLayerTest.cpp - Unit test the global mapping 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/GlobalMappingLayer.h"
  11 #include "gtest/gtest.h"
  12
  13 using namespace llvm;
  14 using namespace llvm::orc;
  15
  16 namespace {
  17
  18 struct MockBaseLayer {
  19
  20   typedef int ModuleSetHandleT;
  21
  22   JITSymbol findSymbol(const std::string &Name, bool ExportedSymbolsOnly) {
  23     if (Name == "bar")
  24       return llvm::orc::JITSymbol(0x4567, JITSymbolFlags::Exported);
  25     return nullptr;
  26   }
  27
  28 };
  29
  30 TEST(GlobalMappingLayerTest, Empty) {
  31   MockBaseLayer M;
  32   GlobalMappingLayer<MockBaseLayer> L(M);
  33
  34   // Test fall-through for missing symbol.
  35   auto FooSym = L.findSymbol("foo", true);
  36   EXPECT_FALSE(FooSym) << "Found unexpected symbol.";
  37
  38   // Test fall-through for symbol in base layer.
  39   auto BarSym = L.findSymbol("bar", true);
  40   EXPECT_EQ(BarSym.getAddress(), static_cast<TargetAddress>(0x4567))
  41     << "Symbol lookup fall-through failed.";
  42
  43   // Test setup of a global mapping.
  44   L.setGlobalMapping("foo", 0x0123);
  45   auto FooSym2 = L.findSymbol("foo", true);
  46   EXPECT_EQ(FooSym2.getAddress(), static_cast<TargetAddress>(0x0123))
  47     << "Symbol mapping setup failed.";
  48
  49   // Test removal of a global mapping.
  50   L.eraseGlobalMapping("foo");
  51   auto FooSym3 = L.findSymbol("foo", true);
  52   EXPECT_FALSE(FooSym3) << "Symbol mapping removal failed.";
  53 }
  54
  55 }