From b32a79f9354951bcb01f4171c3691ffbe020c052 Mon Sep 17 00:00:00 2001
From: Chandler Carruth <chandlerc@gmail.com>
Date: Sat, 9 Nov 2013 04:58:13 +0000
Subject: Test the polymorphic behavior of this utility.

llvm-svn: 194320
---
 llvm/unittests/ADT/polymorphic_ptr_test.cpp | 34 ++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

(limited to 'llvm/unittests/ADT/polymorphic_ptr_test.cpp')

diff --git a/llvm/unittests/ADT/polymorphic_ptr_test.cpp b/llvm/unittests/ADT/polymorphic_ptr_test.cpp
index ce001d1e309..d9a2c3fdd11 100644
--- a/llvm/unittests/ADT/polymorphic_ptr_test.cpp
+++ b/llvm/unittests/ADT/polymorphic_ptr_test.cpp
@@ -22,7 +22,8 @@ struct S {
 };
 
 // A function that forces the return of a copy.
-polymorphic_ptr<S> dummy_copy(const polymorphic_ptr<S> &arg) { return arg; }
+template <typename T>
+T dummy_copy(const T &arg) { return arg; }
 
 TEST(polymorphic_ptr_test, Basic) {
   polymorphic_ptr<S> null;
@@ -83,4 +84,35 @@ TEST(polymorphic_ptr_test, Basic) {
   EXPECT_EQ(42, p3->x);
 }
 
+struct Base {
+  virtual ~Base() {}
+  virtual Base *clone() = 0;
+  virtual StringRef name() { return "Base"; }
+};
+
+struct DerivedA : Base {
+  virtual DerivedA *clone() { return new DerivedA(); }
+  virtual StringRef name() { return "DerivedA"; }
+};
+struct DerivedB : Base {
+  virtual DerivedB *clone() { return new DerivedB(); }
+  virtual StringRef name() { return "DerivedB"; }
+};
+
+TEST(polymorphic_ptr_test, Polymorphism) {
+  polymorphic_ptr<Base> a(new DerivedA());
+  polymorphic_ptr<Base> b(new DerivedB());
+
+  EXPECT_EQ("DerivedA", a->name());
+  EXPECT_EQ("DerivedB", b->name());
+
+  polymorphic_ptr<Base> copy = dummy_copy(a);
+  EXPECT_NE(a, copy);
+  EXPECT_EQ("DerivedA", copy->name());
+
+  copy = dummy_copy(b);
+  EXPECT_NE(b, copy);
+  EXPECT_EQ("DerivedB", copy->name());
+}
+
 }
-- 
cgit v1.2.3