ferencd@0: // ferencd@0: // VMime library (http://www.vmime.org) ferencd@0: // Copyright (C) 2002-2013 Vincent Richard 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/net/smtp/SMTPCommandSet.hpp" ferencd@0: #include "vmime/net/smtp/SMTPCommand.hpp" ferencd@0: ferencd@0: ferencd@0: using namespace vmime::net::smtp; ferencd@0: ferencd@0: ferencd@0: VMIME_TEST_SUITE_BEGIN(SMTPCommandSetTest) ferencd@0: ferencd@0: VMIME_TEST_LIST_BEGIN ferencd@0: VMIME_TEST(testCreate) ferencd@0: VMIME_TEST(testCreatePipeline) ferencd@0: VMIME_TEST(testAddCommand) ferencd@0: VMIME_TEST(testAddCommandPipeline) ferencd@0: VMIME_TEST(testWriteToSocket) ferencd@0: VMIME_TEST(testWriteToSocketPipeline) ferencd@0: VMIME_TEST(testGetLastCommandSent) ferencd@0: VMIME_TEST(testGetLastCommandSentPipeline) ferencd@0: VMIME_TEST_LIST_END ferencd@0: ferencd@0: ferencd@0: void testCreate() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ false); ferencd@0: ferencd@0: VASSERT_NOT_NULL("Not null", cset); ferencd@0: VASSERT_FALSE("Finished", cset->isFinished()); ferencd@0: } ferencd@0: ferencd@0: void testCreatePipeline() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ true); ferencd@0: ferencd@0: VASSERT_NOT_NULL("Not null", cset); ferencd@0: VASSERT_FALSE("Finished", cset->isFinished()); ferencd@0: } ferencd@0: ferencd@0: void testAddCommand() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ false); ferencd@0: ferencd@0: VASSERT_NO_THROW("No throw 1", cset->addCommand(SMTPCommand::createCommand("MY_COMMAND1"))); ferencd@0: VASSERT_EQ("Text", "MY_COMMAND1\r\n", cset->getText()); ferencd@0: VASSERT_NO_THROW("No throw 2", cset->addCommand(SMTPCommand::createCommand("MY_COMMAND2"))); ferencd@0: VASSERT_EQ("Text", "MY_COMMAND1\r\nMY_COMMAND2\r\n", cset->getText()); ferencd@0: ferencd@0: vmime::shared_ptr tracer; ferencd@0: vmime::shared_ptr sok = vmime::make_shared (); ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: VASSERT_FALSE("Finished", cset->isFinished()); ferencd@0: ferencd@0: // Can't add commands when writing to socket has started ferencd@0: VASSERT_THROW("Throw", cset->addCommand(SMTPCommand::createCommand("MY_COMMAND3")), std::runtime_error); ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: VASSERT_TRUE("Finished", cset->isFinished()); ferencd@0: } ferencd@0: ferencd@0: void testAddCommandPipeline() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ true); ferencd@0: ferencd@0: VASSERT_NO_THROW("No throw 1", cset->addCommand(SMTPCommand::createCommand("MY_COMMAND1"))); ferencd@0: VASSERT_EQ("Text", "MY_COMMAND1\r\n", cset->getText()); ferencd@0: VASSERT_NO_THROW("No throw 2", cset->addCommand(SMTPCommand::createCommand("MY_COMMAND2"))); ferencd@0: VASSERT_EQ("Text", "MY_COMMAND1\r\nMY_COMMAND2\r\n", cset->getText()); ferencd@0: ferencd@0: vmime::shared_ptr tracer; ferencd@0: vmime::shared_ptr sok = vmime::make_shared (); ferencd@0: vmime::string response; ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: VASSERT_TRUE("Finished", cset->isFinished()); ferencd@0: ferencd@0: sok->localReceive(response); ferencd@0: VASSERT_EQ("Receive cmds", "MY_COMMAND1\r\nMY_COMMAND2\r\n", response); ferencd@0: ferencd@0: // Can't add commands when writing to socket has started ferencd@0: VASSERT_THROW("Throw", cset->addCommand(SMTPCommand::createCommand("MY_COMMAND3")), std::runtime_error); ferencd@0: } ferencd@0: ferencd@0: void testWriteToSocket() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ false); ferencd@0: ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND1")); ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND2")); ferencd@0: ferencd@0: vmime::shared_ptr tracer; ferencd@0: vmime::shared_ptr sok = vmime::make_shared (); ferencd@0: vmime::string response; ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: ferencd@0: sok->localReceive(response); ferencd@0: VASSERT_EQ("Receive cmd 1", "MY_COMMAND1\r\n", response); ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: ferencd@0: sok->localReceive(response); ferencd@0: VASSERT_EQ("Receive cmd 2", "MY_COMMAND2\r\n", response); ferencd@0: } ferencd@0: ferencd@0: void testWriteToSocketPipeline() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ true); ferencd@0: ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND1")); ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND2")); ferencd@0: ferencd@0: vmime::shared_ptr tracer; ferencd@0: vmime::shared_ptr sok = vmime::make_shared (); ferencd@0: vmime::string response; ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: ferencd@0: sok->localReceive(response); ferencd@0: VASSERT_EQ("Receive cmds", "MY_COMMAND1\r\nMY_COMMAND2\r\n", response); ferencd@0: } ferencd@0: ferencd@0: void testGetLastCommandSent() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ false); ferencd@0: ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND1")); ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND2")); ferencd@0: ferencd@0: vmime::shared_ptr tracer; ferencd@0: vmime::shared_ptr sok = vmime::make_shared (); ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: VASSERT_EQ("Cmd 1", "MY_COMMAND1", cset->getLastCommandSent()->getText()); ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: VASSERT_EQ("Cmd 2", "MY_COMMAND2", cset->getLastCommandSent()->getText()); ferencd@0: } ferencd@0: ferencd@0: void testGetLastCommandSentPipeline() ferencd@0: { ferencd@0: vmime::shared_ptr cset = SMTPCommandSet::create(/* pipelining */ true); ferencd@0: ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND1")); ferencd@0: cset->addCommand(SMTPCommand::createCommand("MY_COMMAND2")); ferencd@0: ferencd@0: vmime::shared_ptr tracer; ferencd@0: vmime::shared_ptr sok = vmime::make_shared (); ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: VASSERT_EQ("Cmd 1", "MY_COMMAND1", cset->getLastCommandSent()->getText()); ferencd@0: ferencd@0: cset->writeToSocket(sok, tracer); ferencd@0: VASSERT_EQ("Cmd 2", "MY_COMMAND2", cset->getLastCommandSent()->getText()); ferencd@0: } ferencd@0: ferencd@0: VMIME_TEST_SUITE_END