aria2/test/DefaultPeerStorageTest.cc

#include "DefaultPeerStorage.h"
#include "DefaultBtContext.h"
#include "Util.h"
#include "Exception.h"
#include <cppunit/extensions/HelperMacros.h>

using namespace std;

class DefaultPeerStorageTest:public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(DefaultPeerStorageTest);
  CPPUNIT_TEST(testCountPeer);
  CPPUNIT_TEST(testDeleteUnusedPeer);
  CPPUNIT_TEST(testAddPeer);
  CPPUNIT_TEST(testGetPeer);
  CPPUNIT_TEST(testIsPeerAvailable);
  CPPUNIT_TEST(testActivatePeer);
  CPPUNIT_TEST(testCalculateStat);
  CPPUNIT_TEST(testAddIncomingPeer);
  CPPUNIT_TEST(testReturnPeer);
  CPPUNIT_TEST(testOnErasingPeer);
  CPPUNIT_TEST(testReturnPeer);
  CPPUNIT_TEST_SUITE_END();
private:
  BtContextHandle btContext;
  BtRuntimeHandle btRuntime;
  Option* option;
public:
  DefaultPeerStorageTest():btContext(0) {}

  void setUp() {
    btContext = BtContextHandle(new DefaultBtContext());
    btContext->load("test.torrent");
    option = new Option();
    btRuntime = BtRuntimeHandle(new BtRuntime());
  }

  void testCountPeer();
  void testDeleteUnusedPeer();
  void testAddPeer();
  void testGetPeer();
  void testIsPeerAvailable();
  void testActivatePeer();
  void testCalculateStat();
  void testAddIncomingPeer();
  void testReturnPeer();
  void testOnErasingPeer();
};


CPPUNIT_TEST_SUITE_REGISTRATION(DefaultPeerStorageTest);

