IdentityContentLengthBodyReaderTest.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 "../IdentityContentLengthBodyReader.h"
#include "../BodyEnd.h"
#include "generated/IdentityContentLengthBodyReaderDriver.h"
#include "Imports.cpp"

namespace Protocols {
namespace Http {
namespace Testing {


class IdentityContentLengthBodyReaderTest : public CppUnit::TestFixture
{
    CPPUNIT_TEST_SUITE (IdentityContentLengthBodyReaderTest);
    CPPUNIT_TEST (testReadContentLengthBytesNoneAvailable);
    CPPUNIT_TEST (testReadContentLengthBytesAtOnce);
    CPPUNIT_TEST (testReadContentLengthBytesInTwoReads);
    CPPUNIT_TEST_SUITE_END();

    auto_ptr <TransportMock>                            transport;
    auto_ptr <IdentityContentLengthBodyReader>          bodyReaderReal;
    auto_ptr <IdentityContentLengthBodyReaderDriver>    bodyReader;

    static const int                                    ContentLength = 50;
    QByteArray                                          content;

public:
    void setUp()
    {
        transport.reset (new TransportMock);
        bodyReaderReal.reset (new IdentityContentLengthBodyReader (
                                ContentLength));
        bodyReader.reset (new IdentityContentLengthBodyReaderDriver (
                                *bodyReaderReal));

        content = QByteArray (ContentLength, '3');
    }

    void tearDown()
    {
        bodyReader.reset();
        bodyReaderReal.reset();
        transport.reset();
    }

    void testReadContentLengthBytesNoneAvailable()
    {
        call (bodyReader->read (*transport))
            .willCall (transport->readAtMost (ContentLength))
            .willReturn (QByteArray())
        .willReturn (Data());
    }

    void testReadContentLengthBytesAtOnce()
    {
        call (bodyReader->read (*transport))
            .willCall (transport->readAtMost (ContentLength))
            .willReturn (content)
        .willReturn (Data (RawData (content)));

        call (bodyReader->read (*transport))
        .willReturn (Data (BodyEnd()));
    }

    void testReadContentLengthBytesInTwoReads()
    {
        QByteArray firstPart = content.left (ContentLength / 2);
        QByteArray secondPart = content.mid (ContentLength / 2);

        call (bodyReader->read (*transport))
            .willCall (transport->readAtMost (ContentLength))
            .willReturn (firstPart)
        .willReturn (Data (RawData (firstPart)));

        call (bodyReader->read (*transport))
            .willCall (transport->readAtMost (ContentLength - firstPart.size()))
            .willReturn (secondPart)
        .willReturn (Data (RawData (secondPart)));

        call (bodyReader->read (*transport))
        .willReturn (Data (BodyEnd()));
    }
};

CPPUNIT_TEST_SUITE_REGISTRATION (IdentityContentLengthBodyReaderTest);

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

Copyright © 2006 Peter Dimov.
Powered by Doxygen.