2007-10-12 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>

	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.

1 files changed, 68 insertions, 0 deletions

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 <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <fstream>
+#include <iomanip>
+#include <cppunit/extensions/HelperMacros.h>
+
+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());
+}