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
|
#include "RarestPieceSelector.h"
#include <cppunit/extensions/HelperMacros.h>
#include "BitfieldMan.h"
#include "PieceStatMan.h"
#include "a2functional.h"
namespace aria2 {
class RarestPieceSelectorTest : public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(RarestPieceSelectorTest);
CPPUNIT_TEST(testSelect);
CPPUNIT_TEST_SUITE_END();
public:
void setUp() {}
void tearDown() {}
void testAddPieceStats_index();
void testAddPieceStats_bitfield();
void testUpdatePieceStats();
void testSubtractPieceStats();
void testSelect();
};
CPPUNIT_TEST_SUITE_REGISTRATION(RarestPieceSelectorTest);
void RarestPieceSelectorTest::testSelect()
{
std::shared_ptr<PieceStatMan> pieceStatMan(new PieceStatMan(10, false));
RarestPieceSelector selector(pieceStatMan);
BitfieldMan bf(1_k, 10_k);
bf.setBitRange(0, 2);
size_t index;
pieceStatMan->addPieceStats(0);
CPPUNIT_ASSERT(selector.select(index, bf.getBitfield(), bf.countBlock()));
CPPUNIT_ASSERT_EQUAL((size_t)1, index);
pieceStatMan->addPieceStats(1);
CPPUNIT_ASSERT(selector.select(index, bf.getBitfield(), bf.countBlock()));
CPPUNIT_ASSERT_EQUAL((size_t)2, index);
}
} // namespace aria2
|
