Add test case for reference-pointer

ropez · Feb 7, 2010 · 76b4254 · 76b4254
1 parent dd1c260
commit 76b4254
Show file tree

Hide file tree

Showing 2 changed files with 80 additions and 1 deletion.
diff --git a/src/test/CMakeLists.txt b/src/test/CMakeLists.txt
@@ -15,6 +15,7 @@ target_link_libraries(testrunner pieces ${CPPUNIT_LIBRARIES} ${OPENTHREADS_LIBRA
 set(tests
     TestSharedData
     TestAutoPointer
+    TestReferencePointer
     TestByteArray
     TestByteArraySlicing
     TestByteArrayChopping

diff --git a/src/test/test_smart_pointers.cpp b/src/test/test_smart_pointers.cpp
@@ -2,9 +2,11 @@
 #include <cppunit/extensions/HelperMacros.h>
 #include <Pieces/SharedData>
 #include <Pieces/AutoPointer>
+#include <Pieces/ReferencePointer>
 #include <Pieces/global>
 
 using pcs::AutoPointer;
+using pcs::ReferencePointer;
 
 namespace {
 class MockObject : public pcs::SharedData
@@ -93,14 +95,18 @@ class TestAutoPointer : public CppUnit::TestFixture
         CPPUNIT_ASSERT_EQUAL(0, MockObject::count);
         p = new MockObject;
         CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+        p = new MockObject;
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
         p = 0;
         CPPUNIT_ASSERT_EQUAL(0, MockObject::count);
     }
 
     void testOwnerChange() {
         AutoPointer<MockObject> p(new MockObject);
         CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
-        AutoPointer<MockObject> q = p;
+        AutoPointer<MockObject> q(new MockObject);
+        CPPUNIT_ASSERT_EQUAL(2, MockObject::count);
+        q = p;
         CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
         CPPUNIT_ASSERT(q.isValid());
         CPPUNIT_ASSERT(p.isNull());
@@ -118,3 +124,75 @@ class TestAutoPointer : public CppUnit::TestFixture
     }
 };
 CPPUNIT_TEST_SUITE_REGISTRATION(TestAutoPointer);
+
+class TestReferencePointer : public CppUnit::TestFixture
+{
+    CPPUNIT_TEST_SUITE(TestReferencePointer);
+    CPPUNIT_TEST(testNullPointer);
+    CPPUNIT_TEST(testScopeCleanUp);
+    CPPUNIT_TEST(testManualCleanUp);
+    CPPUNIT_TEST(testMultipleReferences);
+    CPPUNIT_TEST(testSelfAssignment);
+    CPPUNIT_TEST_SUITE_END();
+
+public:
+    void setUp() {
+        MockObject::count = 0;
+    }
+
+    void tearDown() {
+        CPPUNIT_ASSERT_EQUAL(0, MockObject::count);
+    }
+
+    void testNullPointer() {
+        ReferencePointer<MockObject> p;
+        CPPUNIT_ASSERT(p.isNull());
+        CPPUNIT_ASSERT(!p.isValid());
+        CPPUNIT_ASSERT(!p);
+        CPPUNIT_ASSERT(p.get() == 0);
+    }
+
+    void testScopeCleanUp() {
+        ReferencePointer<MockObject> p(new MockObject);
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+    }
+
+    void testManualCleanUp() {
+        ReferencePointer<MockObject> p;
+        CPPUNIT_ASSERT_EQUAL(0, MockObject::count);
+        p = new MockObject;
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+        p = new MockObject;
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+        p = 0;
+        CPPUNIT_ASSERT_EQUAL(0, MockObject::count);
+    }
+
+    void testMultipleReferences() {
+        ReferencePointer<MockObject> p(new MockObject);
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+        ReferencePointer<MockObject> q(new MockObject);
+        CPPUNIT_ASSERT_EQUAL(2, MockObject::count);
+        q = p;
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+        CPPUNIT_ASSERT(q.isValid());
+        CPPUNIT_ASSERT(p.isValid());
+        CPPUNIT_ASSERT(p->shared());
+        p = q;
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+        CPPUNIT_ASSERT(p.isValid());
+        CPPUNIT_ASSERT(q.isValid());
+        CPPUNIT_ASSERT(p->shared());
+    }
+
+    void testSelfAssignment() {
+        ReferencePointer<MockObject> p(new MockObject);
+        p = p;
+        CPPUNIT_ASSERT_EQUAL(1, MockObject::count);
+        CPPUNIT_ASSERT(p.isValid());
+        CPPUNIT_ASSERT(!p->shared());
+    }
+};
+CPPUNIT_TEST_SUITE_REGISTRATION(TestReferencePointer);
+
+