aria2/test/ShareRatioSeedCriteriaTest.cc

#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