TrackerRequestSessionFactoryImplTest.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 "../MultiTrackerRequestSession.h"
#include "../SingleTrackerRequestSession.h"
#include "../TrackerRequestSession.h"
#include "../TrackerRequestSessionFactory.h"
#include "generated/TrackerRequestSessionFactoryImplDriver.h"
#include "generated/TrackerRequestSessionFactoryMock.h"
#include "Imports.cpp"

namespace Protocols {
namespace BitTorrent {
namespace Trackers {
namespace Testing {


class TrackerRequestSessionFactoryImplTest : public CppUnit::TestFixture
{
    CPPUNIT_TEST_SUITE (TrackerRequestSessionFactoryImplTest);
    CPPUNIT_TEST (testCreateSessionTorrentWithAnnounceUrlWillReturnSingleTrackerRequestSession);
    CPPUNIT_TEST (testCreateSessionTorrentWithAnnounceListWillReturnMultiTrackerRequestSession);
    CPPUNIT_TEST (testDestroySessionOnlyOneSessionCreated);
    CPPUNIT_TEST (testDestroySessionTwoSessionsCreated);
    CPPUNIT_TEST (testDestroyAllCreatedSessionsThreeSessionsCreated);
    CPPUNIT_TEST (testTrackerRequestSessionEstablishedNotificationWillBeForwarded);
    CPPUNIT_TEST (trackerRequestSessionErrorNotificationWillBeForwarded);
    CPPUNIT_TEST (trackerRequestSessionResponseRecievedNotificationWillBeForwarded);
    CPPUNIT_TEST (trackerRequestSessionClosingNotificationWillBeForwarded);
    CPPUNIT_TEST (trackerRequestSessionClosedNotificationWillBeForwarded);
    CPPUNIT_TEST_SUITE_END();

    auto_ptr <TrackerRequestSessionFactoryStatusMock>   factoryStatus;
    auto_ptr <TrackerRequestSessionFactoryImpl>         factoryReal;
    auto_ptr <TrackerRequestSessionFactoryImplDriver>   factory;

    Torrent                                             torrentWithAnnounceUrl;
    Torrent                                             torrentWithAnnounceList;
    TrackerResponse                                     trackerResponse;
    QString                                             errorMessage;

public:
    TrackerRequestSessionFactoryImplTest();

    void setUp()
    {
        factoryStatus.reset (new TrackerRequestSessionFactoryStatusMock());
        factoryReal.reset (new TrackerRequestSessionFactoryImpl (
            &factoryStatus->imp));
        factory.reset (new TrackerRequestSessionFactoryImplDriver (
            *factoryReal.get()));
    }

    void tearDown()
    {
        factory.reset();
        factoryReal.reset();
        factoryStatus.reset();
    }

    void testCreateSessionTorrentWithAnnounceUrlWillReturnSingleTrackerRequestSession()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceUrl);
        CPPUNIT_ASSERT (typeid (*session) == typeid (SingleTrackerRequestSession));
    }

    void testCreateSessionTorrentWithAnnounceListWillReturnMultiTrackerRequestSession()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceList);
        CPPUNIT_ASSERT (typeid (*session) == typeid (MultiTrackerRequestSession));
    }

