mirror of https://github.com/aria2/aria2
88 lines
2.3 KiB
C++
88 lines
2.3 KiB
C++
#include "PiecedSegment.h"
|
|
#include "Piece.h"
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class SegmentTest : public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(SegmentTest);
|
|
CPPUNIT_TEST(testUpdateWrittenLength);
|
|
CPPUNIT_TEST(testUpdateWrittenLength_overflow);
|
|
CPPUNIT_TEST(testUpdateWrittenLength_lastPiece);
|
|
CPPUNIT_TEST(testUpdateWrittenLength_incompleteLastPiece);
|
|
CPPUNIT_TEST(testClear);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void setUp() {}
|
|
|
|
void testUpdateWrittenLength();
|
|
void testUpdateWrittenLength_overflow();
|
|
void testUpdateWrittenLength_lastPiece();
|
|
void testUpdateWrittenLength_incompleteLastPiece();
|
|
void testClear();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( SegmentTest );
|
|
|
|
void SegmentTest::testUpdateWrittenLength()
|
|
{
|
|
SharedHandle<Piece> p(new Piece(0, 16*1024*10));
|
|
PiecedSegment s(16*1024*10, p);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)0, s.getWrittenLength());
|
|
|
|
s.updateWrittenLength(16*1024);
|
|
CPPUNIT_ASSERT(p->hasBlock(0));
|
|
CPPUNIT_ASSERT(!p->hasBlock(1));
|
|
|
|
s.updateWrittenLength(16*1024*9);
|
|
CPPUNIT_ASSERT(p->pieceComplete());
|
|
}
|
|
|
|
void SegmentTest::testUpdateWrittenLength_overflow()
|
|
{
|
|
SharedHandle<Piece> p(new Piece(0, 16*1024*10));
|
|
PiecedSegment s(16*1024*10, p);
|
|
|
|
s.updateWrittenLength(16*1024*11);
|
|
CPPUNIT_ASSERT(p->pieceComplete());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)16*1024, s.getOverflowLength());
|
|
}
|
|
|
|
void SegmentTest::testUpdateWrittenLength_lastPiece()
|
|
{
|
|
SharedHandle<Piece> p(new Piece(0, 16*1024*9+1));
|
|
PiecedSegment s(16*1024*10, p);
|
|
|
|
s.updateWrittenLength(16*1024*10);
|
|
CPPUNIT_ASSERT(p->pieceComplete());
|
|
}
|
|
|
|
void SegmentTest::testUpdateWrittenLength_incompleteLastPiece()
|
|
{
|
|
SharedHandle<Piece> p(new Piece(0, 16*1024*9+2));
|
|
PiecedSegment s(16*1024*10, p);
|
|
|
|
s.updateWrittenLength(16*1024*9+1);
|
|
CPPUNIT_ASSERT(!p->pieceComplete());
|
|
s.updateWrittenLength(1);
|
|
CPPUNIT_ASSERT(p->pieceComplete());
|
|
}
|
|
|
|
void SegmentTest::testClear()
|
|
{
|
|
SharedHandle<Piece> p(new Piece(0, 16*1024*10));
|
|
PiecedSegment s(16*1024*10, p);
|
|
s.updateWrittenLength(16*1024*11);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)16*1024*10, s.getWrittenLength());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)16*1024, s.getOverflowLength());
|
|
s.clear();
|
|
CPPUNIT_ASSERT_EQUAL((size_t)0, s.getWrittenLength());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)0, s.getOverflowLength());
|
|
}
|
|
|
|
} // namespace aria2
|