1 //===--- GlobalMappingLayerTest.cpp - Unit test the global mapping layer --===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/ExecutionEngine/Orc/GlobalMappingLayer.h"
11 #include "gtest/gtest.h"
14 using namespace llvm::orc;
18 struct MockBaseLayer {
20 typedef int ModuleSetHandleT;
22 JITSymbol findSymbol(const std::string &Name, bool ExportedSymbolsOnly) {
24 return llvm::orc::JITSymbol(0x4567, JITSymbolFlags::Exported);
30 TEST(GlobalMappingLayerTest, Empty) {
32 GlobalMappingLayer<MockBaseLayer> L(M);
34 // Test fall-through for missing symbol.
35 auto FooSym = L.findSymbol("foo", true);
36 EXPECT_FALSE(FooSym) << "Found unexpected symbol.";
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.";
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.";
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.";