mirror of
https://github.com/aria2/aria2.git
synced 2025-04-06 22:17:38 +03:00
8ba97188ce
Introduced ValueBase class, which is a replacement of BDE. In this change ValueBase is used instead of BDE except DHT messages, UTMetadata messages and XML-RPC. They'll be replaced in the later commits. DownloadContext::_attrs is now ContextAttribute rather than BDE. * src/ActivePeerConnectionCommand.cc * src/AnnounceList.cc * src/AnnounceList.h * src/BtDependency.cc * src/BtRegistry.cc * src/BtSetup.cc * src/ConsoleStatCalc.cc * src/ContextAttribute.h * src/DefaultBtAnnounce.cc * src/DefaultBtInteractive.cc * src/DownloadContext.cc * src/DownloadContext.h * src/HandshakeExtensionMessage.cc * src/InitiateConnectionCommand.cc * src/LpdReceiveMessageCommand.cc * src/MSEHandshake.cc * src/Makefile.am * src/Makefile.in * src/PeerInteractionCommand.cc * src/PeerListProcessor.h * src/ProtocolDetector.cc * src/RequestGroup.cc * src/RequestGroupMan.cc * src/TorrentAttribute.h * src/TrackerWatcherCommand.cc * src/UTMetadataDataExtensionMessage.cc * src/UTMetadataPostDownloadHandler.cc * src/UTMetadataRequestExtensionMessage.cc * src/ValueBase.cc * src/ValueBase.h * src/XmlRpcMethodImpl.cc * src/XmlRpcMethodImpl.h * src/bencode2.cc * src/bencode2.h * src/bittorrent_helper.cc * src/bittorrent_helper.h * src/download_helper.cc * src/magnet.cc * src/magnet.h * test/AnnounceListTest.cc * test/Bencode2Test.cc * test/BencodeTest.cc * test/BittorrentHelperTest.cc * test/BtDependencyTest.cc * test/BtRegistryTest.cc * test/DefaultBtAnnounceTest.cc * test/DefaultBtProgressInfoFileTest.cc * test/HandshakeExtensionMessageTest.cc * test/MSEHandshakeTest.cc * test/MagnetTest.cc * test/Makefile.am * test/Makefile.in * test/RequestGroupManTest.cc * test/UTMetadataDataExtensionMessageTest.cc * test/UTMetadataPostDownloadHandlerTest.cc * test/UTMetadataRequestExtensionMessageTest.cc * test/ValueBaseTest.cc * test/XmlRpcMethodTest.cc
109 lines
3.1 KiB
C++
109 lines
3.1 KiB
C++
#include "BtRegistry.h"
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "Exception.h"
|
|
#include "DownloadContext.h"
|
|
#include "MockPeerStorage.h"
|
|
#include "MockPieceStorage.h"
|
|
#include "MockBtAnnounce.h"
|
|
#include "MockBtProgressInfoFile.h"
|
|
#include "BtRuntime.h"
|
|
#include "FileEntry.h"
|
|
#include "bittorrent_helper.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class BtRegistryTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(BtRegistryTest);
|
|
CPPUNIT_TEST(testGetDownloadContext);
|
|
CPPUNIT_TEST(testGetDownloadContext_infoHash);
|
|
CPPUNIT_TEST(testGetAllDownloadContext);
|
|
CPPUNIT_TEST(testRemove);
|
|
CPPUNIT_TEST(testRemoveAll);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void testGetDownloadContext();
|
|
void testGetDownloadContext_infoHash();
|
|
void testGetAllDownloadContext();
|
|
void testRemove();
|
|
void testRemoveAll();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( BtRegistryTest );
|
|
|
|
void BtRegistryTest::testGetDownloadContext()
|
|
{
|
|
BtRegistry btRegistry;
|
|
CPPUNIT_ASSERT(btRegistry.getDownloadContext(1).isNull());
|
|
SharedHandle<DownloadContext> dctx(new DownloadContext());
|
|
BtObject btObject;
|
|
btObject._downloadContext = dctx;
|
|
btRegistry.put(1, btObject);
|
|
CPPUNIT_ASSERT_EQUAL(dctx.get(), btRegistry.getDownloadContext(1).get());
|
|
}
|
|
|
|
static void addTwoDownloadContext(BtRegistry& btRegistry)
|
|
{
|
|
SharedHandle<DownloadContext> dctx1(new DownloadContext());
|
|
SharedHandle<DownloadContext> dctx2(new DownloadContext());
|
|
BtObject btObject1;
|
|
btObject1._downloadContext = dctx1;
|
|
BtObject btObject2;
|
|
btObject2._downloadContext = dctx2;
|
|
btRegistry.put(1, btObject1);
|
|
btRegistry.put(2, btObject2);
|
|
}
|
|
|
|
void BtRegistryTest::testGetDownloadContext_infoHash()
|
|
{
|
|
BtRegistry btRegistry;
|
|
addTwoDownloadContext(btRegistry);
|
|
SharedHandle<TorrentAttribute> attrs1(new TorrentAttribute());
|
|
attrs1->infoHash = "hash1";
|
|
SharedHandle<TorrentAttribute> attrs2(new TorrentAttribute());
|
|
attrs2->infoHash = "hash2";
|
|
btRegistry.getDownloadContext(1)->setAttribute
|
|
(bittorrent::BITTORRENT, attrs1);
|
|
btRegistry.getDownloadContext(2)->setAttribute
|
|
(bittorrent::BITTORRENT, attrs2);
|
|
|
|
CPPUNIT_ASSERT(!btRegistry.getDownloadContext("hash1").isNull());
|
|
CPPUNIT_ASSERT(btRegistry.getDownloadContext("hash1").get() ==
|
|
btRegistry.getDownloadContext(1).get());
|
|
CPPUNIT_ASSERT(btRegistry.getDownloadContext("not exists").isNull());
|
|
}
|
|
|
|
void BtRegistryTest::testGetAllDownloadContext()
|
|
{
|
|
BtRegistry btRegistry;
|
|
addTwoDownloadContext(btRegistry);
|
|
|
|
std::vector<SharedHandle<DownloadContext> > result;
|
|
btRegistry.getAllDownloadContext(std::back_inserter(result));
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, result.size());
|
|
}
|
|
|
|
void BtRegistryTest::testRemove()
|
|
{
|
|
BtRegistry btRegistry;
|
|
addTwoDownloadContext(btRegistry);
|
|
CPPUNIT_ASSERT(btRegistry.remove(1));
|
|
CPPUNIT_ASSERT(btRegistry.get(1).isNull());
|
|
CPPUNIT_ASSERT(!btRegistry.get(2).isNull());
|
|
}
|
|
|
|
void BtRegistryTest::testRemoveAll()
|
|
{
|
|
BtRegistry btRegistry;
|
|
addTwoDownloadContext(btRegistry);
|
|
btRegistry.removeAll();
|
|
CPPUNIT_ASSERT(btRegistry.get(1).isNull());
|
|
CPPUNIT_ASSERT(btRegistry.get(2).isNull());
|
|
}
|
|
|
|
} // namespace aria2
|