Unit tests for content handlers.

1 files changed, 185 insertions, 0 deletions

diff --git a/tests/parser/streamContentHandlerTest.cpp b/tests/parser/streamContentHandlerTest.cpp
new file mode 100644
index 00000000..75b3d4d4
--- /dev/null
+++ b/tests/parser/streamContentHandlerTest.cpp
@@ -0,0 +1,185 @@
+//
+// VMime library (http://www.vmime.org)
+// Copyright (C) 2002-2013 Vincent Richard <[email protected]>
+//
+// 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 3 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
+//
+// Linking this library statically or dynamically with other modules is making
+// a combined work based on this library.  Thus, the terms and conditions of
+// the GNU General Public License cover the whole combination.
+//
+
+#include "tests/testUtils.hpp"
+
+#include "vmime/utility/outputStreamAdapter.hpp"
+
+
+VMIME_TEST_SUITE_BEGIN(streamContentHandlerTest)
+
+	VMIME_TEST_LIST_BEGIN
+		VMIME_TEST(testIsEmpty)
+		VMIME_TEST(testGetLength)
+		VMIME_TEST(testIsEncoded)
+		VMIME_TEST(testGetLength_Encoded)
+		VMIME_TEST(testExtract)
+		VMIME_TEST(testExtract_Encoded)
+		VMIME_TEST(testExtractRaw_Encoded)
+		VMIME_TEST(testGenerate)
+		VMIME_TEST(testGenerate_Encoded)
+	VMIME_TEST_LIST_END
+
+
+	void testIsEmpty()
+	{
+		vmime::streamContentHandler cth;
+
+		VASSERT_TRUE("empty", cth.isEmpty());
+	}
+
+	void testGetLength()
+	{
+		vmime::string data("Test Data");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length());
+
+		VASSERT_FALSE("empty", cth.isEmpty());
+		VASSERT_EQ("length", 9, cth.getLength());
+	}
+
+	void testIsEncoded()
+	{
+		vmime::string data("Test Data");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length());
+
+		VASSERT_FALSE("encoded", cth.isEncoded());
+		VASSERT_EQ("encoding", vmime::contentHandler::NO_ENCODING, cth.getEncoding());
+
+
+		vmime::string data2("Zm9vEjRWYmFy=");
+		vmime::ref <vmime::utility::inputStream> stream2 =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data2);
+
+		vmime::streamContentHandler cth2(stream2, data2.length(), vmime::encoding("base64"));
+
+		VASSERT_TRUE("encoded", cth2.isEncoded());
+		VASSERT_EQ("encoding", "base64", cth2.getEncoding().generate());
+	}
+
+	void testGetLength_Encoded()
+	{
+		vmime::string data("foo=12=34=56bar");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length(), vmime::encoding("quoted-printable"));
+
+		// Reported length should be the length of encoded data
+		VASSERT_EQ("length", 15, cth.getLength());
+	}
+
+	void testExtract()
+	{
+		vmime::string data("Test Data");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length());
+
+		std::ostringstream oss;
+		vmime::utility::outputStreamAdapter osa(oss);
+
+		cth.extract(osa);
+
+		VASSERT_EQ("extract", "Test Data", oss.str());
+	}
+
+	void testExtract_Encoded()
+	{
+		vmime::string data
+			("QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5ejAxMjM0NTY3ODk=");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length(), vmime::encoding("base64"));
+
+		std::ostringstream oss;
+		vmime::utility::outputStreamAdapter osa(oss);
+
+		cth.extract(osa);
+
+		// Data should be decoded from B64
+		VASSERT_EQ("extract",
+			"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789", oss.str());
+	}
+
+	void testExtractRaw_Encoded()
+	{
+		vmime::string data
+			("QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5ejAxMjM0NTY3ODk=");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length(), vmime::encoding("base64"));
+
+		std::ostringstream oss;
+		vmime::utility::outputStreamAdapter osa(oss);
+
+		cth.extractRaw(osa);
+
+		// Data should not be decoded
+		VASSERT_EQ("extractRaw",
+			"QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5ejAxMjM0NTY3ODk=", oss.str());
+	}
+
+	void testGenerate()
+	{
+		vmime::string data("foo\x12\x34\x56 bar");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length());
+
+		std::ostringstream oss;
+		vmime::utility::outputStreamAdapter osa(oss);
+
+		cth.generate(osa, vmime::encoding("base64"));
+
+		// Data should be encoded to B64
+		VASSERT_EQ("generate", "Zm9vEjRWIGJhcg==", oss.str());
+	}
+
+	void testGenerate_Encoded()
+	{
+		vmime::string data("foo=12=34=56bar");
+		vmime::ref <vmime::utility::inputStream> stream =
+			vmime::create <vmime::utility::inputStreamStringAdapter>(data);
+
+		vmime::streamContentHandler cth(stream, data.length(), vmime::encoding("quoted-printable"));
+
+		std::ostringstream oss;
+		vmime::utility::outputStreamAdapter osa(oss);
+
+		cth.generate(osa, vmime::encoding("base64"));
+
+		// Data should be reencoded from QP to B64
+		VASSERT_EQ("generate", "Zm9vEjRWYmFy", oss.str());
+	}
+
+VMIME_TEST_SUITE_END