90 lines · cpp
1//===- MemoryFlags.cpp ----------------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// Tests for orc-rt's MemoryFlags.h APIs.10//11//===----------------------------------------------------------------------===//12 13#include "orc-rt/MemoryFlags.h"14#include "gtest/gtest.h"15 16using namespace orc_rt;17 18TEST(MemProtTest, Basics) {19 MemProt MPNone = MemProt::None;20 21 EXPECT_EQ(MPNone & MemProt::Read, MemProt::None);22 EXPECT_EQ(MPNone & MemProt::Write, MemProt::None);23 EXPECT_EQ(MPNone & MemProt::Exec, MemProt::None);24 25 EXPECT_EQ(MPNone | MemProt::Read, MemProt::Read);26 EXPECT_EQ(MPNone | MemProt::Write, MemProt::Write);27 EXPECT_EQ(MPNone | MemProt::Exec, MemProt::Exec);28 29 MemProt MPAll = MemProt::Read | MemProt::Write | MemProt::Exec;30 EXPECT_EQ(MPAll & MemProt::Read, MemProt::Read);31 EXPECT_EQ(MPAll & MemProt::Write, MemProt::Write);32 EXPECT_EQ(MPAll & MemProt::Exec, MemProt::Exec);33}34 35TEST(AllocGroupTest, Default) {36 AllocGroup G;37 EXPECT_EQ(G.getMemProt(), MemProt::None);38 EXPECT_EQ(G.getMemLifetime(), MemLifetime::Standard);39}40 41TEST(AllocGroupTest, InitMemProtOnly) {42 AllocGroup G(MemProt::Read | MemProt::Write);43 EXPECT_EQ(G.getMemProt(), MemProt::Read | MemProt::Write);44 EXPECT_EQ(G.getMemLifetime(), MemLifetime::Standard);45}46 47TEST(AllocGroupTest, InitMemProtAndLifetime) {48 AllocGroup G(MemProt::Read | MemProt::Write, MemLifetime::Finalize);49 EXPECT_EQ(G.getMemProt(), MemProt::Read | MemProt::Write);50 EXPECT_EQ(G.getMemLifetime(), MemLifetime::Finalize);51}52 53TEST(AllocGroupTest, Equality) {54 AllocGroup G(MemProt::Read, MemLifetime::Standard);55 EXPECT_EQ(G, AllocGroup(MemProt::Read, MemLifetime::Standard));56 EXPECT_NE(G, AllocGroup(MemProt::Write, MemLifetime::Standard));57 EXPECT_NE(G, AllocGroup(MemProt::Read, MemLifetime::Finalize));58}59 60TEST(AllocGroupTest, LessThan) {61 // Check that AllocGroups can be compared via less-than so that they can be62 // used as keys in standard containers.63 EXPECT_LT(AllocGroup(MemProt::Read, MemLifetime::Standard),64 AllocGroup(MemProt::Write, MemLifetime::Standard));65 EXPECT_LT(AllocGroup(MemProt::Exec, MemLifetime::Standard),66 AllocGroup(MemProt::Read, MemLifetime::Finalize));67}68 69TEST(AllocGroupSmallMapTest, EmptyMap) {70 AllocGroupSmallMap<bool> EM;71 EXPECT_TRUE(EM.empty());72 EXPECT_EQ(EM.size(), 0u);73}74 75TEST(AllocGroupSmallMapTest, NonEmptyMap) {76 AllocGroupSmallMap<unsigned> NEM;77 NEM[MemProt::Read] = 42;78 79 EXPECT_FALSE(NEM.empty());80 EXPECT_EQ(NEM.size(), 1U);81 EXPECT_EQ(NEM[MemProt::Read], 42U);82 EXPECT_EQ(NEM.find(MemProt::Read), NEM.begin());83 EXPECT_EQ(NEM.find(MemProt::Read | MemProt::Write), NEM.end());84 85 NEM[MemProt::Read | MemProt::Write] = 7;86 EXPECT_EQ(NEM.size(), 2U);87 EXPECT_EQ(NEM.begin()->second, 42U);88 EXPECT_EQ((NEM.begin() + 1)->second, 7U);89}90