1 files changed, 407 insertions, 0 deletions
diff --git a/vmime-master/tests/testUtils.hpp b/vmime-master/tests/testUtils.hpp
new file mode 100644
index 0000000..e6bf1ee
--- /dev/null
+++ b/vmime-master/tests/testUtils.hpp
@@ -0,0 +1,407 @@
+//
+// VMime library (http://www.vmime.org)
+// Copyright (C) 2002 Vincent Richard <vincent@vmime.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 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 <ostream>
+#include <iostream>
+#include <sstream>
+#include <vector>
+#include <string>
+
+
+// VMime
+#include "vmime/vmime.hpp"
+
+
+// CppUnit
+#pragma GCC diagnostic ignored "-Wold-style-cast"
+#include <cppunit/TestAssert.h>
+#include <cppunit/extensions/HelperMacros.h>
+#pragma GCC diagnostic warning "-Wold-style-cast"
+
+#define VASSERT(msg, cond) \
+	CPPUNIT_ASSERT_MESSAGE(std::string(msg), cond)
+
+#define VASSERT_TRUE(msg, cond) \
+	VASSERT(msg, cond)
+#define VASSERT_FALSE(msg, cond) \
+	VASSERT(msg, !(cond))
+
+#define VASSERT_NOT_NULL(msg, cond) \
+	VASSERT(msg, cond != NULL)
+#define VASSERT_NULL(msg, cond) \
+	VASSERT(msg, cond == NULL)
+
+#define VASSERT_EQ(msg, expected, actual) \
+	CPPUNIT_ASSERT_EQUAL_MESSAGE(std::string(msg), expected, actual)
+#define VASSERT_NEQ(msg, expected, actual) \
+	CPPUNIT_ASSERT_MESSAGE(std::string(msg), (expected) != (actual))
+
+#define VASSERT_THROW(msg, expression, exceptionType) \
+	CPPUNIT_ASSERT_THROW(expression, exceptionType)
+#define VASSERT_NO_THROW(msg, expression) \
+	CPPUNIT_ASSERT_NO_THROW(expression)
+
+#define VMIME_TEST_SUITE_BEGIN(testSuiteName) \
+	class testSuiteName; \
+	typedef testSuiteName VMIME_TEST_SUITE; \
+	class testSuiteName : public CppUnit::TestFixture { public:
+#define VMIME_TEST_SUITE_END \
+	}; \
+	\
+	/*static CppUnit::AutoRegisterSuite <VMIME_TEST_SUITE>(autoRegisterRegistry1);*/ \
+	/*static CppUnit::AutoRegisterSuite <VMIME_TEST_SUITE>(autoRegisterRegistry2)(VMIME_TEST_SUITE_MODULE);*/ \
+	extern void registerTestModule(const char* name); \
+	extern const char* getTestModuleNameFromSourceFile(const char *path); \
+	template <typename T> \
+	struct AutoRegisterModule { \
+		AutoRegisterModule() { \
+			static const char* moduleName = getTestModuleNameFromSourceFile(__FILE__); \
+			static CppUnit::AutoRegisterSuite <VMIME_TEST_SUITE>(autoRegisterRegistry1); \
+			static CppUnit::AutoRegisterSuite <VMIME_TEST_SUITE>(autoRegisterRegistry2)(moduleName); \
+			registerTestModule(moduleName); \
+		} \
+	}; \
+	static AutoRegisterModule <VMIME_TEST_SUITE> autoRegisterModule;
+
+#define VMIME_TEST_LIST_BEGIN       CPPUNIT_TEST_SUITE(VMIME_TEST_SUITE);
+#define VMIME_TEST_LIST_END         CPPUNIT_TEST_SUITE_END(); public:
+#define VMIME_TEST(name)            CPPUNIT_TEST(name);
+
+
+namespace CppUnit {
+
+	// Work-around for comparing 'std::string' against 'char*'
+	inline void assertEquals(
+		const char* expected,
+		const std::string actual,
+		SourceLine sourceLine,
+		const std::string &message
+	) {
+
+		assertEquals(std::string(expected), actual, sourceLine, message);
+	}
+
+	template <typename X, typename Y>
+	void assertEquals(
+		const X expected,
+		const Y actual,
+		SourceLine sourceLine,
+		const std::string &message
+	) {
+
+		assertEquals(static_cast <Y>(expected), actual, sourceLine, message);
+	}
+}
+
+
+namespace std {
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::charset& ch) {
+
+	os << "[charset: " << ch.getName() << "]";
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::word& w) {
+
+	os << "[word: charset=" << w.getCharset().getName()
+	   << ", buffer=" << w.getBuffer();
+
+	if (!w.getLanguage().empty()) {
+		os << ", lang=" << w.getLanguage();
+	}
+
+	os << "]";
+
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::text& txt) {
+
+	os << "[text: [";
+
+	for (size_t i = 0 ; i < txt.getWordCount() ; ++i) {
+
+		const vmime::word& w = *txt.getWordAt(i);
+
+		if (i != 0) {
+			os << ",";
+		}
+
+		os << w;
+	}
+
+	os << "]]";
+
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::emailAddress& email) {
+
+	os << email.generate();
+
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::mailbox& mbox) {
+
+	os << "[mailbox: name=" << mbox.getName() << ", email=" << mbox.getEmail() << "]";
+
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::mailboxGroup& group) {
+
+	os << "[mailbox-group: name=" << group.getName() << ", list=[";
+
+	for (size_t i = 0 ; i < group.getMailboxCount() ; ++i) {
+
+		if (i != 0) {
+			os << ",";
+		}
+
+		os << *group.getMailboxAt(i);
+	}
+
+	os << "]]";
+
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::addressList& list) {
+
+	os << "[address-list: [";
+
+	for (size_t i = 0 ; i < list.getAddressCount() ; ++i) {
+
+		const vmime::address& addr = *list.getAddressAt(i);
+
+		if (i != 0) {
+			os << ",";
+		}
+
+		if (addr.isGroup()) {
+
+			const vmime::mailboxGroup& group =
+				dynamic_cast <const vmime::mailboxGroup&>(addr);
+
+			os << group;
+
+		} else {
+
+			const vmime::mailbox& mbox =
+				dynamic_cast <const vmime::mailbox&>(addr);
+
+			os << mbox;
+		}
+	}
+
+	os << "]]";
+
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::datetime& d) {
+
+	os << "[datetime: " << d.getYear() << "/" << d.getMonth() << "/" << d.getDay();
+	os << " " << d.getHour() << ":" << d.getMinute() << ":" << d.getSecond();
+	os << " #" << d.getZone() << "]";
+
+	return os;
+}
+
+
+inline std::ostream& operator<<(std::ostream& os, const vmime::encoding& enc) {
+
+	os << enc.generate();
+
+	return os;
+}
+
+
+}
+
+
+
+// Used to test network features.
+//
+// This works like a local pipe: client reads and writes data using receive()
+// and send(). Server reads incoming data with localReceive() and sends data
+// to client with localSend().
+
+class testSocket : public vmime::net::socket {
+
+public:
+
+	void connect(const vmime::string& address, const vmime::port_t port);
+	void disconnect();
+
+	bool isConnected() const;
+
+	bool waitForWrite(const int msecs = 30000);
+	bool waitForRead(const int msecs = 30000);
+
+	void receive(vmime::string& buffer);
+	void send(const vmime::string& buffer);
+	void send(const char* str);
+
+	size_t receiveRaw(vmime::byte_t* buffer, const size_t count);
+	void sendRaw(const vmime::byte_t* buffer, const size_t count);
+	size_t sendRawNonBlocking(const vmime::byte_t* buffer, const size_t count);
+
+	size_t getBlockSize() const;
+
+	unsigned int getStatus() const;
+
+	const vmime::string getPeerName() const;
+	const vmime::string getPeerAddress() const;
+
+	vmime::shared_ptr <vmime::net::timeoutHandler> getTimeoutHandler();
+
+	void setTracer(const vmime::shared_ptr <vmime::net::tracer>& tracer);
+	vmime::shared_ptr <vmime::net::tracer> getTracer();
+
+	/** Send data to client.
+	  *
+	  * @param buffer data to send
+	  */
+	void localSend(const vmime::string& buffer);
+
+	/** Receive data from client.
+	  *
+	  * @param buffer buffer in which to store received data
+	  */
+	void localReceive(vmime::string& buffer);
+
+	/** Receive a line from client.
+	  *
+	  * @param buffer buffer in which to store received line
+	  * @return true if a line has been read, or false otherwise
+	  */
+	bool localReceiveLine(vmime::string& buffer);
+
+	/** Receive data from client.
+	  *
+	  * @param buffer buffer in which to store received data
+	  * @param count number of bytes to receive
+	  * @return number of bytes received
+	  */
+	vmime::size_t localReceiveRaw(vmime::byte_t* buffer, const size_t count);
+
+protected:
+
+	/** Called when the client has sent some data.
+	  */
+	virtual void onDataReceived();
+
+	/** Called when the client has connected.
+	  */
+	virtual void onConnected();
+
+private:
+
+	vmime::string m_address;
+	vmime::port_t m_port;
+	bool m_connected;
+
+	vmime::string m_inBuffer;
+	vmime::string m_outBuffer;
+};
+
+
+template <typename T>
+class testSocketFactory : public vmime::net::socketFactory {
+
+public:
+
+	vmime::shared_ptr <vmime::net::socket> create() {
+
+		return vmime::make_shared <T>();
+	}
+
+	vmime::shared_ptr <vmime::net::socket> create(
+		const vmime::shared_ptr <vmime::net::timeoutHandler>& /* th */
+	) {
+
+		return vmime::make_shared <T>();
+	}
+};
+
+
+class lineBasedTestSocket : public testSocket {
+
+public:
+
+	void onDataReceived();
+
+	const vmime::string getNextLine();
+	bool haveMoreLines() const;
+
+	virtual void processCommand() = 0;
+
+private:
+
+	std::vector <vmime::string> m_lines;
+	std::string m_buffer;
+};
+
+
+class testTimeoutHandler : public vmime::net::timeoutHandler {
+
+public:
+
+	testTimeoutHandler(const unsigned long delay = 3);
+
+	bool isTimeOut();
+	void resetTimeOut();
+	bool handleTimeOut();
+
+private:
+
+	unsigned long m_delay;
+	unsigned long m_start;
+};
+
+
+class testTimeoutHandlerFactory : public vmime::net::timeoutHandlerFactory {
+
+public:
+
+	vmime::shared_ptr <vmime::net::timeoutHandler> create();
+};
+
+
+// Exception helper
+std::ostream& operator<<(std::ostream& os, const vmime::exception& e);
+
+
+// Conversion to hexadecimal for easier debugging
+const vmime::string toHex(const vmime::string str);