123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147 |
- #include "SegmentMan.h"
- #include <cppunit/extensions/HelperMacros.h>
- #include "DownloadContext.h"
- #include "UnknownLengthPieceStorage.h"
- #include "DefaultPieceStorage.h"
- #include "Segment.h"
- #include "Option.h"
- #include "PieceSelector.h"
- namespace aria2 {
- class SegmentManTest:public CppUnit::TestFixture {
- CPPUNIT_TEST_SUITE(SegmentManTest);
- CPPUNIT_TEST(testNullBitfield);
- CPPUNIT_TEST(testCompleteSegment);
- CPPUNIT_TEST(testGetSegment_sameFileEntry);
- CPPUNIT_TEST(testRegisterPeerStat);
- CPPUNIT_TEST_SUITE_END();
- private:
- public:
- void setUp() {
- }
- void testNullBitfield();
- void testCompleteSegment();
- void testGetPeerStat();
- void testGetSegment_sameFileEntry();
- void testRegisterPeerStat();
- };
- CPPUNIT_TEST_SUITE_REGISTRATION( SegmentManTest );
- void SegmentManTest::testNullBitfield()
- {
- Option op;
- SharedHandle<DownloadContext> dctx
- (new DownloadContext(0, 0, "aria2.tar.bz2"));
- SharedHandle<UnknownLengthPieceStorage> ps
- (new UnknownLengthPieceStorage(dctx, &op));
- SegmentMan segmentMan(&op, dctx, ps);
- SharedHandle<Segment> segment = segmentMan.getSegment(1);
- CPPUNIT_ASSERT(!segment.isNull());
- CPPUNIT_ASSERT_EQUAL((size_t)0, segment->getIndex());
- CPPUNIT_ASSERT_EQUAL((size_t)0, segment->getLength());
- CPPUNIT_ASSERT_EQUAL((size_t)0, segment->getSegmentLength());
- CPPUNIT_ASSERT_EQUAL((size_t)0, segment->getWrittenLength());
- SharedHandle<Segment> segment2 = segmentMan.getSegment(2);
- CPPUNIT_ASSERT(segment2.isNull());
- segmentMan.cancelSegment(1);
- CPPUNIT_ASSERT(!segmentMan.getSegment(2).isNull());
- }
- void SegmentManTest::testCompleteSegment()
- {
- Option op;
- size_t pieceLength = 1024*1024;
- uint64_t totalLength = 64*1024*1024;
- SharedHandle<DownloadContext> dctx
- (new DownloadContext(pieceLength, totalLength, "aria2.tar.bz2"));
- SharedHandle<DefaultPieceStorage> ps(new DefaultPieceStorage(dctx, &op));
- SegmentMan segmentMan(&op, dctx, ps);
- CPPUNIT_ASSERT(!segmentMan.getSegment(1, 0).isNull());
- SharedHandle<Segment> seg = segmentMan.getSegment(1, 1);
- CPPUNIT_ASSERT(!seg.isNull());
- CPPUNIT_ASSERT(!segmentMan.getSegment(1, 2).isNull());
- seg->updateWrittenLength(pieceLength);
- segmentMan.completeSegment(1, seg);
-
- std::vector<SharedHandle<Segment> > segments;
- segmentMan.getInFlightSegment(segments, 1);
- CPPUNIT_ASSERT_EQUAL((size_t)2, segments.size());
- CPPUNIT_ASSERT_EQUAL((size_t)0, segments[0]->getIndex());
- CPPUNIT_ASSERT_EQUAL((size_t)2, segments[1]->getIndex());
- }
- void SegmentManTest::testGetSegment_sameFileEntry()
- {
- Option op;
- SharedHandle<DownloadContext> dctx(new DownloadContext());
- dctx->setPieceLength(2);
- SharedHandle<FileEntry> fileEntries[] = {
- SharedHandle<FileEntry>(new FileEntry("file1", 3, 0)),
- SharedHandle<FileEntry>(new FileEntry("file2", 6, 3)),
- SharedHandle<FileEntry>(new FileEntry("file3", 1, 9))
- };
- dctx->setFileEntries(&fileEntries[0], &fileEntries[3]);
- SharedHandle<DefaultPieceStorage> ps(new DefaultPieceStorage(dctx, &op));
- SegmentMan segman(&op, dctx, ps);
- std::vector<SharedHandle<Segment> > segments;
- segman.getSegment(segments, 1, fileEntries[1], 4);
- // See 3 segments are returned, not 4 because the part of file1 is
- // not filled in segment#1
- CPPUNIT_ASSERT_EQUAL((size_t)3, segments.size());
-
- SharedHandle<Segment> segmentNo1 = segman.getSegment(2, 1);
- // Fill the part of file1 in segment#1
- segmentNo1->updateWrittenLength(1);
- segman.cancelSegment(2);
- segman.cancelSegment(1);
- segments.clear();
- segman.getSegment(segments, 1, fileEntries[1], 4);
- CPPUNIT_ASSERT_EQUAL((size_t)4, segments.size());
- segman.cancelSegment(1);
- SharedHandle<Segment> segmentNo4 = segman.getSegment(1, 4);
- // Fill the part of file2 in segment#4
- segmentNo4->updateWrittenLength(1);
- segman.cancelSegment(1);
- segments.clear();
- segman.getSegment(segments, 1, fileEntries[1], 4);
- // segment#4 is not returned because the part of file2 is filled.
- CPPUNIT_ASSERT_EQUAL((size_t)3, segments.size());
- }
- void SegmentManTest::testRegisterPeerStat()
- {
- Option op;
- SharedHandle<DownloadContext> dctx(new DownloadContext());
- SharedHandle<DefaultPieceStorage> ps(new DefaultPieceStorage(dctx, &op));
- SegmentMan segman(&op, dctx, ps);
-
- SharedHandle<PeerStat> p1(new PeerStat(0, "host1", "http"));
- segman.registerPeerStat(p1);
- CPPUNIT_ASSERT_EQUAL((size_t)1, segman.getPeerStats().size());
- SharedHandle<PeerStat> p2(new PeerStat(0, "host2", "http"));
- segman.registerPeerStat(p2);
- CPPUNIT_ASSERT_EQUAL((size_t)1, segman.getPeerStats().size());
- p2->downloadStart();
- segman.registerPeerStat(p1);
- CPPUNIT_ASSERT_EQUAL((size_t)2, segman.getPeerStats().size());
- }
- } // namespace aria2
|