mirror of
https://github.com/RPCS3/llvm.git
synced 2024-12-14 15:39:06 +00:00
8be7707c14
At the same time, fixes InstructionsTest::CastInst unittest: yes you can leave the IR in an invalid state and exit when you don't destroy the context (like the global one), no longer now. This is the first part of http://reviews.llvm.org/D19094 From: Mehdi Amini <mehdi.amini@apple.com> git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@266379 91177308-0d34-0410-b5e6-96231b3b80d8
121 lines
4.7 KiB
C++
121 lines
4.7 KiB
C++
//===- llvm/unittest/IR/UserTest.cpp - User unit tests --------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/AsmParser/Parser.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/User.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "gtest/gtest.h"
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
TEST(UserTest, ValueOpIteration) {
|
|
LLVMContext C;
|
|
|
|
const char *ModuleString = "define void @f(i32 %x, i32 %y) {\n"
|
|
"entry:\n"
|
|
" switch i32 undef, label %s0\n"
|
|
" [ i32 1, label %s1\n"
|
|
" i32 2, label %s2\n"
|
|
" i32 3, label %s3\n"
|
|
" i32 4, label %s4\n"
|
|
" i32 5, label %s5\n"
|
|
" i32 6, label %s6\n"
|
|
" i32 7, label %s7\n"
|
|
" i32 8, label %s8\n"
|
|
" i32 9, label %s9 ]\n"
|
|
"\n"
|
|
"s0:\n"
|
|
" br label %exit\n"
|
|
"s1:\n"
|
|
" br label %exit\n"
|
|
"s2:\n"
|
|
" br label %exit\n"
|
|
"s3:\n"
|
|
" br label %exit\n"
|
|
"s4:\n"
|
|
" br label %exit\n"
|
|
"s5:\n"
|
|
" br label %exit\n"
|
|
"s6:\n"
|
|
" br label %exit\n"
|
|
"s7:\n"
|
|
" br label %exit\n"
|
|
"s8:\n"
|
|
" br label %exit\n"
|
|
"s9:\n"
|
|
" br label %exit\n"
|
|
"\n"
|
|
"exit:\n"
|
|
" %phi = phi i32 [ 0, %s0 ], [ 1, %s1 ],\n"
|
|
" [ 2, %s2 ], [ 3, %s3 ],\n"
|
|
" [ 4, %s4 ], [ 5, %s5 ],\n"
|
|
" [ 6, %s6 ], [ 7, %s7 ],\n"
|
|
" [ 8, %s8 ], [ 9, %s9 ]\n"
|
|
" ret void\n"
|
|
"}\n";
|
|
SMDiagnostic Err;
|
|
std::unique_ptr<Module> M = parseAssemblyString(ModuleString, Err, C);
|
|
|
|
Function *F = M->getFunction("f");
|
|
BasicBlock &ExitBB = F->back();
|
|
PHINode &P = cast<PHINode>(ExitBB.front());
|
|
EXPECT_TRUE(P.value_op_begin() == P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_begin() == P.value_op_end());
|
|
EXPECT_TRUE(P.value_op_begin() != P.value_op_end());
|
|
EXPECT_FALSE(P.value_op_end() != P.value_op_end());
|
|
EXPECT_TRUE(P.value_op_begin() < P.value_op_end());
|
|
EXPECT_FALSE(P.value_op_begin() < P.value_op_begin());
|
|
EXPECT_TRUE(P.value_op_end() > P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_begin() > P.value_op_begin());
|
|
EXPECT_TRUE(P.value_op_begin() <= P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_end() <= P.value_op_begin());
|
|
EXPECT_TRUE(P.value_op_begin() >= P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_begin() >= P.value_op_end());
|
|
EXPECT_EQ(10, std::distance(P.value_op_begin(), P.value_op_end()));
|
|
|
|
User::value_op_iterator I = P.value_op_begin();
|
|
I += 3;
|
|
EXPECT_EQ(std::next(P.value_op_begin(), 3), I);
|
|
EXPECT_EQ(P.getOperand(3), *I);
|
|
I++;
|
|
EXPECT_EQ(P.getOperand(6), I[2]);
|
|
EXPECT_EQ(P.value_op_end(), (I - 2) + 8);
|
|
}
|
|
|
|
TEST(UserTest, PersonalityUser) {
|
|
LLVMContext Context;
|
|
Module M("", Context);
|
|
FunctionType *RetVoidTy = FunctionType::get(Type::getVoidTy(Context), false);
|
|
Function *PersonalityF = Function::Create(
|
|
RetVoidTy, GlobalValue::ExternalLinkage, "PersonalityFn", &M);
|
|
Function *TestF =
|
|
Function::Create(RetVoidTy, GlobalValue::ExternalLinkage, "TestFn", &M);
|
|
|
|
// Set up the personality function
|
|
TestF->setPersonalityFn(PersonalityF);
|
|
auto PersonalityUsers = PersonalityF->user_begin();
|
|
|
|
// One user and that user is the Test function
|
|
EXPECT_EQ(*PersonalityUsers, TestF);
|
|
EXPECT_EQ(++PersonalityUsers, PersonalityF->user_end());
|
|
|
|
// Reset the personality function
|
|
TestF->setPersonalityFn(nullptr);
|
|
|
|
// No users should remain
|
|
EXPECT_TRUE(TestF->user_empty());
|
|
}
|
|
|
|
} // end anonymous namespace
|