mirrors/aria2
1
0
Fork 0
mirror of https://github.com/aria2/aria2.git synced 2025-04-05 05:27:38 +03:00
Code Issues Projects Releases Packages Wiki Activity

Change have entry indexing method

Browse source 
Now use increasing sequence of integer rather than timer value.
This commit is contained in:
Tatsuhiro Tsujikawa 2016-07-10 22:42:49 +09:00
parent f2aa7564b0
commit babdcb2c7d
12 changed files with 156 additions and 88 deletions
Download patch file Download diff file Expand all files Collapse all files

56
test/DefaultPieceStorageTest.cc
View file

@ -39,6 +39,7 @@ class DefaultPieceStorageTest : public CppUnit::TestFixture {
CPPUNIT_TEST(testGetCompletedLength);
CPPUNIT_TEST(testGetFilteredCompletedLength);
CPPUNIT_TEST(testGetNextUsedIndex);
CPPUNIT_TEST(testAdvertisePiece);
CPPUNIT_TEST_SUITE_END();
private:
@ -77,6 +78,7 @@ public:
void testGetCompletedLength();
void testGetFilteredCompletedLength();
void testGetNextUsedIndex();
void testAdvertisePiece();
};
CPPUNIT_TEST_SUITE_REGISTRATION(DefaultPieceStorageTest);
@ -401,4 +403,58 @@ void DefaultPieceStorageTest::testGetNextUsedIndex()
CPPUNIT_ASSERT_EQUAL((size_t)2, pss.getNextUsedIndex(0));
}
void DefaultPieceStorageTest::testAdvertisePiece()
{
DefaultPieceStorage ps(dctx_, option_.get());
ps.advertisePiece(1, 100, Timer(10_s));
ps.advertisePiece(2, 101, Timer(11_s));
ps.advertisePiece(3, 102, Timer(11_s));
ps.advertisePiece(1, 103, Timer(12_s));
ps.advertisePiece(2, 104, Timer(100_s));
std::vector<size_t> res, ans;
uint64_t lastHaveIndex;
lastHaveIndex = ps.getAdvertisedPieceIndexes(res, 1, 0);
ans = std::vector<size_t>{100, 101, 102, 103, 104};
CPPUNIT_ASSERT_EQUAL((uint64_t)5, lastHaveIndex);
CPPUNIT_ASSERT(ans == res);
res.clear();
lastHaveIndex = ps.getAdvertisedPieceIndexes(res, 1, 3);
ans = std::vector<size_t>{103, 104};
CPPUNIT_ASSERT_EQUAL((uint64_t)5, lastHaveIndex);
CPPUNIT_ASSERT_EQUAL((size_t)2, res.size());
CPPUNIT_ASSERT(ans == res);
res.clear();
lastHaveIndex = ps.getAdvertisedPieceIndexes(res, 1, 5);
CPPUNIT_ASSERT_EQUAL((uint64_t)5, lastHaveIndex);
CPPUNIT_ASSERT_EQUAL((size_t)0, res.size());
// remove haves
ps.removeAdvertisedPiece(Timer(11_s));
res.clear();
lastHaveIndex = ps.getAdvertisedPieceIndexes(res, 1, 0);
ans = std::vector<size_t>{103, 104};
CPPUNIT_ASSERT_EQUAL((uint64_t)5, lastHaveIndex);
CPPUNIT_ASSERT_EQUAL((size_t)2, res.size());
CPPUNIT_ASSERT(ans == res);
ps.removeAdvertisedPiece(Timer(300_s));
res.clear();
lastHaveIndex = ps.getAdvertisedPieceIndexes(res, 1, 0);
CPPUNIT_ASSERT_EQUAL((uint64_t)0, lastHaveIndex);
CPPUNIT_ASSERT_EQUAL((size_t)0, res.size());
}
} // namespace aria2