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
|
#include "RequestGroup.h"
#include <cppunit/extensions/HelperMacros.h>
#include "Option.h"
#include "DownloadContext.h"
#include "FileEntry.h"
#include "PieceStorage.h"
#include "DownloadResult.h"
namespace aria2 {
class RequestGroupTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(RequestGroupTest);
CPPUNIT_TEST(testGetFirstFilePath);
CPPUNIT_TEST(testCreateDownloadResult);
CPPUNIT_TEST_SUITE_END();
private:
std::shared_ptr<Option> option_;
public:
void setUp()
{
option_.reset(new Option());
}
void testGetFirstFilePath();
void testCreateDownloadResult();
};
CPPUNIT_TEST_SUITE_REGISTRATION( RequestGroupTest );
void RequestGroupTest::testGetFirstFilePath()
{
std::shared_ptr<DownloadContext> ctx
(new DownloadContext(1_k, 1_k, "/tmp/myfile"));
RequestGroup group(GroupId::create(), option_);
group.setDownloadContext(ctx);
CPPUNIT_ASSERT_EQUAL(std::string("/tmp/myfile"), group.getFirstFilePath());
group.markInMemoryDownload();
CPPUNIT_ASSERT_EQUAL(std::string("[MEMORY]myfile"), group.getFirstFilePath());
}
void RequestGroupTest::testCreateDownloadResult()
{
std::shared_ptr<DownloadContext> ctx
(new DownloadContext(1_k, 1_m, "/tmp/myfile"));
RequestGroup group(GroupId::create(), option_);
group.setDownloadContext(ctx);
group.initPieceStorage();
{
std::shared_ptr<DownloadResult> result = group.createDownloadResult();
CPPUNIT_ASSERT_EQUAL(std::string("/tmp/myfile"),
result->fileEntries[0]->getPath());
CPPUNIT_ASSERT_EQUAL((int64_t)1_m,
result->fileEntries.back()->getLastOffset());
CPPUNIT_ASSERT_EQUAL((uint64_t)0, result->sessionDownloadLength);
CPPUNIT_ASSERT_EQUAL((int64_t)0, result->sessionTime.count());
// result is UNKNOWN_ERROR if download has not completed and no specific
// error has been reported
CPPUNIT_ASSERT_EQUAL(error_code::UNKNOWN_ERROR, result->result);
// if haltReason is set to RequestGroup::USER_REQUEST, download
// result will become REMOVED.
group.setHaltRequested(true, RequestGroup::USER_REQUEST);
result = group.createDownloadResult();
CPPUNIT_ASSERT_EQUAL(error_code::REMOVED, result->result);
// if haltReason is set to RequestGroup::SHUTDOWN_SIGNAL, download
// result will become IN_PROGRESS.
group.setHaltRequested(true, RequestGroup::SHUTDOWN_SIGNAL);
result = group.createDownloadResult();
CPPUNIT_ASSERT_EQUAL(error_code::IN_PROGRESS, result->result);
}
{
group.setLastErrorCode(error_code::RESOURCE_NOT_FOUND);
std::shared_ptr<DownloadResult> result = group.createDownloadResult();
CPPUNIT_ASSERT_EQUAL(error_code::RESOURCE_NOT_FOUND, result->result);
}
{
group.getPieceStorage()->markAllPiecesDone();
std::shared_ptr<DownloadResult> result = group.createDownloadResult();
CPPUNIT_ASSERT_EQUAL(error_code::FINISHED, result->result);
}
}
} // namespace aria2
|
