mirror of
https://github.com/aria2/aria2.git
synced 2025-02-26 08:22:11 +00:00
3505201f33
Rewritten SharedHandle. Now copy constructor taking raw pointer has keyword explicit and SharedHandle's default constructor initializes its internal obj to null, old implementation initializes it using obj's default constructor. To assign null, write SharedHandle<T> x(...); x.reset(); TODO: test/SharedHandleTest.cc needs more tests. * src/SharedHandle.h
133 lines
3.8 KiB
C++
133 lines
3.8 KiB
C++
#include "MultiDiskAdaptor.h"
|
|
#include "FileEntry.h"
|
|
#include "Exception.h"
|
|
#include <string>
|
|
#include <cerrno>
|
|
#include <cstring>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class MultiDiskAdaptorTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(MultiDiskAdaptorTest);
|
|
CPPUNIT_TEST(testWriteData);
|
|
CPPUNIT_TEST(testReadData);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
SharedHandle<MultiDiskAdaptor> adaptor;
|
|
public:
|
|
void setUp() {
|
|
adaptor.reset(new MultiDiskAdaptor());
|
|
adaptor->setPieceLength(2);
|
|
adaptor->setStoreDir(".");
|
|
adaptor->setTopDir(".");
|
|
}
|
|
|
|
void testWriteData();
|
|
void testReadData();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION( MultiDiskAdaptorTest );
|
|
|
|
std::deque<SharedHandle<FileEntry> > createEntries() {
|
|
SharedHandle<FileEntry> entry1(new FileEntry("file1.txt", 15, 0));
|
|
SharedHandle<FileEntry> entry2(new FileEntry("file2.txt", 7, 15));
|
|
SharedHandle<FileEntry> entry3(new FileEntry("file3.txt", 3, 22));
|
|
unlink("file1.txt");
|
|
unlink("file2.txt");
|
|
unlink("file3.txt");
|
|
std::deque<SharedHandle<FileEntry> > entries;
|
|
entries.push_back(entry1);
|
|
entries.push_back(entry2);
|
|
entries.push_back(entry3);
|
|
return entries;
|
|
}
|
|
|
|
void readFile(const std::string& filename, char* buf, int bufLength) {
|
|
FILE* f = fopen(filename.c_str(), "r");
|
|
if(f == NULL) {
|
|
CPPUNIT_FAIL(strerror(errno));
|
|
}
|
|
int retval = fread(buf, bufLength, 1, f);
|
|
fclose(f);
|
|
if(retval != 1) {
|
|
CPPUNIT_FAIL("return value is not 1");
|
|
}
|
|
}
|
|
|
|
void MultiDiskAdaptorTest::testWriteData() {
|
|
try {
|
|
adaptor->setFileEntries(createEntries());
|
|
|
|
adaptor->openFile();
|
|
std::string msg = "12345";
|
|
adaptor->writeData((const unsigned char*)msg.c_str(), msg.size(), 0);
|
|
adaptor->closeFile();
|
|
|
|
char buf[128];
|
|
readFile("file1.txt", buf, 5);
|
|
buf[5] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(msg, std::string(buf));
|
|
|
|
adaptor->openFile();
|
|
std::string msg2 = "67890ABCDEF";
|
|
adaptor->writeData((const unsigned char*)msg2.c_str(), msg2.size(), 5);
|
|
adaptor->closeFile();
|
|
|
|
readFile("file1.txt", buf, 15);
|
|
buf[15] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("1234567890ABCDE"), std::string(buf));
|
|
readFile("file2.txt", buf, 1);
|
|
buf[1] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("F"), std::string(buf));
|
|
|
|
adaptor->openFile();
|
|
std::string msg3 = "12345123456712";
|
|
adaptor->writeData((const unsigned char*)msg3.c_str(), msg3.size(), 10);
|
|
adaptor->closeFile();
|
|
|
|
readFile("file1.txt", buf, 15);
|
|
buf[15] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("123456789012345"), std::string(buf));
|
|
readFile("file2.txt", buf, 7);
|
|
buf[7] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("1234567"), std::string(buf));
|
|
readFile("file3.txt", buf, 2);
|
|
buf[2] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("12"), std::string(buf));
|
|
} catch(Exception* e) {
|
|
CPPUNIT_FAIL(e->getMsg());
|
|
}
|
|
}
|
|
|
|
void MultiDiskAdaptorTest::testReadData() {
|
|
SharedHandle<FileEntry> entry1(new FileEntry("file1r.txt", 15, 0));
|
|
SharedHandle<FileEntry> entry2(new FileEntry("file2r.txt", 7, 15));
|
|
SharedHandle<FileEntry> entry3(new FileEntry("file3r.txt", 3, 22));
|
|
std::deque<SharedHandle<FileEntry> > entries;
|
|
entries.push_back(entry1);
|
|
entries.push_back(entry2);
|
|
entries.push_back(entry3);
|
|
|
|
adaptor->setFileEntries(entries);
|
|
|
|
adaptor->openFile();
|
|
unsigned char buf[128];
|
|
adaptor->readData(buf, 15, 0);
|
|
buf[15] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("1234567890ABCDE"), std::string((char*)buf));
|
|
adaptor->readData(buf, 10, 6);
|
|
buf[10] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("7890ABCDEF"), std::string((char*)buf));
|
|
adaptor->readData(buf, 4, 20);
|
|
buf[4] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("KLMN"), std::string((char*)buf));
|
|
adaptor->readData(buf, 25, 0);
|
|
buf[25] = '\0';
|
|
CPPUNIT_ASSERT_EQUAL(std::string("1234567890ABCDEFGHIJKLMNO"), std::string((char*)buf));
|
|
}
|
|
|
|
} // namespace aria2
|