[svn] commit: r3561 - /branches/trac397/src/bin/auth/tests/rbt_datasrc_unittest.cc

[BIND 10 source code commits]

Author: chenzhengzhang
Date: Thu Nov 18 07:56:00 2010
New Revision: 3561

Log:
add unittest for getNodeCount

Modified:
    branches/trac397/src/bin/auth/tests/rbt_datasrc_unittest.cc

Modified: branches/trac397/src/bin/auth/tests/rbt_datasrc_unittest.cc
==============================================================================
--- branches/trac397/src/bin/auth/tests/rbt_datasrc_unittest.cc (original)
+++ branches/trac397/src/bin/auth/tests/rbt_datasrc_unittest.cc Thu Nov 18 07:56:00 2010
@@ -71,6 +71,20 @@
     EXPECT_EQ(13, rbtree.getNodeCount());
 }
 
+TEST_F(RBTreeTest, getNameCount) {
+    EXPECT_EQ(11, rbtree.getNameCount());
+    EXPECT_EQ(0, rbtree.insert(Name("d.e.f"), &rbtnode));
+    EXPECT_EQ(12, rbtree.getNameCount());
+    EXPECT_EQ(0, rbtree.erase(Name("d.e.f")));
+    EXPECT_EQ(11, rbtree.getNameCount());
+    EXPECT_EQ(0, rbtree.erase(Name("o.w.y.d.e.f")));
+    EXPECT_EQ(10, rbtree.getNameCount());
+    EXPECT_EQ(0, rbtree.erase(Name("p.w.y.d.e.f")));
+    EXPECT_EQ(9, rbtree.getNameCount());
+    EXPECT_EQ(0, rbtree.erase(Name("q.w.y.d.e.f")));
+    EXPECT_EQ(8, rbtree.getNameCount());
+}
+
 TEST_F(RBTreeTest, insertNames) {
     // a node is considered to "formally" exist only if it has data
     // associated with it