mirror of
https://github.com/aria2/aria2.git
synced 2025-04-06 05:57:36 +03:00
1171a2063f
Throw exception when chunck checksum verification fails.
* src/DownloadCommand.cc (validatePieceHash): New function.
* src/PiecedSegment.{h, cc} (clear): New function.
* src/GrowSegment.{h, cc} (clear): New function.
* src/Segment.h (clear): New function.
* src/SegmentMan.{h, cc} (validatePieceHash): Removed.
* test/SegmentTest.cc
* test/GrowSegmentTest.cc
42 lines
1,018 B
C++
42 lines
1,018 B
C++
#include "GrowSegment.h"
|
|
#include "Piece.h"
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
using namespace std;
|
|
|
|
class GrowSegmentTest : public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(GrowSegmentTest);
|
|
CPPUNIT_TEST(testUpdateWrittenLength);
|
|
CPPUNIT_TEST(testClear);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void setUp() {}
|
|
|
|
void testUpdateWrittenLength();
|
|
void testClear();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( GrowSegmentTest );
|
|
|
|
void GrowSegmentTest::testUpdateWrittenLength()
|
|
{
|
|
GrowSegment segment(new Piece());
|
|
segment.updateWrittenLength(32*1024);
|
|
|
|
CPPUNIT_ASSERT_EQUAL((int64_t)32*1024, segment.getPositionToWrite());
|
|
CPPUNIT_ASSERT(!segment.complete());
|
|
CPPUNIT_ASSERT(segment.getPiece()->pieceComplete());
|
|
}
|
|
|
|
void GrowSegmentTest::testClear()
|
|
{
|
|
GrowSegment segment(new Piece());
|
|
segment.updateWrittenLength(32*1024);
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)32*1024, segment.getWrittenLength());
|
|
segment.clear();
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)0, segment.getWrittenLength());
|
|
}
|