aboutsummaryrefslogtreecommitdiff
path: root/vmime-master/tests/testUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'vmime-master/tests/testUtils.cpp')
-rw-r--r--vmime-master/tests/testUtils.cpp404
1 files changed, 404 insertions, 0 deletions
diff --git a/vmime-master/tests/testUtils.cpp b/vmime-master/tests/testUtils.cpp
new file mode 100644
index 0000000..ae75547
--- /dev/null
+++ b/vmime-master/tests/testUtils.cpp
@@ -0,0 +1,404 @@
+//
+// 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 "testUtils.hpp"
+
+#include "vmime/utility/stringUtils.hpp"
+
+#include <cstring>
+#include <iostream>
+
+
+// Enable to output socket send/receive on standard output
+#define DEBUG_SOCKET_IN_OUT 0
+
+
+
+// testSocket
+
+void testSocket::connect(const vmime::string& address, const vmime::port_t port) {
+
+ m_address = address;
+ m_port = port;
+ m_connected = true;
+
+ onConnected();
+}
+
+
+void testSocket::disconnect() {
+
+ m_address.clear();
+ m_port = 0;
+ m_connected = false;
+}
+
+
+bool testSocket::isConnected() const {
+
+ return m_connected;
+}
+
+
+vmime::size_t testSocket::getBlockSize() const {
+
+ return 16384;
+}
+
+
+unsigned int testSocket::getStatus() const {
+
+ return 0;
+}
+
+
+const vmime::string testSocket::getPeerName() const {
+
+ return "test.vmime.org";
+}
+
+
+const vmime::string testSocket::getPeerAddress() const {
+
+ return "127.0.0.1";
+}
+
+
+vmime::shared_ptr <vmime::net::timeoutHandler> testSocket::getTimeoutHandler() {
+
+ return vmime::null;
+}
+
+
+void testSocket::setTracer(const vmime::shared_ptr <vmime::net::tracer>& /* tracer */) {
+
+}
+
+
+vmime::shared_ptr <vmime::net::tracer> testSocket::getTracer() {
+
+ return vmime::null;
+}
+
+
+bool testSocket::waitForRead(const int /* msecs */) {
+
+ return true;
+}
+
+
+bool testSocket::waitForWrite(const int /* msecs */) {
+
+ return true;
+}
+
+
+void testSocket::receive(vmime::string& buffer) {
+
+ buffer = m_inBuffer;
+ m_inBuffer.clear();
+}
+
+
+void testSocket::send(const vmime::string& buffer) {
+
+ m_outBuffer += buffer;
+
+ onDataReceived();
+}
+
+
+void testSocket::send(const char* str) {
+
+ sendRaw(reinterpret_cast <const vmime::byte_t*>(str), strlen(str));
+}
+
+
+vmime::size_t testSocket::receiveRaw(vmime::byte_t* buffer, const size_t count) {
+
+ const size_t n = std::min(count, static_cast <size_t>(m_inBuffer.size()));
+
+ std::copy(m_inBuffer.begin(), m_inBuffer.begin() + n, buffer);
+ m_inBuffer.erase(m_inBuffer.begin(), m_inBuffer.begin() + n);
+
+ return n;
+}
+
+
+void testSocket::sendRaw(const vmime::byte_t* buffer, const size_t count) {
+
+ send(vmime::utility::stringUtils::makeStringFromBytes(buffer, count));
+}
+
+
+vmime::size_t testSocket::sendRawNonBlocking(const vmime::byte_t* buffer, const size_t count) {
+
+ sendRaw(buffer, count);
+ return count;
+}
+
+
+void testSocket::localSend(const vmime::string& buffer) {
+
+ m_inBuffer += buffer;
+
+#if DEBUG_SOCKET_IN_OUT
+ std::cout << "> " << vmime::utility::stringUtils::trim(buffer) << std::endl;
+#endif // DEBUG_SOCKET_IN_OUT
+
+}
+
+
+void testSocket::localReceive(vmime::string& buffer) {
+
+ buffer = m_outBuffer;
+ m_outBuffer.clear();
+}
+
+
+bool testSocket::localReceiveLine(vmime::string& line) {
+
+ vmime::size_t eol;
+
+ if ((eol = m_outBuffer.find('\n')) != vmime::string::npos) {
+
+ line = vmime::string(m_outBuffer.begin(), m_outBuffer.begin() + eol);
+
+ if (!line.empty() && line[line.length() - 1] == '\r') {
+ line.erase(line.end() - 1, line.end());
+ }
+
+ m_outBuffer.erase(m_outBuffer.begin(), m_outBuffer.begin() + eol + 1);
+
+ return true;
+ }
+
+ return false;
+}
+
+
+vmime::size_t testSocket::localReceiveRaw(vmime::byte_t* buffer, const size_t count) {
+
+ const size_t received = std::min(count, static_cast <size_t>(m_outBuffer.size()));
+
+ if (received != 0) {
+
+ if (buffer) {
+ std::copy(m_outBuffer.begin(), m_outBuffer.begin() + received, buffer);
+ }
+
+ m_outBuffer.erase(m_outBuffer.begin(), m_outBuffer.begin() + received);
+ }
+
+ return received;
+}
+
+
+void testSocket::onDataReceived() {
+
+ // Override
+}
+
+
+void testSocket::onConnected() {
+
+ // Override
+}
+
+
+// lineBasedTestSocket
+
+void lineBasedTestSocket::onDataReceived() {
+
+ vmime::string chunk;
+ localReceive(chunk);
+
+ m_buffer += chunk;
+
+ vmime::size_t eol;
+
+ while ((eol = m_buffer.find('\n')) != vmime::string::npos) {
+
+ vmime::string line(std::string(m_buffer.begin(), m_buffer.begin() + eol));
+
+ if (!line.empty() && line[line.length() - 1] == '\r') {
+ line.erase(line.end() - 1, line.end());
+ }
+
+#if DEBUG_SOCKET_IN_OUT
+ std::cout << "< " << vmime::utility::stringUtils::trim(line) << std::endl;
+#endif // DEBUG_SOCKET_IN_OUT
+
+ m_lines.push_back(line);
+ m_buffer.erase(m_buffer.begin(), m_buffer.begin() + eol + 1);
+ }
+
+ while (!m_lines.empty()) {
+ processCommand();
+ }
+}
+
+
+const vmime::string lineBasedTestSocket::getNextLine() {
+
+ const vmime::string line = m_lines.front();
+ m_lines.erase(m_lines.begin(), m_lines.begin() + 1);
+ return line;
+}
+
+
+bool lineBasedTestSocket::haveMoreLines() const {
+
+ return !m_lines.empty();
+}
+
+
+// testTimeoutHandler
+
+testTimeoutHandler::testTimeoutHandler(const unsigned long delay)
+ : m_delay(delay),
+ m_start(0) {
+
+}
+
+
+bool testTimeoutHandler::isTimeOut() {
+
+ return (vmime::platform::getHandler()->getUnixTime() - m_start) >= m_delay;
+}
+
+
+void testTimeoutHandler::resetTimeOut() {
+
+ m_start = vmime::platform::getHandler()->getUnixTime();
+}
+
+
+bool testTimeoutHandler::handleTimeOut() {
+
+ return false;
+}
+
+
+// testTimeoutHandlerFactory : public vmime::net::timeoutHandlerFactory
+
+vmime::shared_ptr <vmime::net::timeoutHandler> testTimeoutHandlerFactory::create() {
+
+ return vmime::make_shared <testTimeoutHandler>();
+}
+
+
+
+// Exception helper
+std::ostream& operator<<(std::ostream& os, const vmime::exception& e) {
+
+ os << "* vmime::exceptions::" << e.name() << std::endl;
+ os << " what = " << e.what() << std::endl;
+
+ // More information for special exceptions
+ if (dynamic_cast <const vmime::exceptions::command_error*>(&e)) {
+
+ const vmime::exceptions::command_error& cee =
+ dynamic_cast <const vmime::exceptions::command_error&>(e);
+
+ os << " command = " << cee.command() << std::endl;
+ os << " response = " << cee.response() << std::endl;
+ }
+
+ if (dynamic_cast <const vmime::exceptions::invalid_response*>(&e)) {
+
+ const vmime::exceptions::invalid_response& ir =
+ dynamic_cast <const vmime::exceptions::invalid_response&>(e);
+
+ os << " response = " << ir.response() << std::endl;
+ }
+
+ if (dynamic_cast <const vmime::exceptions::connection_greeting_error*>(&e)) {
+
+ const vmime::exceptions::connection_greeting_error& cgee =
+ dynamic_cast <const vmime::exceptions::connection_greeting_error&>(e);
+
+ os << " response = " << cgee.response() << std::endl;
+ }
+
+ if (dynamic_cast <const vmime::exceptions::authentication_error*>(&e)) {
+
+ const vmime::exceptions::authentication_error& aee =
+ dynamic_cast <const vmime::exceptions::authentication_error&>(e);
+
+ os << " response = " << aee.response() << std::endl;
+ }
+
+ if (dynamic_cast <const vmime::exceptions::filesystem_exception*>(&e)) {
+
+ const vmime::exceptions::filesystem_exception& fse =
+ dynamic_cast <const vmime::exceptions::filesystem_exception&>(e);
+
+ os << " path = " << vmime::platform::getHandler()->
+ getFileSystemFactory()->pathToString(fse.path()) << std::endl;
+ }
+
+ if (e.other()) {
+ os << *e.other();
+ }
+
+ return os;
+}
+
+
+const vmime::string toHex(const vmime::string str) {
+
+ static const char hexChars[] = "0123456789abcdef";
+
+ vmime::string res = "\n";
+
+ for (size_t i = 0 ; i < str.length() ; i += 16) {
+
+ size_t r = std::min(static_cast <size_t>(16), str.length() - i);
+
+ vmime::string hex;
+ vmime::string chr;
+
+ for (size_t j = 0 ; j < r ; ++j) {
+
+ const unsigned char c = str[i + j];
+
+ hex += hexChars[c / 16];
+ hex += hexChars[c % 16];
+ hex += " ";
+
+ if (c >= 32 && c <= 127)
+ chr += c;
+ else
+ chr += '.';
+ }
+
+ for (size_t j = r ; j < 16 ; ++j) {
+ hex += " ";
+ }
+
+ res += hex + " " + chr + "\n";
+ }
+
+ return res;
+}