    void testDestroySessionOnlyOneSessionCreated()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceUrl);
        CPPUNIT_ASSERT (factoryReal->holdedSessionsCount() == 1);
        factoryReal->destroySession (session);
        CPPUNIT_ASSERT (factoryReal->holdedSessionsCount() == 0);
    }

    void testDestroySessionTwoSessionsCreated()
    {
        TrackerRequestSession *session1 = factoryReal->createSession (
            torrentWithAnnounceUrl);
        /*TrackerRequestSession *session2 =*/ factoryReal->createSession (
            torrentWithAnnounceList);
        CPPUNIT_ASSERT (factoryReal->holdedSessionsCount() == 2);
        factoryReal->destroySession (session1);
        CPPUNIT_ASSERT (factoryReal->holdedSessionsCount() == 1);
    }

    void testDestroyAllCreatedSessionsThreeSessionsCreated()
    {
        TrackerRequestSession *session1 = factoryReal->createSession (
            torrentWithAnnounceUrl);
        TrackerRequestSession *session2 = factoryReal->createSession (
            torrentWithAnnounceList);
        TrackerRequestSession *session3 = factoryReal->createSession (
            torrentWithAnnounceList);
        CPPUNIT_ASSERT (factoryReal->holdedSessionsCount() == 3);
        factoryReal->destroySession (session1);
        factoryReal->destroySession (session2);
        factoryReal->destroySession (session3);
        CPPUNIT_ASSERT (factoryReal->holdedSessionsCount() == 0);
    }

    void testTrackerRequestSessionEstablishedNotificationWillBeForwarded()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceUrl);

        call (factory->trackerRequestSessionEstablished (session))
            .willCall (factoryStatus->trackerRequestSessionFactorySessionEstablished (
                session))
            .returns()
        .returns();
    }

    void trackerRequestSessionErrorNotificationWillBeForwarded()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceUrl);

        call (factory->trackerRequestSessionError (session, errorMessage))
            .willCall (factoryStatus->trackerRequestSessionFactorySessionError (
                session, errorMessage))
            .returns()
        .returns();
    }

    void trackerRequestSessionResponseRecievedNotificationWillBeForwarded()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceUrl);

        call (factory->trackerRequestSessionResponseRecieved (session,
            trackerResponse))
            .willCall (factoryStatus->trackerRequestSessionFactoryResponseRecieved (
                session, trackerResponse))
            .returns()
        .returns();
    }

    void trackerRequestSessionClosingNotificationWillBeForwarded()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceUrl);

        call (factory->trackerRequestSessionClosing (session))
            .willCall (factoryStatus->trackerRequestSessionFactorySessionClosing (
                session))
            .returns()
        .returns();
    }

    void trackerRequestSessionClosedNotificationWillBeForwarded()
    {
        TrackerRequestSession *session = factoryReal->createSession (
            torrentWithAnnounceUrl);

        call (factory->trackerRequestSessionClosed (session))
            .willCall (factoryStatus->trackerRequestSessionFactorySessionClosed (
                session))
            .returns()
        .returns();
    }
};

CPPUNIT_TEST_SUITE_REGISTRATION(TrackerRequestSessionFactoryImplTest);

TrackerRequestSessionFactoryImplTest::TrackerRequestSessionFactoryImplTest()
    : factoryStatus(),
      factoryReal(),
      factory(),
      torrentWithAnnounceUrl(),
      torrentWithAnnounceList(),
      trackerResponse(),
      errorMessage (QString ("Error"))
{
    const QByteArray rawTorrentWithAnnounceUrlData (
        "d8:announce18:http://tracker.com"
        "4:infod6:lengthi6e4:name8:file.txt"
        "12:piece lengthi32768e6:pieces20:f64hx6dr3cz940fgd52see");
    bool torrentWithAnnounceUrloadedOk = TorrentParser::parseAndLoadTorrent (
        rawTorrentWithAnnounceUrlData, torrentWithAnnounceUrl);
    CPPUNIT_ASSERT (torrentWithAnnounceUrloadedOk);

    const QByteArray rawTorrentWithAnnounceListData (
        "d8:announce18:http://tracker.com"
        "13:announce-listll19:http://tracker1.com19:http://tracker2.comee"
        "4:infod6:lengthi6e4:name8:file.txt"
        "12:piece lengthi32768e6:pieces20:cx33lza06nvgseqwt3b9ee");
    bool torrentWithAnnounceListLoadedOk = TorrentParser::parseAndLoadTorrent (
        rawTorrentWithAnnounceListData, torrentWithAnnounceList);
    CPPUNIT_ASSERT (torrentWithAnnounceListLoadedOk);
}

} // namespace Testing
} // namespace Trackers
} // namespace BitTorrent
} // namespace Protocols

Copyright © 2006 Peter Dimov.
Powered by Doxygen.