Logo Search packages:      
Sourcecode: poco version File versions  Download package

MailMessageTest.cpp

//
// MailMessageTest.cpp
//
// $Id: //poco/1.2/Net/testsuite/src/MailMessageTest.cpp#2 $
//
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
// 
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//


#include "MailMessageTest.h"
#include "CppUnit/TestCaller.h"
#include "CppUnit/TestSuite.h"
#include "Poco/Net/MailMessage.h"
#include "Poco/Net/MailRecipient.h"
#include "Poco/Net/PartHandler.h"
#include "Poco/Net/StringPartSource.h"
#include "Poco/Net/MediaType.h"
#include "Poco/Timestamp.h"
#include <sstream>
#include <vector>


using Poco::Net::MailMessage;
using Poco::Net::MailRecipient;
using Poco::Net::MessageHeader;
using Poco::Net::PartHandler;
using Poco::Net::MediaType;
using Poco::Net::StringPartSource;
using Poco::Timestamp;


namespace
{
      class StringPartHandler: public PartHandler
      {
      public:
            StringPartHandler()
            {
            }
            
            void handlePart(const MessageHeader& header, std::istream& stream)
            {
                  _disp.push_back(header["Content-Disposition"]);
                  _type.push_back(header["Content-Type"]);
                  std::string data;
                  int ch = stream.get();
                  while (ch > 0)
                  {
                        data += (char) ch;
                        ch = stream.get();
                  }
                  _data.push_back(data);
            }
            
            const std::vector<std::string>& data() const
            {
                  return _data;
            }

            const std::vector<std::string>& disp() const
            {
                  return _disp;
            }

            const std::vector<std::string>& type() const
            {
                  return _type;
            }
            
      private:
            std::vector<std::string> _data;
            std::vector<std::string> _disp;
            std::vector<std::string> _type;
      };
}


MailMessageTest::MailMessageTest(const std::string& name): CppUnit::TestCase(name)
{
}


MailMessageTest::~MailMessageTest()
{
}


void MailMessageTest::testWriteQP()
{
      MailMessage message;
      MailRecipient r1(MailRecipient::PRIMARY_RECIPIENT, "john.doe@no.where", "John Doe");
      MailRecipient r2(MailRecipient::CC_RECIPIENT, "jane.doe@no.where", "Jane Doe");
      MailRecipient r3(MailRecipient::BCC_RECIPIENT, "walter.foo@no.where", "Frank Foo");
      MailRecipient r4(MailRecipient::BCC_RECIPIENT, "bernie.bar@no.where", "Bernie Bar");
      message.addRecipient(r1);
      message.addRecipient(r2);
      message.addRecipient(r3);
      message.addRecipient(r4);
      message.setSubject("Test Message");
      message.setSender("poco@appinf.com");
      message.setContent(
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
            "To test the quoted-printable encoding, we'll put an extra long line here. This should be enough.\r\n"
            "And here is some more =fe.\r\n"
      );
      Timestamp ts(0);
      message.setDate(ts);
      
      assert (!message.isMultipart());
      
      std::ostringstream str;
      message.write(str);
      std::string s = str.str();
      assert (s == 
            "CC: Jane Doe <jane.doe@no.where>\r\n"
            "Content-Transfer-Encoding: quoted-printable\r\n"
            "Content-Type: text/plain\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>\r\n"
            "\r\n"
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
            "To test the quoted-printable encoding, we'll put an extra long line here. T=\r\n"
            "his should be enough.\r\n"
            "And here is some more =3Dfe.\r\n"
      );
}


void MailMessageTest::testWrite8Bit()
{
      MailMessage message;
      MailRecipient r1(MailRecipient::PRIMARY_RECIPIENT, "john.doe@no.where", "John Doe");
      message.addRecipient(r1);
      message.setSubject("Test Message");
      message.setSender("poco@appinf.com");
      message.setContent(
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n",
            MailMessage::ENCODING_8BIT
      );
      Timestamp ts(0);
      message.setDate(ts);
      
      std::ostringstream str;
      message.write(str);
      std::string s = str.str();
      assert (s == 
            "Content-Transfer-Encoding: 8bit\r\n"
            "Content-Type: text/plain\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>\r\n"
            "\r\n"
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
      );
}


