IdentityContentLengthBodyWriterTest.cpp

Go to the documentation of this file.

/*

Copyright (C) 2006-2007 by Peter Dimov.

This file is part of Calitko (http://www.calitko.org).

Calitko 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 2 of the License, or
(at your option) any later version.

Calitko 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 Calitko; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

*/

#include "Qt.h"
#include "../IdentityContentLengthBodyWriter.h"
#include "../BodyEnd.h"
#include "generated/IdentityContentLengthBodyWriterDriver.h"
#include "Imports.cpp"

namespace Protocols {
namespace Http {
namespace Testing {


class IdentityContentLengthBodyWriterTest : public CppUnit::TestFixture
{
    CPPUNIT_TEST_SUITE (IdentityContentLengthBodyWriterTest);
    CPPUNIT_TEST (testWriteBodyAtOnce);
    CPPUNIT_TEST (testWriteBodyFailsFirstButThenSucceeds);
    CPPUNIT_TEST (testWriteBodyInTwoChunksFailsFirstButThenSucceed);
    CPPUNIT_TEST_SUITE_END();

    auto_ptr <TransportMock>                                transport;
    auto_ptr <IdentityContentLengthBodyWriter>              writerReal;
    auto_ptr <IdentityContentLengthBodyWriterDriver>        writer;

    static const int                                        ContentLength = 40;
    const QByteArray                                        chunk1;
    const QByteArray                                        chunk2;
    const QByteArray                                        body;

public:
    IdentityContentLengthBodyWriterTest()
     :  chunk1 (20, '1'), chunk2 (20, '2'), body (chunk1 + chunk2)
    {
    }

    void setUp()
    {
        transport.reset (new TransportMock);
        writerReal.reset (new IdentityContentLengthBodyWriter (ContentLength));
        writer.reset (new IdentityContentLengthBodyWriterDriver (*writerReal));
    }

    void tearDown()
    {
        writer.reset();
        writerReal.reset();
        transport.reset();
    }

    void refWriteBodyChunk (const QByteArray &chunk, bool willSucceed)
    {
        call (writer->write (RawData (chunk), *transport))
            .willCall (transport->write (chunk))
            .willReturn (willSucceed)
        .willReturn (willSucceed);
    }

    void refWriteBodyEnd()
    {
        call (writer->write (BodyEnd(), *transport))
        .willReturn (true);
    }

    void testWriteBodyAtOnce()
    {
        refWriteBodyChunk (body, true);
        refWriteBodyEnd();
    }

    void testWriteBodyFailsFirstButThenSucceeds()
    {
        refWriteBodyChunk (body, false);
        refWriteBodyChunk (body, true);
        refWriteBodyEnd();
    }

    void testWriteBodyInTwoChunksFailsFirstButThenSucceed()
    {
        refWriteBodyChunk (chunk1, false);
        refWriteBodyChunk (chunk1, true);
        refWriteBodyChunk (chunk2, false);
        refWriteBodyChunk (chunk2, true);
        refWriteBodyEnd();
    }
};

CPPUNIT_TEST_SUITE_REGISTRATION (IdentityContentLengthBodyWriterTest);

} // namespace Testing
} // namespace Http
} // namespace Protocols

Copyright © 2006 Peter Dimov.
Powered by Doxygen.