Add unit test for Stream::ReadLine.

Fixes #5077

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 94c13a626d3e26881cfdc73e6347a566dedfbc35..baee4592b29ec2d11b2115c5487cdd444f8c4e39 100644 (file)
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -20,8 +20,9 @@ include(BoostTestTargets)
 add_boost_test(base
   SOURCES base-array.cpp base-convert.cpp base-dictionary.cpp base-fifo.cpp
           base-match.cpp base-netstring.cpp base-object.cpp base-serialize.cpp
-          base-shellescape.cpp base-stacktrace.cpp base-string.cpp base-timer.cpp
-          base-type.cpp base-value.cpp icinga-perfdata.cpp test.cpp
+          base-shellescape.cpp base-stacktrace.cpp base-stream.cpp
+          base-string.cpp base-timer.cpp base-type.cpp base-value.cpp
+          icinga-perfdata.cpp test.cpp
   LIBRARIES base config icinga
   TESTS base_array/construct
         base_array/getset
@@ -54,6 +55,7 @@ add_boost_test(base
         base_shellescape/escape_basic
         base_shellescape/escape_quoted
         base_stacktrace/stacktrace
+        base_stream/readline_stdio
         base_string/construct
         base_string/equal
         base_string/clear

diff --git a/test/base-stream.cpp b/test/base-stream.cpp
new file mode 100644 (file)
index 0000000..36bdf8f
--- /dev/null
+++ b/test/base-stream.cpp
@@ -0,0 +1,54 @@
+/******************************************************************************
+ * Icinga 2                                                                   *
+ * Copyright (C) 2012-2013 Icinga Development Team (http://www.icinga.org/)   *
+ *                                                                            *
+ * This program is free software; you can redistribute it and/or              *
+ * modify it under the terms of the GNU General Public License                *
+ * as published by the Free Software Foundation; either version 2             *
+ * of the License, or (at your option) any later version.                     *
+ *                                                                            *
+ * This program is distributed in the hope that it will be useful,            *
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of             *
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              *
+ * GNU General Public License for more details.                               *
+ *                                                                            *
+ * You should have received a copy of the GNU General Public License          *
+ * along with this program; if not, write to the Free Software Foundation     *
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.             *
+ ******************************************************************************/
+
+#include "base/stdiostream.h"
+#include <boost/test/unit_test.hpp>
+#include <boost/foreach.hpp>
+#include <sstream>
+
+using namespace icinga;
+
+BOOST_AUTO_TEST_SUITE(base_stream)
+
+BOOST_AUTO_TEST_CASE(readline_stdio)
+{
+       std::stringstream msgbuf;
+       msgbuf << "Hello\nWorld\n\n";
+
+       StdioStream::Ptr stdstream = make_shared<StdioStream>(&msgbuf, false);
+
+       ReadLineContext rlc;
+
+       String line;
+       BOOST_CHECK(stdstream->ReadLine(&line, rlc));
+       BOOST_CHECK(line == "Hello");
+
+       BOOST_CHECK(stdstream->ReadLine(&line, rlc));
+       BOOST_CHECK(line == "World");
+
+       BOOST_CHECK(stdstream->ReadLine(&line, rlc));
+       BOOST_CHECK(line == "");
+
+       BOOST_CHECK(stdstream->ReadLine(&line, rlc));
+       BOOST_CHECK(line == "");
+
+       stdstream->Close();
+}
+
+BOOST_AUTO_TEST_SUITE_END()