blob: 8c48b09a4cc52c5072a6d7d38438f2eee23e17ef (
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
|
#include "ShareRatioSeedCriteria.h"
#include <cppunit/extensions/HelperMacros.h>
#include "DownloadContext.h"
#include "BtRuntime.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()
{
std::shared_ptr<DownloadContext> dctx(new DownloadContext(1_m, 1000000));
std::shared_ptr<BtRuntime> btRuntime(new BtRuntime());
btRuntime->setUploadLengthAtStartup(1000000);
std::shared_ptr<MockPieceStorage> pieceStorage(new MockPieceStorage());
pieceStorage->setCompletedLength(1000000);
ShareRatioSeedCriteria cri(1.0, dctx);
cri.setBtRuntime(btRuntime);
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
|
