Local.cpp revision d8b4fb4aab4d6fedb2b14bed1b846451b17bde7c
1//===- Local.cpp - Unit tests for Local -----------------------------------===// 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 "gtest/gtest.h" 11#include "llvm/BasicBlock.h" 12#include "llvm/Instructions.h" 13#include "llvm/LLVMContext.h" 14#include "llvm/Support/IRBuilder.h" 15#include "llvm/Transforms/Utils/Local.h" 16 17using namespace llvm; 18 19TEST(Local, RecursivelyDeleteDeadPHINodes) { 20 LLVMContext &C(getGlobalContext()); 21 22 IRBuilder<> builder(C); 23 24 // Make blocks 25 BasicBlock *bb0 = BasicBlock::Create(C); 26 BasicBlock *bb1 = BasicBlock::Create(C); 27 28 builder.SetInsertPoint(bb0); 29 PHINode *phi = builder.CreatePHI(Type::getInt32Ty(C)); 30 phi->reserveOperandSpace(2); 31 BranchInst *br0 = builder.CreateCondBr(builder.getTrue(), bb0, bb1); 32 33 builder.SetInsertPoint(bb1); 34 BranchInst *br1 = builder.CreateBr(bb0); 35 36 phi->addIncoming(phi, bb0); 37 phi->addIncoming(phi, bb1); 38 39 // The PHI will be removed 40 EXPECT_TRUE(RecursivelyDeleteDeadPHINode(phi)); 41 42 // Make sure the blocks only contain the branches 43 EXPECT_EQ(&bb0->front(), br0); 44 EXPECT_EQ(&bb1->front(), br1); 45 46 builder.SetInsertPoint(bb0); 47 phi = builder.CreatePHI(Type::getInt32Ty(C)); 48 49 EXPECT_TRUE(RecursivelyDeleteDeadPHINode(phi)); 50 51 builder.SetInsertPoint(bb0); 52 phi = builder.CreatePHI(Type::getInt32Ty(C)); 53 builder.CreateAdd(phi, phi); 54 55 EXPECT_TRUE(RecursivelyDeleteDeadPHINode(phi)); 56 57 bb0->dropAllReferences(); 58 bb1->dropAllReferences(); 59 delete bb0; 60 delete bb1; 61} 62