void MailMessageTest::testWriteBase64()
{
      MailMessage message;
      MailRecipient r1(MailRecipient::PRIMARY_RECIPIENT, "john.doe@no.where", "John Doe");
      message.addRecipient(r1);
      message.setSubject("Test Message");
      message.setSender("poco@appinf.com");
      message.setContent(
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n",
            MailMessage::ENCODING_BASE64
      );
      Timestamp ts(0);
      message.setDate(ts);

      std::ostringstream str;
      message.write(str);
      std::string s = str.str();
      assert (s == 
            "Content-Transfer-Encoding: base64\r\n"
            "Content-Type: text/plain\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>\r\n"
            "\r\n"
            "SGVsbG8sIHdvcmxkIQ0KVGhpcyBpcyBhIHRlc3QgZm9yIHRoZSBNYWlsTWVzc2FnZSBjbGFz\r\n"
            "cy4NCg=="
      );
}


void MailMessageTest::testWriteManyRecipients()
{
      MailMessage message;
      MailRecipient r1(MailRecipient::PRIMARY_RECIPIENT, "john.doe@no.where", "John Doe");
      MailRecipient r2(MailRecipient::PRIMARY_RECIPIENT, "jane.doe@no.where", "Jane Doe");
      MailRecipient r3(MailRecipient::PRIMARY_RECIPIENT, "walter.foo@no.where", "Frank Foo");
      MailRecipient r4(MailRecipient::PRIMARY_RECIPIENT, "bernie.bar@no.where", "Bernie Bar");
      MailRecipient r5(MailRecipient::PRIMARY_RECIPIENT, "joe.spammer@no.where", "Joe Spammer");
      message.addRecipient(r1);
      message.addRecipient(r2);
      message.addRecipient(r3);
      message.addRecipient(r4);
      message.addRecipient(r5);
      message.setSubject("Test Message");
      message.setSender("poco@appinf.com");
      message.setContent(
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n",
            MailMessage::ENCODING_8BIT
      );
      Timestamp ts(0);
      message.setDate(ts);
      
      std::ostringstream str;
      message.write(str);
      std::string s = str.str();
      assert (s == 
            "Content-Transfer-Encoding: 8bit\r\n"
            "Content-Type: text/plain\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>, Jane Doe <jane.doe@no.where>, \r\n"
        "\tFrank Foo <walter.foo@no.where>, Bernie Bar <bernie.bar@no.where>, \r\n"
        "\tJoe Spammer <joe.spammer@no.where>\r\n"
            "\r\n"
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
      );
}


void MailMessageTest::testWriteMultiPart()
{
      MailMessage message;
      MailRecipient r1(MailRecipient::PRIMARY_RECIPIENT, "john.doe@no.where", "John Doe");
      message.addRecipient(r1);
      message.setSubject("Test Message");
      message.setSender("poco@appinf.com");
      Timestamp ts(0);
      message.setDate(ts);
      message.addContent(new StringPartSource("Hello World!\r\n", "text/plain"), MailMessage::ENCODING_8BIT);
      message.addAttachment("sample", new StringPartSource("This is some binary data. Really.", "application/octet-stream", "sample.dat"));

      assert (message.isMultipart());

      std::ostringstream str;
      message.write(str);
      std::string s = str.str();
      std::string rawMsg(
            "Content-Type: multipart/mixed; boundary=$\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Mime-Version: 1.0\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>\r\n"
            "\r\n"
            "--$\r\n"
            "Content-Disposition: inline\r\n"
            "Content-Transfer-Encoding: 8bit\r\n"
            "Content-Type: text/plain\r\n"
            "\r\n"
            "Hello World!\r\n"
            "\r\n"
            "--$\r\n"
            "Content-Disposition: attachment; filename=sample.dat\r\n"
            "Content-Transfer-Encoding: base64\r\n"
            "Content-Type: application/octet-stream; name=sample\r\n"
            "\r\n"
            "VGhpcyBpcyBzb21lIGJpbmFyeSBkYXRhLiBSZWFsbHku\r\n"
            "--$--\r\n"
      );
      std::string::size_type p1 = s.find('=') + 1;
      std::string::size_type p2 = s.find('\r');
      std::string boundary(s, p1, p2 - p1);
      std::string msg;
      for (std::string::const_iterator it = rawMsg.begin(); it != rawMsg.end(); ++it)
      {
            if (*it == '$')
                  msg += boundary;
            else
                  msg += *it;
      }
      assert (s == msg);
}


