blob: 3197ed147827ba5a479b52147c21664ed705efa2 (
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
45
46
47
48
|
#include "ShareRatioSeedCriteria.h"
#include <cppunit/extensions/HelperMacros.h>
#include "DownloadContext.h"
#include "MockPeerStorage.h"
#include "MockPieceStorage.h"
#include "FileEntry.h"
namespace aria2 {
class ShareRatioSeedCriteriaTest:public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(ShareRatioSeedCriteriaTest);
CPPUNIT_TEST(testEvaluate);
CPPUNIT_TEST_SUITE_END();
public:
void testEvaluate();
};
CPPUNIT_TEST_SUITE_REGISTRATION(ShareRatioSeedCriteriaTest);
void ShareRatioSeedCriteriaTest::testEvaluate() {
SharedHandle<DownloadContext> dctx(new DownloadContext(1024*1024, 1000000));
SharedHandle<MockPeerStorage> peerStorage(new MockPeerStorage());
TransferStat stat;
stat.setAllTimeUploadLength(1000000);
peerStorage->setStat(stat);
SharedHandle<MockPieceStorage> pieceStorage(new MockPieceStorage());
pieceStorage->setCompletedLength(1000000);
ShareRatioSeedCriteria cri(1.0, dctx);
cri.setPeerStorage(peerStorage);
cri.setPieceStorage(pieceStorage);
CPPUNIT_ASSERT(cri.evaluate());
cri.setRatio(2.0);
CPPUNIT_ASSERT(!cri.evaluate());
// check div by zero
dctx->getFirstFileEntry()->setLength(0);
CPPUNIT_ASSERT(!cri.evaluate());
}
} // namespace aria2
|
