QueryHitsTest.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 "../QueryHits.h"
#include "Imports.cpp"

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


class QueryHitsTest : public CppUnit::TestFixture
{
    CPPUNIT_TEST_SUITE(QueryHitsTest);
    CPPUNIT_TEST(testPacketProperties);
    CPPUNIT_TEST(testReadPacketOK);
    CPPUNIT_TEST(testReadPacketBAD);
    CPPUNIT_TEST(testWritePacketOK);
    CPPUNIT_TEST(testReadSetWritePacketOK);
    CPPUNIT_TEST_SUITE_END();

    Result                  simpleResult;
    QueryHitsData           simpleQueryHitsData;
    auto_ptr <QueryHits>    packet;

public:
    QueryHitsTest() : simpleResult(), simpleQueryHitsData(), packet()
    {
    }

    void setUp()
    {
        simpleResult.fileIndex = 0x12345678;
        simpleResult.fileSize = 0x23456789;
        simpleResult.fullFileName = QString("TAOCP");

        simpleQueryHitsData.serventId = QUuid (0x01020304, 0x0005, 0x0607,
            '\x8', '\x9', '\xa', '\xb', '\xc', '\xd', '\xe', '\xf');
        packet.reset (new QueryHits());
    }

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


    void scenarioReadPacket (const QByteArray &rawHeader,
                             const QByteArray &rawPayload,
                             bool parsedOk,
                             quint16 port,
                             const QHostAddress &ipAddress,
                             quint32 speed,
                             const ResultSet &resultSet,
                             const QueryHitsData &queryHitsData) {
        CPPUNIT_ASSERT (packet->parse (rawHeader, rawPayload) == parsedOk);

        if (parsedOk) {
            //CPPUNIT_ASSERT (packet->numberOfHits() == numberOfHits);
            CPPUNIT_ASSERT (packet->port() == port);
            CPPUNIT_ASSERT (packet->ipAddress() == ipAddress);
            CPPUNIT_ASSERT (packet->speed() == speed);
            CPPUNIT_ASSERT (packet->resultSet() == resultSet);
            CPPUNIT_ASSERT (packet->queryHitsData() == queryHitsData);
        }
    }


    void scenarioWritePacket (//uchar numberOfHits,
                              quint16 port,
                              const QHostAddress &ipAddress,
                              quint32 speed,
                              const ResultSet &resultSet,
                              const QueryHitsData &queryHitsData,
                              const QByteArray &rawHeader,
                              const QByteArray &rawPayload) {
        //packet->setNumberOfHits (numberOfHits);
        packet->setPort (port);
        packet->setIpAddress (ipAddress);
        packet->setSpeed (speed);
        packet->setResultSet (resultSet);
        packet->setQueryHitsData (queryHitsData);
        CPPUNIT_ASSERT (packet->rawHeader() == rawHeader);
        CPPUNIT_ASSERT (packet->rawPayload() == rawPayload);
    }


    void scenarioReadSetWritePacket (const QByteArray &rawHeader,
                                     const QByteArray &rawPayload,
                                     //uchar numberOfHits,
                                     quint16 port,
                                     const QHostAddress &ipAddress,
                                     quint32 speed,
                                     const ResultSet &resultSet,
                                     const QueryHitsData &queryHitsData,
                                     const QByteArray &rawHeader2,
                                     const QByteArray &rawPayload2) {
        CPPUNIT_ASSERT (packet->parse (rawHeader, rawPayload));

        //packet->setNumberOfHits (numberOfHits);
        packet->setPort (port);
        packet->setIpAddress (ipAddress);
        packet->setSpeed (speed);
        packet->setResultSet (resultSet);
        packet->setQueryHitsData (queryHitsData);

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

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

    void testReadPacketOK() {
        scenarioReadPacket(
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x81\x7\x8\x2a\0\0\0", 23),
            QByteArray ("\x01" "\x67\x45" "\x55\xd6\x3c\xb9" "\x70\x56\x34\x12" "\x78\x56\x34\x12" "\x89\x67\x45\x23" "TAOCP\0" "\0" "UNKN\x02\x3c\x01" "\x1\x2\x3\x4\x0\x5\x6\x7\x8\x9\xa\xb\xc\xd\xe\xf", 42),
            true,
            //1,
            0x4567,
            QHostAddress (0x55d63cb9),
            0x12345670,
            ResultSet() += simpleResult,
            simpleQueryHitsData);
    }

    void testReadPacketBAD() {
        scenarioReadPacket(
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x81\x7\x8\x29\0\0\0", 23),
            QByteArray ("\x01" "\x67\x15" "\x55\xd6\x3c\xb9" "\x70\x56\x34\x12" "\x78\x56\x24\x12" "\x83\x67\x45\x23" "TAOCP\0" "\x1\x2\x3\x4\x0\x5\x6\x7\x8\x9\xa\xb\xc\xd\xe\xf", 41),
            false,
            //1,
            0x4567,
            QHostAddress (0x55d63cb9),
            0x12345670,
            ResultSet() += simpleResult,
            simpleQueryHitsData);
    }

    void testWritePacketOK() {
        scenarioWritePacket(
            //1,
            0x4567,
            QHostAddress(0x55d63cb9),
            0x12345670,
            ResultSet() += simpleResult,
            simpleQueryHitsData,
            packet->rawHeader().replace(19, 4, QByteArray("\x31\0\0\0",4)),
            QByteArray ("\x01" "\x67\x45" "\x55\xd6\x3c\xb9" "\x70\x56\x34\x12" "\x78\x56\x34\x12" "\x89\x67\x45\x23" "TAOCP\0" "\0" "UNKN\x02\x3c\x01" "\x1\x2\x3\x4\x0\x5\x6\x7\x8\x9\xa\xb\xc\xd\xe\xf", 49));
    }

    void testReadSetWritePacketOK() {
        scenarioReadSetWritePacket(
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x81\x7\x8\x29\0\0\0", 23),
            QByteArray ("\x1" "\x07\x45" "\x55\xd2\x3c\xb9" "\x70\x55\x34\x12" "\x78\x56\x24\x12" "\x89\x62\x35\x23" "TAOP\0" "\0" "\x1\x2\x3\x4\x0\x5\x6\x3\x8\x9\xa\xb\x3\xd\xf\xf", 41),
            //1,
            0x4567,
            QHostAddress (0x55d63cb9),
            0x12345670,
            ResultSet() += simpleResult,
            simpleQueryHitsData,
            QByteArray ("\x1\x2\x3\x4\x5\x6\x7\x8\x9\0\x1\x2\x3\x4\x5\x6\x81\x7\x8\x31\0\0\0", 23),
            QByteArray ("\x1" "\x67\x45" "\x55\xd6\x3c\xb9" "\x70\x56\x34\x12" "\x78\x56\x34\x12" "\x89\x67\x45\x23" "TAOCP\0" "\0" "UNKN\x02\x3c\x01" "\x1\x2\x3\x4\x0\x5\x6\x7\x8\x9\xa\xb\xc\xd\xe\xf", 49));
    }
};

CPPUNIT_TEST_SUITE_REGISTRATION(QueryHitsTest);

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

Copyright © 2006 Peter Dimov.
Powered by Doxygen.