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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
#include "MetalinkEntry.h"
#include <cppunit/extensions/HelperMacros.h>
#include "MetalinkResource.h"
#include "a2functional.h"
namespace aria2 {
class MetalinkEntryTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(MetalinkEntryTest);
CPPUNIT_TEST(testDropUnsupportedResource);
CPPUNIT_TEST(testReorderResourcesByPriority);
CPPUNIT_TEST(testSetLocationPriority);
CPPUNIT_TEST(testSetProtocolPriority);
CPPUNIT_TEST_SUITE_END();
private:
public:
void setUp() {}
void tearDown() {}
void testDropUnsupportedResource();
void testReorderResourcesByPriority();
void testSetLocationPriority();
void testSetProtocolPriority();
};
CPPUNIT_TEST_SUITE_REGISTRATION(MetalinkEntryTest);
std::unique_ptr<MetalinkEntry> createTestEntry()
{
auto entry = make_unique<MetalinkEntry>();
auto res1 = make_unique<MetalinkResource>();
res1->url = "ftp://myhost/aria2.tar.bz2";
res1->type = MetalinkResource::TYPE_FTP;
res1->location = "ro";
res1->priority = 50;
auto res2 = make_unique<MetalinkResource>();
res2->url = "http://myhost/aria2.tar.bz2";
res2->type = MetalinkResource::TYPE_HTTP;
res2->location = "at";
res2->priority = 1;
auto res3 = make_unique<MetalinkResource>();
res3->url = "http://myhost/aria2.torrent";
res3->type = MetalinkResource::TYPE_BITTORRENT;
res3->location = "al";
res3->priority = 40;
auto res4 = make_unique<MetalinkResource>();
res4->url = "http://myhost/aria2.ext";
res4->type = MetalinkResource::TYPE_NOT_SUPPORTED;
res4->location = "ad";
res4->priority = 90;
auto res5 = make_unique<MetalinkResource>();
res5->url = "https://myhost/aria2.tar.bz2";
res5->type = MetalinkResource::TYPE_HTTPS;
res5->location = "jp";
res5->priority = 10;
entry->resources.push_back(std::move(res1));
entry->resources.push_back(std::move(res2));
entry->resources.push_back(std::move(res3));
entry->resources.push_back(std::move(res4));
entry->resources.push_back(std::move(res5));
return entry;
}
void MetalinkEntryTest::testDropUnsupportedResource()
{
auto entry = createTestEntry();
entry->dropUnsupportedResource();
#if defined(ENABLE_SSL) && defined(ENABLE_BITTORRENT)
CPPUNIT_ASSERT_EQUAL((size_t)4, entry->resources.size());
#elif defined(ENABLE_SSL) || defined(ENABLE_BITTORRENT)
CPPUNIT_ASSERT_EQUAL((size_t)3, entry->resources.size());
#else // defined(ENABLE_SSL) || defined(ENABLE_BITTORRENT)
CPPUNIT_ASSERT_EQUAL((size_t)2, entry->resources.size());
#endif // defined(ENABLE_SSL) || defined(ENABLE_BITTORRENT)
auto itr = std::begin(entry->resources);
CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_FTP, (*itr++)->type);
CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_HTTP, (*itr++)->type);
#ifdef ENABLE_BITTORRENT
CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_BITTORRENT, (*itr++)->type);
#endif // ENABLE_BITTORRENT
#ifdef ENABLE_SSL
CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_HTTPS, (*itr++)->type);
#endif // ENABLE_SSL
}
void MetalinkEntryTest::testReorderResourcesByPriority()
{
auto entry = createTestEntry();
entry->reorderResourcesByPriority();
CPPUNIT_ASSERT_EQUAL(1, entry->resources.at(0)->priority);
CPPUNIT_ASSERT_EQUAL(10, entry->resources.at(1)->priority);
CPPUNIT_ASSERT_EQUAL(40, entry->resources.at(2)->priority);
CPPUNIT_ASSERT_EQUAL(50, entry->resources.at(3)->priority);
CPPUNIT_ASSERT_EQUAL(90, entry->resources.at(4)->priority);
}
void MetalinkEntryTest::testSetLocationPriority()
{
auto entry = createTestEntry();
auto locations = std::vector<std::string>{"jp", "al", "ro"};
entry->setLocationPriority(locations, -100);
CPPUNIT_ASSERT_EQUAL(std::string("ro"), entry->resources[0]->location);
CPPUNIT_ASSERT_EQUAL(-50, entry->resources[0]->priority);
CPPUNIT_ASSERT_EQUAL(std::string("at"), entry->resources[1]->location);
CPPUNIT_ASSERT_EQUAL(1, entry->resources[1]->priority);
CPPUNIT_ASSERT_EQUAL(std::string("al"), entry->resources[2]->location);
CPPUNIT_ASSERT_EQUAL(-60, entry->resources[2]->priority);
CPPUNIT_ASSERT_EQUAL(std::string("ad"), entry->resources[3]->location);
CPPUNIT_ASSERT_EQUAL(90, entry->resources[3]->priority);
CPPUNIT_ASSERT_EQUAL(std::string("jp"), entry->resources[4]->location);
CPPUNIT_ASSERT_EQUAL(-90, entry->resources[4]->priority);
}
void MetalinkEntryTest::testSetProtocolPriority()
{
auto entry = createTestEntry();
entry->setProtocolPriority("http", -1);
CPPUNIT_ASSERT_EQUAL(50, entry->resources[0]->priority); // ftp
CPPUNIT_ASSERT_EQUAL(0, entry->resources[1]->priority); // http, -1
CPPUNIT_ASSERT_EQUAL(40, entry->resources[2]->priority); // bittorrent
CPPUNIT_ASSERT_EQUAL(90, entry->resources[3]->priority); // not supported
CPPUNIT_ASSERT_EQUAL(10, entry->resources[4]->priority); // https
}
} // namespace aria2
|
