From 048a2cf59737f761876e81a77e8d064c45c752d7 Mon Sep 17 00:00:00 2001 From: Tatsuhiro Tsujikawa Date: Thu, 11 Oct 2007 16:58:24 +0000 Subject: 2007-10-12 Tatsuhiro Tsujikawa Implemented BitTorrent/http/ftp integrated download. I've rewritten lots of files and now some headers have forward class declarations to reduce compile time. The implementation is extremely alpha stage, I recommend to use this for testing purpose only. --- test/SingleFileAllocationIteratorTest.cc | 68 ++++++++++++++++++++++++++++++++ 1 file changed, 68 insertions(+) create mode 100644 test/SingleFileAllocationIteratorTest.cc (limited to 'test/SingleFileAllocationIteratorTest.cc') diff --git a/test/SingleFileAllocationIteratorTest.cc b/test/SingleFileAllocationIteratorTest.cc new file mode 100644 index 00000000..050d0a19 --- /dev/null +++ b/test/SingleFileAllocationIteratorTest.cc @@ -0,0 +1,68 @@ +#include "SingleFileAllocationIterator.h" +#include "File.h" +#include "DefaultDiskWriter.h" +#include "DirectDiskAdaptor.h" +#include +#include +#include +#include +#include +#include + +class SingleFileAllocationIteratorTest:public CppUnit::TestFixture { + + CPPUNIT_TEST_SUITE(SingleFileAllocationIteratorTest); + CPPUNIT_TEST(testAllocate); + CPPUNIT_TEST_SUITE_END(); +private: + +public: + void setUp() {} + + void testAllocate(); +}; + + +CPPUNIT_TEST_SUITE_REGISTRATION( SingleFileAllocationIteratorTest ); + +void SingleFileAllocationIteratorTest::testAllocate() +{ + string dir = "/tmp"; + string fname = "aria2_SingleFileAllocationIteratorTest_testAllocate"; + string fn = dir+"/"+fname; + ofstream of(fn.c_str()); + + of << "0123456789"; + + of.close(); + + File x(fn); + CPPUNIT_ASSERT_EQUAL((int64_t)10, x.size()); + + + DefaultDiskWriterHandle writer = new DefaultDiskWriter(); + DirectDiskAdaptorHandle diskAdaptor = new DirectDiskAdaptor(); + diskAdaptor->setDiskWriter(writer); + diskAdaptor->setTotalLength(16*1024*2+8*1024); + diskAdaptor->setStoreDir(dir); + FileEntryHandle fileEntry = new FileEntry(fname, + diskAdaptor->getTotalLength(), + 0); + FileEntries fs; + fs.push_back(fileEntry); + diskAdaptor->setFileEntries(fs); + + // we have to open file first. + diskAdaptor->openFile(); + SingleFileAllocationIteratorHandle itr = diskAdaptor->fileAllocationIterator(); + + itr->allocateChunk(); + CPPUNIT_ASSERT(!itr->finished()); + itr->allocateChunk(); + CPPUNIT_ASSERT(!itr->finished()); + itr->allocateChunk(); + CPPUNIT_ASSERT(itr->finished()); + + File f(fn); + CPPUNIT_ASSERT_EQUAL((int64_t)40960, f.size()); +} -- cgit