mirror of
https://github.com/aria2/aria2.git
synced 2026-04-02 10:55:00 +00:00
ba69f5c0c3
Peer list is now divided into 2: unusedPeers_ and usedPeers_. Duplicate check is done using std::set by comparing pair of IP address and port. For this, only IP address and port given to the Peer constructor are used. In other words, TCP port received from extended message is not used for this purpose.
61 lines
1.7 KiB
C++
61 lines
1.7 KiB
C++
#include "Peer.h"
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class PeerTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(PeerTest);
|
|
CPPUNIT_TEST(testPeerAllowedIndexSet);
|
|
CPPUNIT_TEST(testAmAllowedIndexSet);
|
|
CPPUNIT_TEST(testCountSeeder);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
SharedHandle<Peer> peer;
|
|
public:
|
|
void setUp() {
|
|
peer.reset(new Peer("localhost", 6969));
|
|
peer->allocateSessionResource(1024, 1024*1024);
|
|
}
|
|
|
|
void testPeerAllowedIndexSet();
|
|
void testAmAllowedIndexSet();
|
|
void testCountSeeder();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(PeerTest);
|
|
|
|
void PeerTest::testPeerAllowedIndexSet() {
|
|
CPPUNIT_ASSERT(!peer->isInPeerAllowedIndexSet(0));
|
|
peer->addPeerAllowedIndex(0);
|
|
CPPUNIT_ASSERT(peer->isInPeerAllowedIndexSet(0));
|
|
}
|
|
|
|
void PeerTest::testAmAllowedIndexSet() {
|
|
CPPUNIT_ASSERT(!peer->isInAmAllowedIndexSet(0));
|
|
peer->addAmAllowedIndex(0);
|
|
CPPUNIT_ASSERT(peer->isInAmAllowedIndexSet(0));
|
|
}
|
|
|
|
void PeerTest::testCountSeeder()
|
|
{
|
|
std::vector<SharedHandle<Peer> > peers(5);
|
|
peers[0].reset(new Peer("192.168.0.1", 7000));
|
|
peers[1].reset(new Peer("192.168.0.2", 7000));
|
|
peers[2].reset(new Peer("192.168.0.3", 7000));
|
|
peers[3].reset(new Peer("192.168.0.4", 7000));
|
|
peers[4].reset(new Peer("192.168.0.5", 7000));
|
|
for(std::vector<SharedHandle<Peer> >::iterator i = peers.begin();
|
|
i != peers.end(); ++i) {
|
|
(*i)->allocateSessionResource(1024, 1024*8);
|
|
}
|
|
unsigned char bitfield[] = { 0xff };
|
|
peers[1]->setBitfield(bitfield, 1);
|
|
peers[3]->setBitfield(bitfield, 1);
|
|
peers[4]->setBitfield(bitfield, 1);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)3, countSeeder(peers.begin(), peers.end()));
|
|
}
|
|
|
|
} // namespace aria2
|