ferencd@0: //
ferencd@0: // VMime library (http://www.vmime.org)
ferencd@0: // Copyright (C) 2002-2013 Vincent Richard <vincent@vmime.org>
ferencd@0: //
ferencd@0: // This program is free software; you can redistribute it and/or
ferencd@0: // modify it under the terms of the GNU General Public License as
ferencd@0: // published by the Free Software Foundation; either version 3 of
ferencd@0: // the License, or (at your option) any later version.
ferencd@0: //
ferencd@0: // This program is distributed in the hope that it will be useful,
ferencd@0: // but WITHOUT ANY WARRANTY; without even the implied warranty of
ferencd@0: // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
ferencd@0: // General Public License for more details.
ferencd@0: //
ferencd@0: // You should have received a copy of the GNU General Public License along
ferencd@0: // with this program; if not, write to the Free Software Foundation, Inc.,
ferencd@0: // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
ferencd@0: //
ferencd@0: // Linking this library statically or dynamically with other modules is making
ferencd@0: // a combined work based on this library.  Thus, the terms and conditions of
ferencd@0: // the GNU General Public License cover the whole combination.
ferencd@0: //
ferencd@0: 
ferencd@0: #include "tests/testUtils.hpp"
ferencd@0: 
ferencd@0: #include "vmime/utility/outputStreamAdapter.hpp"
ferencd@0: 
ferencd@0: 
ferencd@0: VMIME_TEST_SUITE_BEGIN(stringContentHandlerTest)
ferencd@0: 
ferencd@0: 	VMIME_TEST_LIST_BEGIN
ferencd@0: 		VMIME_TEST(testIsEmpty)
ferencd@0: 		VMIME_TEST(testGetLength)
ferencd@0: 		VMIME_TEST(testIsEncoded)
ferencd@0: 		VMIME_TEST(testGetLength_Encoded)
ferencd@0: 		VMIME_TEST(testExtract)
ferencd@0: 		VMIME_TEST(testExtract_Encoded)
ferencd@0: 		VMIME_TEST(testExtractRaw_Encoded)
ferencd@0: 		VMIME_TEST(testGenerate)
ferencd@0: 		VMIME_TEST(testGenerate_Encoded)
ferencd@0: 		VMIME_TEST(testStringProxy)
ferencd@0: 	VMIME_TEST_LIST_END
ferencd@0: 
ferencd@0: 
ferencd@0: 	void testIsEmpty()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth;
ferencd@0: 
ferencd@0: 		VASSERT_TRUE("empty", cth.isEmpty());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testGetLength()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth("Test Data");
ferencd@0: 
ferencd@0: 		VASSERT_FALSE("empty", cth.isEmpty());
ferencd@0: 		VASSERT_EQ("length", 9, cth.getLength());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testIsEncoded()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth("Test Data");
ferencd@0: 
ferencd@0: 		VASSERT_FALSE("encoded", cth.isEncoded());
ferencd@0: 		VASSERT_EQ("encoding", vmime::contentHandler::NO_ENCODING, cth.getEncoding());
ferencd@0: 
ferencd@0: 
ferencd@0: 		vmime::stringContentHandler cth2("Zm9vEjRWYmFy=", vmime::encoding("base64"));
ferencd@0: 
ferencd@0: 		VASSERT_TRUE("encoded", cth2.isEncoded());
ferencd@0: 		VASSERT_EQ("encoding", "base64", cth2.getEncoding().generate());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testGetLength_Encoded()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth
ferencd@0: 			("foo=12=34=56bar", vmime::encoding("quoted-printable"));
ferencd@0: 
ferencd@0: 		// Reported length should be the length of encoded data
ferencd@0: 		VASSERT_EQ("length", 15, cth.getLength());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testExtract()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth("Test Data");
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter osa(oss);
ferencd@0: 
ferencd@0: 		cth.extract(osa);
ferencd@0: 
ferencd@0: 		VASSERT_EQ("extract", "Test Data", oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testExtract_Encoded()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth
ferencd@0: 			("QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5ejAxMjM0NTY3ODk=",
ferencd@0: 			 vmime::encoding("base64"));
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter osa(oss);
ferencd@0: 
ferencd@0: 		cth.extract(osa);
ferencd@0: 
ferencd@0: 		// Data should be decoded from B64
ferencd@0: 		VASSERT_EQ("extract",
ferencd@0: 			"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789", oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testExtractRaw_Encoded()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth
ferencd@0: 			("QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5ejAxMjM0NTY3ODk=",
ferencd@0: 			 vmime::encoding("base64"));
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter osa(oss);
ferencd@0: 
ferencd@0: 		cth.extractRaw(osa);
ferencd@0: 
ferencd@0: 		// Data should not be decoded
ferencd@0: 		VASSERT_EQ("extractRaw",
ferencd@0: 			"QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5ejAxMjM0NTY3ODk=", oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testGenerate()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth("foo\x12\x34\x56 bar");
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter osa(oss);
ferencd@0: 
ferencd@0: 		cth.generate(osa, vmime::encoding("base64"));
ferencd@0: 
ferencd@0: 		// Data should be encoded to B64
ferencd@0: 		VASSERT_EQ("generate", "Zm9vEjRWIGJhcg==", oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testGenerate_Encoded()
ferencd@0: 	{
ferencd@0: 		vmime::stringContentHandler cth
ferencd@0: 			("foo=12=34=56bar", vmime::encoding("quoted-printable"));
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter osa(oss);
ferencd@0: 
ferencd@0: 		cth.generate(osa, vmime::encoding("base64"));
ferencd@0: 
ferencd@0: 		// Data should be reencoded from QP to B64
ferencd@0: 		VASSERT_EQ("generate", "Zm9vEjRWYmFy", oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testStringProxy()
ferencd@0: 	{
ferencd@0: 		// With 'stringProxy' object
ferencd@0: 		vmime::utility::stringProxy str("This is the data buffer", 12, 12 + 4);
ferencd@0: 		vmime::stringContentHandler cth(str);
ferencd@0: 
ferencd@0: 		VASSERT_FALSE("empty", cth.isEmpty());
ferencd@0: 		VASSERT_EQ("length", 4, cth.getLength());
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter osa(oss);
ferencd@0: 
ferencd@0: 		cth.extract(osa);
ferencd@0: 
ferencd@0: 		VASSERT_EQ("extract", "data", oss.str());
ferencd@0: 
ferencd@0: 		// Directly with constructor
ferencd@0: 		vmime::stringContentHandler cth2("This is the data buffer", 12, 12 + 4);
ferencd@0: 
ferencd@0: 		VASSERT_FALSE("empty", cth2.isEmpty());
ferencd@0: 		VASSERT_EQ("length", 4, cth2.getLength());
ferencd@0: 
ferencd@0: 		std::ostringstream oss2;
ferencd@0: 		vmime::utility::outputStreamAdapter osa2(oss2);
ferencd@0: 
ferencd@0: 		cth2.extract(osa2);
ferencd@0: 
ferencd@0: 		VASSERT_EQ("extract", "data", oss2.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: VMIME_TEST_SUITE_END