summaryrefslogtreecommitdiff
path: root/test/GrowSegmentTest.cc
blob: 2d895ba4efcd6892508d96fe8b9af07519675d4c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include "GrowSegment.h"
#include "Piece.h"
#include <cppunit/extensions/HelperMacros.h>

namespace aria2 {

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(std::shared_ptr<Piece>(new Piece()));
  segment.updateWrittenLength(32_k);

  CPPUNIT_ASSERT_EQUAL((int64_t)32_k, segment.getPositionToWrite());
  CPPUNIT_ASSERT(!segment.complete());
  CPPUNIT_ASSERT(segment.getPiece()->pieceComplete());
}

void GrowSegmentTest::testClear()
{
  GrowSegment segment(std::shared_ptr<Piece>(new Piece()));
  segment.updateWrittenLength(32_k);
  CPPUNIT_ASSERT_EQUAL((int64_t)32_k, segment.getWrittenLength());
  segment.clear(nullptr);
  CPPUNIT_ASSERT_EQUAL((int64_t)0, segment.getWrittenLength());
}

} // namespace aria2