aria2/test/DefaultDiskWriterTest.cc

#include "DefaultDiskWriter.h"
#include <string>
#include <cppunit/extensions/HelperMacros.h>

using namespace std;

class DefaultDiskWriterTest:public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(DefaultDiskWriterTest);
  CPPUNIT_TEST(testMessageDigest);
  CPPUNIT_TEST_SUITE_END();
private:

public:
  void setUp() {
  }

  void testMessageDigest();
};


CPPUNIT_TEST_SUITE_REGISTRATION( DefaultDiskWriterTest );

void DefaultDiskWriterTest::testMessageDigest() {
#ifdef ENABLE_MESSAGE_DIGEST
  
  DefaultDiskWriter dw;
  dw.openExistingFile("4096chunk.txt");
  
  CPPUNIT_ASSERT_EQUAL(string("608cabc0f2fa18c260cafd974516865c772363d5"),
		       dw.messageDigest(0, 4096, DIGEST_ALGO_SHA1));

  CPPUNIT_ASSERT_EQUAL(string("7a4a9ae537ebbbb826b1060e704490ad0f365ead"),
		       dw.messageDigest(5, 100, DIGEST_ALGO_SHA1));

  dw.closeFile();
#endif // ENABLE_MESSAGE_DIGEST
}