mirror of
https://github.com/aria2/aria2.git
synced 2025-04-06 14:07:37 +03:00
c064a2cd9e
Extract the Peer class's member variables, which are only needed
after
PeerInteractionCommand, into PeerSessionResource class.
This class is instantiated in PeerInteractionCommand class's
ctor and
released in its dtor.
This will make Peer class lightweight and uses less memory for
peers
which are not connected and wait in the queue.
* src/PeerChokeCommand.cc
* src/PeerSessionResource.{h, cc}
* src/PeerInteractionCommand.cc
* src/PeerAbstractCommand.cc: Note: 0 is given to onAbort()
function.
* src/DefaultBtInteractive.cc
* src/BtPieceMessage.cc
* src/BtInterestedMessage.cc
* src/BtUnchokeMessage.cc
* src/DefaultPeerStorage.{h, cc}
* src/PeerInitiateConnectionCommand.cc
* src/ActivePeerConnectionCommand.cc
* src/BtNotInterestedMessage.cc
* src/DefaultBtMessageDispatcher.cc
* src/BtChokeMessage.cc
* src/BtRequestMessage.cc
* src/Peer.{h, cc}
* src/BtRegistry.h
* src/TrackerWatcherCommand.cc
* src/PeerReceiveHandshakeCommand.cc
* test/BtExtendedMessageTest.cc
* test/BtAllowedFastMessageTest.cc
* test/BtCancelMessageTest.cc
* test/DefaultPieceStorageTest.cc
* test/BtBitfieldMessageTest.cc
* test/BtHaveMessageTest.cc
* test/BtNotInterestedMessageTest.cc
* test/BtRequestMessageTest.cc
* test/PeerSessionResourceTest.cc
* test/DefaultBtMessageDispatcherTest.cc
* test/PeerTest.cc
* test/BtInterestedMessageTest.cc
* test/BtRejectMessageTest.cc
* test/BtChokeMessageTest.cc
* test/DefaultPeerStorageTest.cc
* test/BtHaveNoneMessageTest.cc
* test/BtHaveAllMessageTest.cc
* test/DefaultExtensionMessageFactoryTest.cc
* test/BtUnchokeMessageTest.cc
* test/DefaultBtMessageFactoryTest.cc
* test/HandshakeExtensionMessageTest.cc
* test/UTPexExtensionMessageTest.cc
* test/DefaultBtRequestFactoryTest.cc
* test/BtPieceMessageTest.cc
Removed typedef PeerStats.
* src/PeerStat.h
* src/SegmentMan.cc
106 lines
3 KiB
C++
106 lines
3 KiB
C++
#include "BtAllowedFastMessage.h"
|
|
#include "PeerMessageUtil.h"
|
|
#include "Util.h"
|
|
#include "Peer.h"
|
|
#include <cstring>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class BtAllowedFastMessageTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(BtAllowedFastMessageTest);
|
|
CPPUNIT_TEST(testCreate);
|
|
CPPUNIT_TEST(testGetMessage);
|
|
CPPUNIT_TEST(testDoReceivedAction);
|
|
CPPUNIT_TEST(testOnSendComplete);
|
|
CPPUNIT_TEST(testToString);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void setUp() {
|
|
}
|
|
|
|
void testCreate();
|
|
void testGetMessage();
|
|
void testDoReceivedAction();
|
|
void testOnSendComplete();
|
|
void testToString();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(BtAllowedFastMessageTest);
|
|
|
|
void BtAllowedFastMessageTest::testCreate() {
|
|
unsigned char msg[9];
|
|
PeerMessageUtil::createPeerMessageString(msg, sizeof(msg), 5, 17);
|
|
PeerMessageUtil::setIntParam(&msg[5], 12345);
|
|
SharedHandle<BtAllowedFastMessage> pm = BtAllowedFastMessage::create(&msg[4], 5);
|
|
CPPUNIT_ASSERT_EQUAL((int8_t)17, pm->getId());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)12345, pm->getIndex());
|
|
|
|
// case: payload size is wrong
|
|
try {
|
|
unsigned char msg[10];
|
|
PeerMessageUtil::createPeerMessageString(msg, sizeof(msg), 6, 17);
|
|
BtAllowedFastMessage::create(&msg[4], 6);
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(...) {
|
|
}
|
|
// case: id is wrong
|
|
try {
|
|
unsigned char msg[9];
|
|
PeerMessageUtil::createPeerMessageString(msg, sizeof(msg), 5, 18);
|
|
BtAllowedFastMessage::create(&msg[4], 5);
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(...) {
|
|
}
|
|
}
|
|
|
|
void BtAllowedFastMessageTest::testGetMessage() {
|
|
BtAllowedFastMessage msg;
|
|
msg.setIndex(12345);
|
|
unsigned char data[9];
|
|
PeerMessageUtil::createPeerMessageString(data, sizeof(data), 5, 17);
|
|
PeerMessageUtil::setIntParam(&data[5], 12345);
|
|
CPPUNIT_ASSERT(memcmp(msg.getMessage(), data, 9) == 0);
|
|
}
|
|
|
|
void BtAllowedFastMessageTest::testDoReceivedAction() {
|
|
BtAllowedFastMessage msg;
|
|
msg.setIndex(1);
|
|
SharedHandle<Peer> peer = new Peer("localhost", 6969);
|
|
peer->allocateSessionResource(1024, 1024*1024);
|
|
peer->setFastExtensionEnabled(true);
|
|
msg.setPeer(peer);
|
|
CPPUNIT_ASSERT(!peer->isInPeerAllowedIndexSet(1));
|
|
msg.doReceivedAction();
|
|
CPPUNIT_ASSERT(peer->isInPeerAllowedIndexSet(1));
|
|
|
|
peer->setFastExtensionEnabled(false);
|
|
try {
|
|
msg.doReceivedAction();
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(...) {}
|
|
}
|
|
|
|
void BtAllowedFastMessageTest::testOnSendComplete() {
|
|
BtAllowedFastMessage msg;
|
|
msg.setIndex(1);
|
|
SharedHandle<Peer> peer = new Peer("localhost", 6969);
|
|
peer->allocateSessionResource(1024, 1024*1024);
|
|
peer->setFastExtensionEnabled(true);
|
|
msg.setPeer(peer);
|
|
CPPUNIT_ASSERT(!peer->isInAmAllowedIndexSet(1));
|
|
msg.onSendComplete();
|
|
CPPUNIT_ASSERT(peer->isInAmAllowedIndexSet(1));
|
|
}
|
|
|
|
void BtAllowedFastMessageTest::testToString() {
|
|
BtAllowedFastMessage msg;
|
|
msg.setIndex(1);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("allowed fast index=1"), msg.toString());
|
|
}
|
|
|
|
} // namespace aria2
|