[llvm-commits] [llvm] r108854 - /llvm/trunk/unittests/Support/Casting.cpp

[Gabor Greif]

Author: ggreif
Date: Tue Jul 20 11:51:18 2010
New Revision: 108854

URL: http://llvm.org/viewvc/llvm-project?rev=108854&view=rev
Log:
extend to cast<> and cast_or_null<> tests

Modified:
    llvm/trunk/unittests/Support/Casting.cpp

Modified: llvm/trunk/unittests/Support/Casting.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/Support/Casting.cpp?rev=108854&r1=108853&r2=108854&view=diff
==============================================================================
--- llvm/trunk/unittests/Support/Casting.cpp (original)
+++ llvm/trunk/unittests/Support/Casting.cpp Tue Jul 20 11:51:18 2010
@@ -19,19 +19,47 @@
 
 namespace {
 
+const foo *null_foo = NULL;
+
 extern bar &B1;
 extern const bar *B2;
+// test various configurations of const
+const bar &B3 = B1;
+const bar *const B4 = B2;
 
 TEST(CastingTest, isa) {
-  // test various configurations of const
-  const bar &B3 = B1;
-  const bar *const B4 = B2;
   EXPECT_TRUE(isa<foo>(B1));
   EXPECT_TRUE(isa<foo>(B2));
   EXPECT_TRUE(isa<foo>(B3));
   EXPECT_TRUE(isa<foo>(B4));
 }
 
+TEST(CastingTest, cast) {
+  foo &F1 = cast<foo>(B1);
+  EXPECT_NE(&F1, null_foo);
+  const foo *F3 = cast<foo>(B2);
+  EXPECT_NE(F3, null_foo);
+  const foo *F4 = cast<foo>(B2);
+  EXPECT_NE(F4, null_foo);
+  const foo &F8 = cast<foo>(B3);
+  EXPECT_NE(&F8, null_foo);
+  const foo *F9 = cast<foo>(B4);
+  EXPECT_NE(F9, null_foo);
+  foo *F10 = cast<foo>(fub());
+  EXPECT_EQ(F10, null_foo);
+}
+
+TEST(CastingTest, cast_or_null) {
+  const foo *F11 = cast_or_null<foo>(B2);
+  EXPECT_NE(F11, null_foo);
+  const foo *F12 = cast_or_null<foo>(B2);
+  EXPECT_NE(F12, null_foo);
+  const foo *F13 = cast_or_null<foo>(B4);
+  EXPECT_NE(F13, null_foo);
+  const foo *F14 = cast_or_null<foo>(fub());  // Shouldn't print.
+  EXPECT_EQ(F14, null_foo);
+}
+
 bar B;
 bar &B1 = B;
 const bar *B2 = &B;