void MailMessageTest::testReadQP()
{
      std::istringstream istr(
            "Content-Transfer-Encoding: quoted-printable\r\n"
            "Content-Type: text/plain\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>\r\n"
            "\r\n"
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
            "To test the quoted-printable encoding, we'll put an extra long line here. T=\r\n"
            "his should be enough.\r\n"
            "And here is some more =3Dfe.\r\n"
      );
      
      MailMessage message;
      message.read(istr);
      
      assert (message.getSender() == "poco@appinf.com");
      assert (message.getContentType() == "text/plain");
      assert (message.getContent() == 
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
            "To test the quoted-printable encoding, we'll put an extra long line here. This should be enough.\r\n"
            "And here is some more =fe.\r\n"
      );
}


void MailMessageTest::testRead8Bit()
{
      std::istringstream istr(
            "Content-Transfer-Encoding: 8bit\r\n"
            "Content-Type: text/plain\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>\r\n"
            "\r\n"
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
      );
      
      MailMessage message;
      message.read(istr);
      
      assert (message.getSender() == "poco@appinf.com");
      assert (message.getContentType() == "text/plain");
      assert (message.getContent() == 
            "Hello, world!\r\n"
            "This is a test for the MailMessage class.\r\n"
      );
}


void MailMessageTest::testReadMultiPart()
{
      std::istringstream istr(
            "Content-Type: multipart/mixed; boundary=MIME_boundary_01234567\r\n"
            "Date: Thu, 1 Jan 1970 00:00:00 GMT\r\n"
            "From: poco@appinf.com\r\n"
            "Mime-Version: 1.0\r\n"
            "Subject: Test Message\r\n"
            "To: John Doe <john.doe@no.where>\r\n"
            "\r\n"
            "\r\n"
            "--MIME_boundary_01234567\r\n"
            "Content-Disposition: inline\r\n"
            "Content-Transfer-Encoding: 8bit\r\n"
            "Content-Type: text/plain\r\n"
            "\r\n"
            "Hello World!\r\n"
            "\r\n"
            "--MIME_boundary_01234567\r\n"
            "Content-Disposition: attachment; filename=sample.dat\r\n"
            "Content-Transfer-Encoding: base64\r\n"
            "Content-Type: application/octet-stream; name=sample\r\n"
            "\r\n"
            "VGhpcyBpcyBzb21lIGJpbmFyeSBkYXRhLiBSZWFsbHku\r\n"
            "--MIME_boundary_01234567--\r\n"
      );
      
      StringPartHandler handler;
      MailMessage message;
      message.read(istr, handler);
      
      assert (handler.data().size() == 2);
      assert (handler.data()[0] == "Hello World!\r\n");
      assert (handler.type()[0] == "text/plain");
      assert (handler.disp()[0] == "inline");

      assert (handler.data()[1] == "This is some binary data. Really.");
      assert (handler.type()[1] == "application/octet-stream; name=sample");
      assert (handler.disp()[1] == "attachment; filename=sample.dat");
}


void MailMessageTest::setUp()
{
}


void MailMessageTest::tearDown()
{
}


CppUnit::Test* MailMessageTest::suite()
{
      CppUnit::TestSuite* pSuite = new CppUnit::TestSuite("MailMessageTest");

      CppUnit_addTest(pSuite, MailMessageTest, testWriteQP);
      CppUnit_addTest(pSuite, MailMessageTest, testWrite8Bit);
      CppUnit_addTest(pSuite, MailMessageTest, testWriteBase64);
      CppUnit_addTest(pSuite, MailMessageTest, testWriteManyRecipients);
      CppUnit_addTest(pSuite, MailMessageTest, testWriteMultiPart);
      CppUnit_addTest(pSuite, MailMessageTest, testReadQP);
      CppUnit_addTest(pSuite, MailMessageTest, testRead8Bit);
      CppUnit_addTest(pSuite, MailMessageTest, testReadMultiPart);

      return pSuite;
}

Generated by  Doxygen 1.6.0   Back to index