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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#include "SegmentMan.h"
#include "File.h"
#include "prefs.h"
#include "Util.h"
#include <cppunit/extensions/HelperMacros.h>
using namespace std;
class SegmentManTest:public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(SegmentManTest);
CPPUNIT_TEST(testSaveAndLoad);
CPPUNIT_TEST(testNullBitfield);
CPPUNIT_TEST(testCancelSegmentOnNullBitfield);
CPPUNIT_TEST_SUITE_END();
private:
public:
void setUp() {
}
void testSaveAndLoad();
void testNullBitfield();
void testCancelSegmentOnNullBitfield();
};
CPPUNIT_TEST_SUITE_REGISTRATION( SegmentManTest );
void SegmentManTest::testSaveAndLoad() {
string filename = "save-temp";
string aria2filename = filename+".aria2";
File saveFile(aria2filename);
if(saveFile.exists()) {
saveFile.remove();
assert(!saveFile.exists());
}
try {
SegmentMan segmentMan;
segmentMan.totalSize = 10*1024*1024;
segmentMan.ufilename = filename;
segmentMan.initBitfield(1024*1024, segmentMan.totalSize);
Segment seg1;
segmentMan.getSegment(seg1, 1);
seg1.writtenLength = seg1.length;
segmentMan.completeSegment(1, seg1);
Segment seg2;
segmentMan.getSegment(seg2, 2);
seg2.writtenLength = 512*1024;
segmentMan.updateSegment(2, seg2);
Segment seg3;
segmentMan.getSegment(seg3, 3);
seg2.writtenLength = 512*1024;
segmentMan.updateSegment(2, seg2);
segmentMan.save();
CPPUNIT_ASSERT(saveFile.exists());
SegmentMan segmentManLoad;
segmentManLoad.ufilename = filename;
segmentManLoad.load();
CPPUNIT_ASSERT_EQUAL(segmentMan.totalSize, segmentManLoad.totalSize);
Segment seg2Load;
segmentManLoad.getSegment(seg2Load, 2, seg2.index);
CPPUNIT_ASSERT_EQUAL(seg2, seg2Load);
Segment seg3Load;
segmentManLoad.getSegment(seg3Load, 3, seg3.index);
CPPUNIT_ASSERT_EQUAL(seg3, seg3Load);
CPPUNIT_ASSERT_EQUAL(segmentMan.getDownloadLength(), segmentManLoad.getDownloadLength());
} catch(Exception* e) {
cerr << e->getMsg() << endl;
delete e;
}
}
void SegmentManTest::testNullBitfield() {
SegmentMan segmentMan;
Option op;
op.put(PREF_SEGMENT_SIZE, Util::itos(1024*1024));
segmentMan.option = &op;
Segment segment;
CPPUNIT_ASSERT(segmentMan.getSegment(segment, 1));
CPPUNIT_ASSERT_EQUAL(Segment(0, 0, 0), segment);
Segment segment2;
CPPUNIT_ASSERT(!segmentMan.getSegment(segment2, 2));
long long int totalLength = 1024*1024;
segment.writtenLength = totalLength;
CPPUNIT_ASSERT(segmentMan.updateSegment(1, segment));
CPPUNIT_ASSERT_EQUAL(totalLength, segmentMan.getDownloadLength());
CPPUNIT_ASSERT(segmentMan.completeSegment(1, segment));
CPPUNIT_ASSERT_EQUAL(totalLength, segmentMan.getDownloadLength());
}
void SegmentManTest::testCancelSegmentOnNullBitfield() {
SegmentMan segmentMan;
Segment segment;
CPPUNIT_ASSERT(segmentMan.getSegment(segment, 1));
segmentMan.cancelSegment(1);
CPPUNIT_ASSERT(segmentMan.getSegment(segment, 1));
}
|
