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(fileContentHandlerTest)
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(testExtract)
ferencd@0: 		VMIME_TEST(testExtractRaw)
ferencd@0: 		VMIME_TEST(testGenerate)
ferencd@0: 	VMIME_TEST_LIST_END
ferencd@0: 
ferencd@0: 
ferencd@0: 	vmime::shared_ptr <vmime::utility::file> testFile;
ferencd@0: 	vmime::string testDataEncoded, testDataDecoded;
ferencd@0: 
ferencd@0: 
ferencd@0: 	void setUp()
ferencd@0: 	{
ferencd@0: 		testDataDecoded = "ABCDEFGHIJKLMNOPQRSTUVWXYZ \x12\x34\x56\x78\x90 abcdefghijklmnopqrstuvwxyz0123456789";
ferencd@0: 		testDataEncoded = "QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVogEjRWeJAgYWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXowMTIzNDU2Nzg5";
ferencd@0: 
ferencd@0: 		std::ostringstream testFilePath;
ferencd@0: 		testFilePath << "/tmp/vmime_test_" << (rand() % 999999999);
ferencd@0: 
ferencd@0: 		vmime::shared_ptr <vmime::utility::fileSystemFactory> fsf =
ferencd@0: 			vmime::platform::getHandler()->getFileSystemFactory();
ferencd@0: 
ferencd@0: 		testFile = fsf->create(fsf->stringToPath(testFilePath.str()));
ferencd@0: 		testFile->createFile();
ferencd@0: 		testFile->getFileWriter()->getOutputStream()->write(testDataEncoded.data(), testDataEncoded.length());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void tearDown()
ferencd@0: 	{
ferencd@0: 		testFile->remove();
ferencd@0: 		testFile = vmime::null;
ferencd@0: 	}
ferencd@0: 
ferencd@0: 
ferencd@0: 	void testIsEmpty()
ferencd@0: 	{
ferencd@0: 		vmime::fileContentHandler cth;
ferencd@0: 
ferencd@0: 		VASSERT_TRUE("empty", cth.isEmpty());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testGetLength()
ferencd@0: 	{
ferencd@0: 		vmime::fileContentHandler cth(testFile);
ferencd@0: 
ferencd@0: 		VASSERT_FALSE("empty", cth.isEmpty());
ferencd@0: 		VASSERT_EQ("length", testDataEncoded.length(), cth.getLength());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testIsEncoded()
ferencd@0: 	{
ferencd@0: 		vmime::fileContentHandler cth(testFile, vmime::encoding("base64"));
ferencd@0: 
ferencd@0: 		VASSERT_TRUE("encoded", cth.isEncoded());
ferencd@0: 		VASSERT_EQ("encoding", "base64", cth.getEncoding().generate());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testExtract()
ferencd@0: 	{
ferencd@0: 		vmime::fileContentHandler cth(testFile, 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", testDataDecoded, oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testExtractRaw()
ferencd@0: 	{
ferencd@0: 		vmime::fileContentHandler cth(testFile, 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", testDataEncoded, oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	void testGenerate()
ferencd@0: 	{
ferencd@0: 		vmime::fileContentHandler cth(testFile, vmime::encoding("base64"));
ferencd@0: 
ferencd@0: 		std::ostringstream oss;
ferencd@0: 		vmime::utility::outputStreamAdapter osa(oss);
ferencd@0: 
ferencd@0: 		cth.generate(osa, vmime::encoding("quoted-printable"));
ferencd@0: 
ferencd@0: 		// Data should be reencoded from B64 to QP
ferencd@0: 		VASSERT_EQ("generate",
ferencd@0: 			"ABCDEFGHIJKLMNOPQRSTUVWXYZ =124Vx=90 abcdefghijklmnopqrstuvwxyz0123456789", oss.str());
ferencd@0: 	}
ferencd@0: 
ferencd@0: VMIME_TEST_SUITE_END