[llvm] r185977 - Add some SpecialCaseList unit tests.

[Peter Collingbourne]

Author: pcc
Date: Tue Jul  9 17:03:12 2013
New Revision: 185977

URL: http://llvm.org/viewvc/llvm-project?rev=185977&view=rev
Log:
Add some SpecialCaseList unit tests.

Differential Revision: http://llvm-reviews.chandlerc.com/D1091

Added:
    llvm/trunk/unittests/Transforms/Utils/SpecialCaseList.cpp
Modified:
    llvm/trunk/unittests/Transforms/Utils/CMakeLists.txt

Modified: llvm/trunk/unittests/Transforms/Utils/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/Transforms/Utils/CMakeLists.txt?rev=185977&r1=185976&r2=185977&view=diff
==============================================================================
--- llvm/trunk/unittests/Transforms/Utils/CMakeLists.txt (original)
+++ llvm/trunk/unittests/Transforms/Utils/CMakeLists.txt Tue Jul  9 17:03:12 2013
@@ -6,4 +6,5 @@ add_llvm_unittest(UtilsTests
   Cloning.cpp
   IntegerDivision.cpp
   Local.cpp
+  SpecialCaseList.cpp
   )

Added: llvm/trunk/unittests/Transforms/Utils/SpecialCaseList.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/Transforms/Utils/SpecialCaseList.cpp?rev=185977&view=auto
==============================================================================
--- llvm/trunk/unittests/Transforms/Utils/SpecialCaseList.cpp (added)
+++ llvm/trunk/unittests/Transforms/Utils/SpecialCaseList.cpp Tue Jul  9 17:03:12 2013
@@ -0,0 +1,117 @@
+//===- SpecialCaseList.cpp - Unit tests for SpecialCaseList ---------------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/IR/Function.h"
+#include "llvm/IR/LLVMContext.h"
+#include "llvm/IR/Module.h"
+#include "llvm/Support/MemoryBuffer.h"
+#include "llvm/Transforms/Utils/SpecialCaseList.h"
+#include "gtest/gtest.h"
+
+using namespace llvm;
+
+namespace {
+
+class SpecialCaseListTest : public ::testing::Test {
+protected:
+  Function *makeFunction(StringRef Name, Module &M) {
+    return Function::Create(FunctionType::get(Type::getVoidTy(Ctx), false),
+                            GlobalValue::ExternalLinkage,
+                            Name,
+                            &M);
+  }
+
+  GlobalVariable *makeGlobal(StringRef Name, StringRef StructName, Module &M) {
+    StructType *ST =
+        StructType::create(StructName, Type::getInt32Ty(Ctx), (Type*)0);
+    return new GlobalVariable(
+        M, ST, false, GlobalValue::ExternalLinkage, 0, Name);
+  }
+
+  SpecialCaseList *makeSpecialCaseList(StringRef List) {
+    OwningPtr<MemoryBuffer> MB(MemoryBuffer::getMemBuffer(List));
+    return new SpecialCaseList(MB.get());
+  }
+
+  LLVMContext Ctx;
+};
+
+TEST_F(SpecialCaseListTest, ModuleIsIn) {
+  Module M("hello", Ctx);
+  Function *F = makeFunction("foo", M);
+  GlobalVariable *GV = makeGlobal("bar", "t", M);
+
+  OwningPtr<SpecialCaseList> SCL(makeSpecialCaseList("# This is a comment.\n"
+                                                     "\n"
+                                                     "src:hello\n"));
+  EXPECT_TRUE(SCL->isIn(M));
+  EXPECT_TRUE(SCL->isIn(*F));
+  EXPECT_TRUE(SCL->isIn(*GV));
+
+  SCL.reset(makeSpecialCaseList("src:he*o\n"));
+  EXPECT_TRUE(SCL->isIn(M));
+  EXPECT_TRUE(SCL->isIn(*F));
+  EXPECT_TRUE(SCL->isIn(*GV));
+
+  SCL.reset(makeSpecialCaseList("src:hi\n"));
+  EXPECT_FALSE(SCL->isIn(M));
+  EXPECT_FALSE(SCL->isIn(*F));
+  EXPECT_FALSE(SCL->isIn(*GV));
+}
+
+TEST_F(SpecialCaseListTest, FunctionIsIn) {
+  Module M("hello", Ctx);
+  Function *Foo = makeFunction("foo", M);
+  Function *Bar = makeFunction("bar", M);
+
+  OwningPtr<SpecialCaseList> SCL(makeSpecialCaseList("fun:foo\n"));
+  EXPECT_TRUE(SCL->isIn(*Foo));
+  EXPECT_FALSE(SCL->isIn(*Bar));
+
+  SCL.reset(makeSpecialCaseList("fun:b*\n"));
+  EXPECT_FALSE(SCL->isIn(*Foo));
+  EXPECT_TRUE(SCL->isIn(*Bar));
+
+  SCL.reset(makeSpecialCaseList("fun:f*\n"
+                                "fun:bar\n"));
+  EXPECT_TRUE(SCL->isIn(*Foo));
+  EXPECT_TRUE(SCL->isIn(*Bar));
+}
+
+TEST_F(SpecialCaseListTest, GlobalIsIn) {
+  Module M("hello", Ctx);
+  GlobalVariable *Foo = makeGlobal("foo", "t1", M);
+  GlobalVariable *Bar = makeGlobal("bar", "t2", M);
+
+  OwningPtr<SpecialCaseList> SCL(makeSpecialCaseList("global:foo\n"));
+  EXPECT_TRUE(SCL->isIn(*Foo));
+  EXPECT_FALSE(SCL->isIn(*Bar));
+  EXPECT_FALSE(SCL->isInInit(*Foo));
+  EXPECT_FALSE(SCL->isInInit(*Bar));
+
+  SCL.reset(makeSpecialCaseList("global-init:foo\n"));
+  EXPECT_FALSE(SCL->isIn(*Foo));
+  EXPECT_FALSE(SCL->isIn(*Bar));
+  EXPECT_TRUE(SCL->isInInit(*Foo));
+  EXPECT_FALSE(SCL->isInInit(*Bar));
+
+  SCL.reset(makeSpecialCaseList("global-init-type:t2\n"));
+  EXPECT_FALSE(SCL->isIn(*Foo));
+  EXPECT_FALSE(SCL->isIn(*Bar));
+  EXPECT_FALSE(SCL->isInInit(*Foo));
+  EXPECT_TRUE(SCL->isInInit(*Bar));
+
+  SCL.reset(makeSpecialCaseList("global-init-src:hello\n"));
+  EXPECT_FALSE(SCL->isIn(*Foo));
+  EXPECT_FALSE(SCL->isIn(*Bar));
+  EXPECT_TRUE(SCL->isInInit(*Foo));
+  EXPECT_TRUE(SCL->isInInit(*Bar));
+}
+
+}