void DefaultPeerStorageTest::testCountPeer() {
  DefaultPeerStorage ps(btContext, option);

  CPPUNIT_ASSERT_EQUAL(0,
		       ps.countPeer());

  PeerHandle peer(new Peer("192.168.0.1", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  ps.addPeer(peer);
  CPPUNIT_ASSERT_EQUAL(1,
		       ps.countPeer());
}

void DefaultPeerStorageTest::testDeleteUnusedPeer() {
  DefaultPeerStorage ps(btContext, option);

  PeerHandle peer1(new Peer("192.168.0.1", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  PeerHandle peer2(new Peer("192.168.0.2", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  PeerHandle peer3(new Peer("192.168.0.3", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));

  ps.addPeer(peer1);
  ps.addPeer(peer2);
  ps.addPeer(peer3);

  ps.deleteUnusedPeer(2);

  CPPUNIT_ASSERT_EQUAL(1, ps.countPeer());
  CPPUNIT_ASSERT_EQUAL(string("192.168.0.3"),
		       ps.getPeer("192.168.0.3", 6889)->ipaddr);

  ps.addPeer(peer1);
  ps.addPeer(peer2);

  peer2->cuid = 1;

  ps.deleteUnusedPeer(3);

  // peer2 has been in use, so it did't deleted.
  CPPUNIT_ASSERT_EQUAL(1, ps.countPeer());
  CPPUNIT_ASSERT_EQUAL(string("192.168.0.2"),
		       ps.getPeer("192.168.0.2", 6889)->ipaddr);
  
}

void DefaultPeerStorageTest::testAddPeer() {
  DefaultPeerStorage ps(btContext, option);

  PeerHandle peer1(new Peer("192.168.0.1", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  PeerHandle peer2(new Peer("192.168.0.2", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  PeerHandle peer3(new Peer("192.168.0.3", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));

  ps.addPeer(peer1);
  ps.addPeer(peer2);
  ps.addPeer(peer3);

  CPPUNIT_ASSERT_EQUAL(3, ps.countPeer());

  // this is true, because peer1 in the container has no errors and
  // it is replaced by peer1(self assignment).
  CPPUNIT_ASSERT_EQUAL(true, ps.addPeer(peer1));
  // the number of peers doesn't change.
  CPPUNIT_ASSERT_EQUAL(3, ps.countPeer());

  ps.setMaxPeerListSize(3);

  PeerHandle peer4(new Peer("192.168.0.4", 6889, btContext->getPieceLength(),
			    btContext->getTotalLength()));

  CPPUNIT_ASSERT(ps.addPeer(peer4));
  // peer1 was deleted.
  CPPUNIT_ASSERT_EQUAL(3, ps.countPeer());
  
  peer4->startBadCondition();

  PeerHandle peer5(new Peer("192.168.0.4", 6889, btContext->getPieceLength(),
			    btContext->getTotalLength()));

  // this returns false, because peer4 in the container has error.
  CPPUNIT_ASSERT_EQUAL(false, ps.addPeer(peer5));

  peer3->cuid = 1;

  PeerHandle peer6(new Peer("192.168.0.3", 6889, btContext->getPieceLength(),
			    btContext->getTotalLength()));

  // this is false, because peer3's cuid is not zero
  CPPUNIT_ASSERT_EQUAL(false, ps.addPeer(peer6));
  
}

void DefaultPeerStorageTest::testGetPeer() {
  DefaultPeerStorage ps(btContext, option);
  ps.setBtRuntime(btRuntime);

  PeerHandle peer1(new Peer("192.168.0.1", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));

  ps.addPeer(peer1);

  CPPUNIT_ASSERT_EQUAL(string("192.168.0.1"),
		       ps.getUnusedPeer()->ipaddr);

  peer1->cuid = 1;

  CPPUNIT_ASSERT(ps.getUnusedPeer().isNull());

  peer1->resetStatus();
  peer1->startBadCondition();

  CPPUNIT_ASSERT(ps.getUnusedPeer().isNull());
}

void DefaultPeerStorageTest::testIsPeerAvailable() {
  DefaultPeerStorage ps(btContext, option);
  ps.setBtRuntime(btRuntime);

  CPPUNIT_ASSERT_EQUAL(false, ps.isPeerAvailable());

  PeerHandle peer1(new Peer("192.168.0.1", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));

  ps.addPeer(peer1);

  CPPUNIT_ASSERT_EQUAL(true, ps.isPeerAvailable());

  peer1->cuid = 1;

  CPPUNIT_ASSERT_EQUAL(false, ps.isPeerAvailable());

  peer1->resetStatus();

  peer1->startBadCondition();

  CPPUNIT_ASSERT_EQUAL(false, ps.isPeerAvailable());
}

void DefaultPeerStorageTest::testActivatePeer() {
  DefaultPeerStorage ps(btContext, option);

  CPPUNIT_ASSERT_EQUAL((size_t)0, ps.getActivePeers().size());

  PeerHandle peer1(new Peer("192.168.0.1", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));

  ps.addPeer(peer1);

  Peers activePeer = ps.getActivePeers();

  CPPUNIT_ASSERT_EQUAL((size_t)0, ps.getActivePeers().size());

  peer1->activate();

  CPPUNIT_ASSERT_EQUAL((size_t)1, ps.getActivePeers().size());
}

void DefaultPeerStorageTest::testCalculateStat() {
}


void DefaultPeerStorageTest::testAddIncomingPeer()
{
  DefaultPeerStorage ps(btContext, option);
 
  CPPUNIT_ASSERT(ps.addIncomingPeer(new Peer("192.168.0.1", 6889,
					     btContext->getPieceLength(),
					     btContext->getTotalLength())));

  CPPUNIT_ASSERT_EQUAL((size_t)1, ps.getIncomingPeers().size());
}

void DefaultPeerStorageTest::testReturnPeer()
{
  DefaultPeerStorage ps(btContext, option);

  PeerHandle peer1(new Peer("192.168.0.1", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  PeerHandle peer2(new Peer("192.168.0.2", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  ps.addPeer(peer1);
  ps.addPeer(peer2);

  PeerHandle peer3(new Peer("192.168.0.3", 6889, btContext->getPieceLength(),
			   btContext->getTotalLength()));
  
  ps.addIncomingPeer(peer3);

  CPPUNIT_ASSERT_EQUAL(string("192.168.0.2"),
		       ps.getPeers().front()->ipaddr);
  ps.returnPeer(peer2);
  CPPUNIT_ASSERT_EQUAL(string("192.168.0.1"),
		       ps.getPeers().front()->ipaddr);

  ps.returnPeer(peer3);
  CPPUNIT_ASSERT_EQUAL((size_t)0, ps.getIncomingPeers().size());

}

void DefaultPeerStorageTest::testOnErasingPeer()
{
  // test this
}