Browse Source

[2108] Add two more InMemoryClient unittests

Mukund Sivaraman 12 years ago
parent
commit
90b10c0f2a
1 changed files with 7 additions and 6 deletions
  1. 7 6
      src/lib/datasrc/memory/tests/memory_client_unittest.cc

+ 7 - 6
src/lib/datasrc/memory/tests/memory_client_unittest.cc

@@ -66,15 +66,16 @@ protected:
     memory::ZoneTable* zone_table;
 };
 
-TEST_F(MemoryClientTest, load) {
-    // Zone "example.org." doesn't exist
-    EXPECT_THROW(client_->getIterator(Name("example.org")), DataSourceError);
-    EXPECT_EQ(0, client_->getZoneCount());
+TEST_F(MemoryClientTest, loadRRsetDoesntMatchOrigin) {
+    EXPECT_THROW(client_->load(Name("example.com"),
+                               TEST_DATA_DIR "/example.org-empty.zone"),
+                 MasterLoadError);
+}
 
+TEST_F(MemoryClientTest, getZoneCount) {
+    EXPECT_EQ(0, client_->getZoneCount());
     client_->load(Name("example.org"), TEST_DATA_DIR "/example.org-empty.zone");
     EXPECT_EQ(1, client_->getZoneCount());
-
-    ZoneIteratorPtr iterator(client_->getIterator(Name("example.org")));
 }
 
 TEST_F(MemoryClientTest, getIteratorForNonExistentZone) {