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/charset.hpp"
ferencd@0: #include "vmime/charsetConverter.hpp"
ferencd@0: 
ferencd@0: #include "charsetTestSuites.hpp"
ferencd@0: 
ferencd@0: 
ferencd@0: VMIME_TEST_SUITE_BEGIN(charsetFilteredOutputStreamTest)
ferencd@0: 
ferencd@0: 	VMIME_TEST_LIST_BEGIN
ferencd@0: 		VMIME_TEST(testInputBufferUnderflow)
ferencd@0: 		VMIME_TEST(testInvalidInput1)
ferencd@0: 		VMIME_TEST(testStreamCopy)
ferencd@0: 		VMIME_TEST(testOneByteAtTime)
ferencd@0: 		VMIME_TEST(testVariableInputChunk)
ferencd@0: 	VMIME_TEST_LIST_END
ferencd@0: 
ferencd@0: 
ferencd@0: 	void testInputBufferUnderflow()
ferencd@0: 	{
ferencd@0: 		vmime::shared_ptr <vmime::charsetConverter> cc =
ferencd@0: 			vmime::charsetConverter::create("utf-8", "iso-8859-1");
ferencd@0: 
ferencd@0: 		vmime::string output;
ferencd@0: 		vmime::utility::outputStreamStringAdapter os(output);
ferencd@0: 		vmime::shared_ptr <vmime::utility::filteredOutputStream> cfos = cc->getFilteredOutputStream(os);
ferencd@0: 
ferencd@0: 		VASSERT_NOT_NULL("filteredOutputStream availability", cfos);
ferencd@0: 
ferencd@0: 		// føo = 66 c3 b8 6f [UTF8]
ferencd@0: 		// føo = 66 f8 6f [latin1]
ferencd@0: 
ferencd@0: 		cfos->write("\x66\xc3", 2);
ferencd@0: 
ferencd@0: 		// Incomplete UTF-8 sequence was not converted
ferencd@0: 		VASSERT_EQ("chunk 1", toHex("f"), toHex(output));
ferencd@0: 
ferencd@0: 		// Write second byte of UTF-8 sequence
ferencd@0: 		cfos->write("\xb8\x6f", 2);
ferencd@0: 
ferencd@0: 		VASSERT_EQ("chunk 2", toHex("f\xf8o"), toHex(output));
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testInvalidInput1()
ferencd@0: 	{
ferencd@0: 		vmime::string in("foo\xab\xcd\xef bar");
ferencd@0: 		vmime::string expectedOut("foo??? bar");
ferencd@0: 
ferencd@0: 		vmime::string actualOut;
ferencd@0: 		vmime::utility::outputStreamStringAdapter osa(actualOut);
ferencd@0: 
ferencd@0: 		vmime::shared_ptr <vmime::charsetConverter> conv =
ferencd@0: 			vmime::charsetConverter::create
ferencd@0: 				(vmime::charset("utf-8"),
ferencd@0: 				 vmime::charset("iso-8859-1"));
ferencd@0: 
ferencd@0: 		vmime::shared_ptr <vmime::utility::charsetFilteredOutputStream> os =
ferencd@0: 			conv->getFilteredOutputStream(osa);
ferencd@0: 
ferencd@0: 		VASSERT_NOT_NULL("filteredOutputStream availability", os);
ferencd@0: 
ferencd@0: 		vmime::utility::inputStreamStringAdapter is(in);
ferencd@0: 
ferencd@0: 		vmime::byte_t buffer[16];
ferencd@0: 
ferencd@0: 		for (int i = 0 ; !is.eof() ; ++i)
ferencd@0: 			os->write(buffer, is.read(buffer, 1));
ferencd@0: 
ferencd@0: 		os->flush();
ferencd@0: 
ferencd@0: 		VASSERT_EQ("1", toHex(expectedOut), toHex(actualOut));
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	// Using 'bufferedStreamCopy'
ferencd@0: 	void testStreamCopy()
ferencd@0: 	{
ferencd@0: 		for (unsigned int i = 0 ; i < charsetTestSuitesCount ; ++i)
ferencd@0: 		{
ferencd@0: 			const charsetTestSuiteStruct& entry = charsetTestSuites[i];
ferencd@0: 
ferencd@0: 			std::ostringstream testName;
ferencd@0: 			testName << i << ": " << entry.fromCharset << " -> " << entry.toCharset;
ferencd@0: 
ferencd@0: 			const unsigned long inLength = (entry.fromLength == 0 ? strlen(entry.fromBytes) : entry.fromLength);
ferencd@0: 			vmime::string in(entry.fromBytes, entry.fromBytes + inLength);
ferencd@0: 
ferencd@0: 			const unsigned long outLength = (entry.toLength == 0 ? strlen(entry.toBytes) : entry.toLength);
ferencd@0: 			vmime::string expectedOut(entry.toBytes, entry.toBytes + outLength);
ferencd@0: 
ferencd@0: 			vmime::string actualOut;
ferencd@0: 			vmime::utility::outputStreamStringAdapter osa(actualOut);
ferencd@0: 
ferencd@0: 			vmime::shared_ptr <vmime::charsetConverter> conv =
ferencd@0: 				vmime::charsetConverter::create(entry.fromCharset, entry.toCharset);
ferencd@0: 
ferencd@0: 			vmime::shared_ptr <vmime::utility::charsetFilteredOutputStream> os =
ferencd@0: 				conv->getFilteredOutputStream(osa);
ferencd@0: 
ferencd@0: 			VASSERT_NOT_NULL("filteredOutputStream availability", os);
ferencd@0: 
ferencd@0: 			vmime::utility::inputStreamStringAdapter is(in);
ferencd@0: 
ferencd@0: 			vmime::utility::bufferedStreamCopy(is, *os);
ferencd@0: 
ferencd@0: 			os->flush();
ferencd@0: 
ferencd@0: 			VASSERT_EQ(testName.str(), toHex(expectedOut), toHex(actualOut));
ferencd@0: 		}
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	// One byte at a time
ferencd@0: 	void testOneByteAtTime()
ferencd@0: 	{
ferencd@0: 		for (unsigned int i = 0 ; i < charsetTestSuitesCount ; ++i)
ferencd@0: 		{
ferencd@0: 			const charsetTestSuiteStruct& entry = charsetTestSuites[i];
ferencd@0: 
ferencd@0: 			std::ostringstream testName;
ferencd@0: 			testName << i << ": " << entry.fromCharset << " -> " << entry.toCharset;
ferencd@0: 
ferencd@0: 			const unsigned long inLength = (entry.fromLength == 0 ? strlen(entry.fromBytes) : entry.fromLength);
ferencd@0: 			vmime::string in(entry.fromBytes, entry.fromBytes + inLength);
ferencd@0: 
ferencd@0: 			const unsigned long outLength = (entry.toLength == 0 ? strlen(entry.toBytes) : entry.toLength);
ferencd@0: 			vmime::string expectedOut(entry.toBytes, entry.toBytes + outLength);
ferencd@0: 
ferencd@0: 			vmime::string actualOut;
ferencd@0: 			vmime::utility::outputStreamStringAdapter osa(actualOut);
ferencd@0: 
ferencd@0: 			vmime::shared_ptr <vmime::charsetConverter> conv =
ferencd@0: 				vmime::charsetConverter::create(entry.fromCharset, entry.toCharset);
ferencd@0: 
ferencd@0: 			vmime::shared_ptr <vmime::utility::charsetFilteredOutputStream> os =
ferencd@0: 				conv->getFilteredOutputStream(osa);
ferencd@0: 
ferencd@0: 			VASSERT_NOT_NULL("filteredOutputStream availability", os);
ferencd@0: 
ferencd@0: 			vmime::utility::inputStreamStringAdapter is(in);
ferencd@0: 
ferencd@0: 			vmime::byte_t buffer[16];
ferencd@0: 
ferencd@0: 			for (int i = 0 ; !is.eof() ; ++i)
ferencd@0: 				os->write(buffer, is.read(buffer, 1));
ferencd@0: 
ferencd@0: 			os->flush();
ferencd@0: 
ferencd@0: 			VASSERT_EQ(testName.str(), toHex(expectedOut), toHex(actualOut));
ferencd@0: 		}
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	// Variable chunks
ferencd@0: 	void testVariableInputChunk()
ferencd@0: 	{
ferencd@0: 		for (unsigned int i = 0 ; i < charsetTestSuitesCount ; ++i)
ferencd@0: 		{
ferencd@0: 			const charsetTestSuiteStruct& entry = charsetTestSuites[i];
ferencd@0: 
ferencd@0: 			std::ostringstream testName;
ferencd@0: 			testName << i << ": " << entry.fromCharset << " -> " << entry.toCharset;
ferencd@0: 
ferencd@0: 			const unsigned long inLength = (entry.fromLength == 0 ? strlen(entry.fromBytes) : entry.fromLength);
ferencd@0: 			vmime::string in(entry.fromBytes, entry.fromBytes + inLength);
ferencd@0: 
ferencd@0: 			const unsigned long outLength = (entry.toLength == 0 ? strlen(entry.toBytes) : entry.toLength);
ferencd@0: 			vmime::string expectedOut(entry.toBytes, entry.toBytes + outLength);
ferencd@0: 
ferencd@0: 			vmime::string actualOut;
ferencd@0: 			vmime::utility::outputStreamStringAdapter osa(actualOut);
ferencd@0: 
ferencd@0: 			vmime::shared_ptr <vmime::charsetConverter> conv =
ferencd@0: 				vmime::charsetConverter::create(entry.fromCharset, entry.toCharset);
ferencd@0: 
ferencd@0: 			vmime::shared_ptr <vmime::utility::charsetFilteredOutputStream> os =
ferencd@0: 				conv->getFilteredOutputStream(osa);
ferencd@0: 
ferencd@0: 			VASSERT_NOT_NULL("filteredOutputStream availability", os);
ferencd@0: 
ferencd@0: 			vmime::utility::inputStreamStringAdapter is(in);
ferencd@0: 
ferencd@0: 			vmime::byte_t buffer[16];
ferencd@0: 
ferencd@0: 			for (int i = 0 ; !is.eof() ; ++i)
ferencd@0: 				os->write(buffer, is.read(buffer, (i % 5) + 1));
ferencd@0: 
ferencd@0: 			os->flush();
ferencd@0: 
ferencd@0: 			VASSERT_EQ(testName.str(), toHex(expectedOut), toHex(actualOut));
ferencd@0: 		}
ferencd@0: 	}
ferencd@0: 
ferencd@0: VMIME_TEST_SUITE_END