From: Mukund Sivaraman <muks@isc.org>
Date: Mon, 29 Oct 2012 04:15:52 +0000 (+0530)
Subject: [2369] Check saveLine() + ungetAll()
X-Git-Tag: trac2487_base~18^2~27
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=2c8d3ac2d8d62ef77c0f888a7c334689ebcd9b5b;p=thirdparty%2Fkea.git

[2369] Check saveLine() + ungetAll()
---

diff --git a/src/lib/dns/tests/inputsource_unittest.cc b/src/lib/dns/tests/inputsource_unittest.cc
index dc0a39a500..4c7bf5c334 100644
--- a/src/lib/dns/tests/inputsource_unittest.cc
+++ b/src/lib/dns/tests/inputsource_unittest.cc
@@ -168,4 +168,36 @@ TEST_F(InputSourceTest, lines) {
     EXPECT_EQ(1, source_.getCurrentLine());
 }
 
+// ungetAll() after saveLine() should skip back to the last-saved place.
+TEST_F(InputSourceTest, saveLine) {
+    // First, skip to line 2.
+    while (!source_.atEOF() &&
+           (source_.getCurrentLine() != 2)) {
+        source_.getChar();
+    }
+    EXPECT_FALSE(source_.atEOF());
+    size_t line = source_.getCurrentLine();
+    EXPECT_EQ(2, line);
+
+    // Now, save the line.
+    source_.saveLine();
+
+    // Now, go to EOF
+    while (!source_.atEOF()) {
+        source_.getChar();
+    }
+    line = source_.getCurrentLine();
+
+    // Now, we are at EOF.
+    EXPECT_TRUE(source_.atEOF());
+    EXPECT_EQ(4, line);
+
+    // Now, ungetAll() and check where it goes back.
+    source_.ungetAll();
+
+    // Now we are back to where we last-saved.
+    EXPECT_EQ(2, source_.getCurrentLine());
+    EXPECT_FALSE(source_.atEOF());
+}
+
 } // end namespace