mirror of https://github.com/aria2/aria2
89 lines
2.2 KiB
C++
89 lines
2.2 KiB
C++
#include "BtRegistry.h"
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "Exception.h"
|
|
#include "MockBtContext.h"
|
|
#include "MockPeerStorage.h"
|
|
#include "MockPieceStorage.h"
|
|
#include "MockBtAnnounce.h"
|
|
#include "MockBtProgressInfoFile.h"
|
|
#include "BtRuntime.h"
|
|
#include "FileEntry.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class BtRegistryTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(BtRegistryTest);
|
|
CPPUNIT_TEST(testGetBtContext);
|
|
CPPUNIT_TEST(testGetAllBtContext);
|
|
CPPUNIT_TEST(testRemove);
|
|
CPPUNIT_TEST(testRemoveAll);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void testGetBtContext();
|
|
void testGetAllBtContext();
|
|
void testRemove();
|
|
void testRemoveAll();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( BtRegistryTest );
|
|
|
|
void BtRegistryTest::testGetBtContext()
|
|
{
|
|
BtRegistry btRegistry;
|
|
CPPUNIT_ASSERT(btRegistry.getBtContext("test").isNull());
|
|
SharedHandle<BtContext> btContext(new MockBtContext());
|
|
BtObject btObject;
|
|
btObject._btContext = btContext;
|
|
btRegistry.put("test", btObject);
|
|
CPPUNIT_ASSERT_EQUAL(btContext.get(),
|
|
btRegistry.getBtContext("test").get());
|
|
}
|
|
|
|
static void addTwoBtContext(BtRegistry& btRegistry)
|
|
{
|
|
SharedHandle<BtContext> btContext1(new MockBtContext());
|
|
SharedHandle<BtContext> btContext2(new MockBtContext());
|
|
BtObject btObject1;
|
|
btObject1._btContext = btContext1;
|
|
BtObject btObject2;
|
|
btObject2._btContext = btContext2;
|
|
btRegistry.put("ctx1", btObject1);
|
|
btRegistry.put("ctx2", btObject2);
|
|
}
|
|
|
|
void BtRegistryTest::testGetAllBtContext()
|
|
{
|
|
BtRegistry btRegistry;
|
|
addTwoBtContext(btRegistry);
|
|
|
|
std::vector<SharedHandle<BtContext> > result;
|
|
btRegistry.getAllBtContext(std::back_inserter(result));
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, result.size());
|
|
}
|
|
|
|
void BtRegistryTest::testRemove()
|
|
{
|
|
BtRegistry btRegistry;
|
|
addTwoBtContext(btRegistry);
|
|
CPPUNIT_ASSERT(btRegistry.remove("ctx1"));
|
|
CPPUNIT_ASSERT(btRegistry.get("ctx1").isNull());
|
|
CPPUNIT_ASSERT(!btRegistry.get("ctx2").isNull());
|
|
}
|
|
|
|
void BtRegistryTest::testRemoveAll()
|
|
{
|
|
BtRegistry btRegistry;
|
|
addTwoBtContext(btRegistry);
|
|
btRegistry.removeAll();
|
|
CPPUNIT_ASSERT(btRegistry.get("ctx1").isNull());
|
|
CPPUNIT_ASSERT(btRegistry.get("ctx2").isNull());
|
|
}
|
|
|
|
} // namespace aria2
|