summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGabe Black <gblack@eecs.umich.edu>2011-01-10 03:56:42 -0800
committerGabe Black <gblack@eecs.umich.edu>2011-01-10 03:56:42 -0800
commitdc64732dee916535b555cbddf4c7590c14e72b9e (patch)
tree09c3b2b442d9205392c477e785409bdfe37de4be
parent6f1187943cf78c2fd0334bd7e4372ae79a587fa4 (diff)
downloadgem5-dc64732dee916535b555cbddf4c7590c14e72b9e.tar.xz
RefCount: Add a unit test for reference counting pointers.
This test exercises each of the functions in the reference counting pointer implementation individually (except get()) and verifies they have some minimially expected behavior. It also checks that reference counted objects are freed when their usage count goes to 0 in some basic situations, specifically a pointer being set to NULL and a pointer being deleted.
-rw-r--r--src/unittest/SConscript1
-rw-r--r--src/unittest/refcnttest.cc187
2 files changed, 188 insertions, 0 deletions
diff --git a/src/unittest/SConscript b/src/unittest/SConscript
index 91ead522f..e7e2a6013 100644
--- a/src/unittest/SConscript
+++ b/src/unittest/SConscript
@@ -44,6 +44,7 @@ UnitTest('offtest', 'offtest.cc')
UnitTest('rangetest', 'rangetest.cc')
UnitTest('rangemaptest', 'rangemaptest.cc')
UnitTest('rangemultimaptest', 'rangemultimaptest.cc')
+UnitTest('refcnttest', 'refcnttest.cc')
UnitTest('stattest', 'stattest.cc')
UnitTest('strnumtest', 'strnumtest.cc')
UnitTest('symtest', 'symtest.cc')
diff --git a/src/unittest/refcnttest.cc b/src/unittest/refcnttest.cc
new file mode 100644
index 000000000..e7dfd3693
--- /dev/null
+++ b/src/unittest/refcnttest.cc
@@ -0,0 +1,187 @@
+/*
+ * Copyright (c) 2010 The Regents of The University of Michigan
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met: redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer;
+ * redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution;
+ * neither the name of the copyright holders nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * Authors: Gabe Black
+ */
+
+#include <cassert>
+#include <iostream>
+#include <list>
+
+#include "base/refcnt.hh"
+
+using namespace std;
+
+namespace {
+
+class TestRC;
+typedef list<TestRC *> LiveList;
+LiveList liveList;
+
+int
+live()
+{
+ return liveList.size();
+}
+
+int
+liveChange()
+{
+ static int oldLive = 0;
+ int newLive = live();
+ int diff = newLive - oldLive;
+ oldLive = newLive;
+ return diff;
+}
+
+class TestRC : public RefCounted
+{
+ protected:
+ const char *_tag;
+ LiveList::iterator liveIt;
+
+ public:
+ TestRC(const char *newTag) : _tag(newTag)
+ {
+ cout << " Creating object \"" << _tag << "\"\n";
+ liveList.push_front(this);
+ liveIt = liveList.begin();
+ }
+
+ ~TestRC()
+ {
+ cout << " Destroying object \"" << _tag << "\"\n";
+ liveList.erase(liveIt);
+ }
+
+ const char *
+ tag()
+ {
+ return _tag;
+ }
+
+ int testVal;
+};
+
+typedef RefCountingPtr<TestRC> Ptr;
+
+}
+
+int
+main()
+{
+ assert(live() == 0);
+ assert(liveChange() == 0);
+
+ // Create an empty Ptr and verify it's data pointer is NULL.
+ cout << "NULL check.\n";
+ Ptr nullCheck;
+ assert(nullCheck.get() == NULL);
+
+ assert(liveChange() == 0);
+
+ // Construct a Ptr from a TestRC pointer.
+ cout << "Construction from pointer.\n";
+ Ptr constFromPointer = new TestRC("construction from pointer");
+
+ assert(liveChange() == 1);
+
+ // Construct a Ptr from an existing Ptr.
+ cout << "Construction from a Ptr.\n";
+ Ptr constFromPtr = constFromPointer;
+
+ assert(liveChange() == 0);
+
+ // Test a Ptr being destroyed.
+ cout << "Destroying a Ptr.\n";
+ Ptr *ptrPtr = new Ptr(new TestRC("destroying a ptr"));
+ assert(liveChange() == 1);
+ delete ptrPtr;
+ assert(liveChange() == -1);
+
+ // Test assignment from a pointer and from a Ptr.
+ cout << "Assignment operators.\n";
+ Ptr assignmentTarget;
+ TestRC *assignmentSourcePointer = new TestRC("assignment source 1");
+ assert(liveChange() == 1);
+ assignmentTarget = assignmentSourcePointer;
+ assert(liveChange() == 0);
+ assignmentTarget = NULL;
+ assert(liveChange() == -1);
+ Ptr assignmentSourcePtr(new TestRC("assignment source 2"));
+ assert(liveChange() == 1);
+ assignmentTarget = assignmentSourcePtr;
+ assert(liveChange() == 0);
+ assignmentSourcePtr = NULL;
+ assert(liveChange() == 0);
+ assignmentTarget = NULL;
+ assert(liveChange() == -1);
+
+ // Test access to members of the pointed to class and dereferencing.
+ cout << "Access to members.\n";
+ TestRC *accessTest = new TestRC("access test");
+ Ptr accessTestPtr = accessTest;
+ accessTest->testVal = 1;
+ assert(accessTestPtr->testVal == 1);
+ assert((*accessTestPtr).testVal == 1);
+ accessTest->testVal = 2;
+ assert(accessTestPtr->testVal == 2);
+ assert((*accessTestPtr).testVal == 2);
+ accessTestPtr->testVal = 3;
+ assert(accessTest->testVal == 3);
+ (*accessTestPtr).testVal = 4;
+ assert(accessTest->testVal == 4);
+ accessTestPtr = NULL;
+ accessTest = NULL;
+ assert(liveChange() == 0);
+
+ // Test bool and ! operator overloads.
+ cout << "Conversion to bool and ! overload.\n";
+ Ptr boolTest = new TestRC("bool test");
+ assert(boolTest == true);
+ assert(!boolTest == false);
+ boolTest = NULL;
+ assert(boolTest == false);
+ assert(!boolTest == true);
+ assert(liveChange() == 0);
+
+ // Test the equality operators.
+ cout << "Equality operators.\n";
+ TestRC *equalTestA = new TestRC("equal test a");
+ Ptr equalTestAPtr = equalTestA;
+ Ptr equalTestAPtr2 = equalTestA;
+ TestRC *equalTestB = new TestRC("equal test b");
+ Ptr equalTestBPtr = equalTestB;
+ assert(equalTestA == equalTestAPtr);
+ assert(equalTestAPtr == equalTestA);
+ assert(equalTestAPtr == equalTestAPtr2);
+ assert(equalTestA != equalTestBPtr);
+ assert(equalTestAPtr != equalTestB);
+ assert(equalTestAPtr != equalTestBPtr);
+
+ cout << flush;
+}