blob: 995dd05a812fcc1131bb93086c34ec2fb4ef57cf (
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
49
50
51
52
|
#include "ShareRatioSeedCriteria.h"
#include "MockBtContext.h"
#include "MockPeerStorage.h"
#include "MockPieceStorage.h"
#include <cppunit/extensions/HelperMacros.h>
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() {
string infoHash = "01234567890123456789";
string infoHashString = Util::toHex((const unsigned char*)infoHash.c_str(), infoHash.size());
MockBtContext* mockBtContext = new MockBtContext();
mockBtContext->setTotalLength(1000000);
mockBtContext->setInfoHash((const unsigned char*)infoHash.c_str());
BtContextHandle btContext(mockBtContext);
BtRuntimeHandle btRuntime(new BtRuntime());
btRuntime->setUploadLengthAtStartup(500000);
BtRegistry::registerBtRuntime(infoHashString, btRuntime);
MockPeerStorage* mockPeerStorage = new MockPeerStorage();
TransferStat stat;
stat.setSessionDownloadLength(1000000);
stat.setSessionUploadLength(500000);
mockPeerStorage->setStat(stat);
PeerStorageHandle peerStorage(mockPeerStorage);
BtRegistry::registerPeerStorage(infoHashString, peerStorage);
MockPieceStorage* mockPieceStorage = new MockPieceStorage();
mockPieceStorage->setCompletedLength(1000000);
BtRegistry::registerPieceStorage(infoHashString, PieceStorageHandle(mockPieceStorage));
ShareRatioSeedCriteria cri(1.0, btContext);
CPPUNIT_ASSERT(cri.evaluate());
cri.setRatio(2.0);
CPPUNIT_ASSERT(!cri.evaluate());
// check div by zero
mockBtContext->setTotalLength(0);
CPPUNIT_ASSERT(!cri.evaluate());
}
|
