QueryTest.cpp

Go to the documentation of this file.

/*

Copyright (C) 2005-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 "../Query.h"
#include "Imports.cpp"

namespace Protocols {
namespace Gnutella {
namespace Packets {
namespace Testing {


class QueryTest : public CppUnit::TestFixture
{
    CPPUNIT_TEST_SUITE(QueryTest);
    CPPUNIT_TEST(testPacketProperties);
    CPPUNIT_TEST(testReadPacketHasExtension);
    CPPUNIT_TEST(testReadPacketNoExtension);
    CPPUNIT_TEST(testWritePacketHasExtension);
    CPPUNIT_TEST(testWritePacketNoExtension);
    CPPUNIT_TEST(testReadSetWritePacketOK);
    CPPUNIT_TEST_SUITE_END();

    auto_ptr <Query>    packet;

public:
    void setUp()
    {
        packet.reset (new Query());
    }

    void tearDown()
    {
        packet.reset();
    }


    void scenarioReadPacket (const QByteArray &rawHeader,
                             const QByteArray &rawPayload,
                             bool parsedOk,const MinSpeed &minSpeed,
                             const QByteArray &searchCriteria,
                             const QueryData &extension) {
        CPPUNIT_ASSERT (packet->parse (rawHeader, rawPayload) == parsedOk);

        if (parsedOk) {
            CPPUNIT_ASSERT (packet->minSpeed() == minSpeed);
            CPPUNIT_ASSERT (packet->searchCriteria() == searchCriteria);
            CPPUNIT_ASSERT (packet->extension() == extension);
        }
    }


    void scenarioWritePacket (const MinSpeed &minSpeed,
                              const QByteArray &searchCriteria,
                              const QueryData &extension,
                              const QByteArray &rawHeader,
                              const QByteArray &rawPayload) {
        packet->setMinSpeed (minSpeed);
        packet->setSearchCriteria (searchCriteria);
        packet->setExtension (extension);

        CPPUNIT_ASSERT (packet->rawHeader() == rawHeader);
        CPPUNIT_ASSERT (packet->rawPayload() == rawPayload);
    }


    void scenarioReadSetWritePacket (const QByteArray &rawHeader,
                                     const QByteArray &rawPayload,
                                     const MinSpeed &minSpeed,
                                     const QByteArray &searchCriteria,
                                     const QueryData &extension,
                                     const QByteArray &rawHeader2,
                                     const QByteArray &rawPayload2) {
        CPPUNIT_ASSERT (packet->parse (rawHeader, rawPayload));

        packet->setMinSpeed (minSpeed);
        packet->setSearchCriteria (searchCriteria);
        packet->setExtension (extension);

        CPPUNIT_ASSERT (packet->rawHeader() == rawHeader2);
        CPPUNIT_ASSERT (packet->rawPayload() == rawPayload2);
    }

    void testPacketProperties()
    {
        CPPUNIT_ASSERT (QueryPacket == packet->packetType());
        CPPUNIT_ASSERT (QString ("Query") == packet->name());
    }

    void testReadPacketHasExtension() {
        scenarioReadPacket(
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x80\x7\x8\x14\0\0\0", 23),
            QByteArray ("\xa4\x7c\143alitko\0\074xmltest/\076", 20),
            true,
            MinSpeed(true, false, true, false,false, true, '\x7c'),
            QByteArray("calitko"),
            QueryData::xmlExtension ("\074xmltest/\076"));
    }

    void testReadPacketNoExtension() {
        scenarioReadPacket(
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x80\x7\x8\xa\0\0\0", 23),
            QByteArray ("\xa4\x7c\143alitko\0", 10),
            true,
            MinSpeed(true, false, true, false,false, true, '\x7c'),
            QByteArray("calitko"),
            QueryData());
    }

    void testWritePacketHasExtension() {
        scenarioWritePacket(
            MinSpeed(true, false, true, false,false, true, '\x7c'),
            QByteArray("calitko"),
            QueryData::xmlExtension  ("\074xmltest/\076"),
            packet->rawHeader().replace(19, 4, QByteArray("\x14\0\0\0",4)),
            QByteArray ("\xa4\x7c\143alitko\0\074xmltest/\076", 20));
    }

    void testWritePacketNoExtension() {
        scenarioWritePacket(
            MinSpeed(true, false, true, false,false, true, '\x7c'),
            QByteArray("calitko"),
            QueryData(),
            packet->rawHeader().replace(19, 4, QByteArray("\xa\0\0\0",4)),
            QByteArray ("\xa4\x7c\143alitko\0", 10));
    }

    void testReadSetWritePacketOK() {
        scenarioReadSetWritePacket(
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x80\x7\x8\x14\0\0\0", 23),
            QByteArray ("\xa4\x7c\143alitko\0\074xmltest/\076", 20),
            MinSpeed(true, true, false, false,true, true, '\x3b'),
            QByteArray("peterdemov"),
            QueryData::urnExtension ("urn: hugetest"),
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x80\x7\x8\x1a\0\0\0", 23),
            QByteArray ("\xcc\x3bpeterdemov\0urn: hugetest", 26));
    }
};

CPPUNIT_TEST_SUITE_REGISTRATION(QueryTest);

} // namespace Testing
} // namespace Packets
} // namespace Gnutella
} // namespace Protocols

Copyright © 2006 Peter Dimov.
Powered by Doxygen.