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, 102 insertions, 0 deletions

diff --git a/test/MultiFileAllocationIteratorTest.cc b/test/MultiFileAllocationIteratorTest.cc
new file mode 100644
index 00000000..e64fa225
--- /dev/null
+++ b/test/MultiFileAllocationIteratorTest.cc
@@ -0,0 +1,102 @@
+#include "MultiFileAllocationIterator.h"
+#include "File.h"
+#include "MultiDiskAdaptor.h"
+#include <cppunit/extensions/HelperMacros.h>
+
+class MultiFileAllocationIteratorTest:public CppUnit::TestFixture {
+
+  CPPUNIT_TEST_SUITE(MultiFileAllocationIteratorTest);
+  CPPUNIT_TEST(testAllocate);
+  CPPUNIT_TEST_SUITE_END();
+private:
+
+public:
+  void setUp() {}
+
+  void testAllocate();
+};
+
+
+CPPUNIT_TEST_SUITE_REGISTRATION( MultiFileAllocationIteratorTest );
+
+void MultiFileAllocationIteratorTest::testAllocate()
+{
+  string dir = "/tmp";
+  string topDir = "aria2_MultiFileAllocationIteratorTest_testAllocate";
+  string fname1 = "file1";
+  string fname2 = "file2";
+  string fname3 = "file3";
+  string fname4 = "file4";
+  string fname5 = "file5";
+  string fname6 = "file6";
+  int64_t length1 = 32769;
+  int64_t length2 = 0;
+  int64_t length3 = 8;
+  int64_t length4 = 10;
+  int64_t length5 = 20;
+  int64_t length6 = 30;
+
+  try {
+  MultiDiskAdaptorHandle diskAdaptor = new MultiDiskAdaptor();
+  diskAdaptor->setStoreDir(dir);
+  diskAdaptor->setTopDir(topDir);
+
+  int64_t offset = 0;
+  FileEntryHandle fileEntry1 = new FileEntry(fname1,
+					     length1,
+					     offset);
+  offset += length1;
+  FileEntryHandle fileEntry2 = new FileEntry(fname2,
+					     length2,
+					     offset);
+
+  offset += length2;
+  FileEntryHandle fileEntry3 = new FileEntry(fname3,
+					     length3,
+					     offset);
+
+  offset += length3;
+  FileEntryHandle fileEntry4 = new FileEntry(fname4,
+					     length4,
+					     offset);
+  fileEntry4->setRequested(false);
+
+  offset += length4;
+  FileEntryHandle fileEntry5 = new FileEntry(fname5,
+					     length5,
+					     offset);
+
+  offset += length5;
+  FileEntryHandle fileEntry6 = new FileEntry(fname6,
+					     length6,
+					     offset);
+  fileEntry6->setRequested(false);
+
+  FileEntries fs;
+  fs.push_back(fileEntry1);
+  fs.push_back(fileEntry2);
+  fs.push_back(fileEntry3);
+  fs.push_back(fileEntry4);
+  fs.push_back(fileEntry5);
+  fs.push_back(fileEntry6);
+  diskAdaptor->setFileEntries(fs);
+
+  // we have to open file first.
+  diskAdaptor->initAndOpenFile();
+  MultiFileAllocationIteratorHandle itr = diskAdaptor->fileAllocationIterator();
+  while(!itr->finished()) {
+    itr->allocateChunk();
+  }
+  CPPUNIT_ASSERT_EQUAL((int64_t)length1, File(dir+"/"+topDir+"/"+fname1).size());
+  CPPUNIT_ASSERT_EQUAL((int64_t)length2, File(dir+"/"+topDir+"/"+fname2).size());
+  CPPUNIT_ASSERT_EQUAL((int64_t)length3, File(dir+"/"+topDir+"/"+fname3).size());
+  CPPUNIT_ASSERT_EQUAL((int64_t)0, File(dir+"/"+topDir+"/"+fname4).size());
+  CPPUNIT_ASSERT_EQUAL((int64_t)length5, File(dir+"/"+topDir+"/"+fname5).size());
+  CPPUNIT_ASSERT_EQUAL((int64_t)0, File(dir+"/"+topDir+"/"+fname6).size());
+
+  } catch(Exception* e) {
+    cerr << e->getMsg() << endl;
+    delete e;
+    CPPUNIT_FAIL("exception was thrown");
+  }
+}