mirror of
https://github.com/aria2/aria2.git
synced 2025-01-04 09:03:46 +00:00
55 lines
1.2 KiB
C++
55 lines
1.2 KiB
C++
#include "SingleFileAllocationIterator.h"
|
|
#include <fstream>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "File.h"
|
|
#include "DefaultDiskWriter.h"
|
|
#include "a2functional.h"
|
|
|
|
namespace aria2 {
|
|
|
|
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()
|
|
{
|
|
std::string dir = A2_TEST_OUT_DIR;
|
|
std::string fname = "aria2_SingleFileAllocationIteratorTest_testAllocate";
|
|
std::string fn = dir + "/" + fname;
|
|
std::ofstream of(fn.c_str(), std::ios::binary);
|
|
of << "0123456789";
|
|
of.close();
|
|
|
|
File x(fn);
|
|
CPPUNIT_ASSERT_EQUAL((int64_t)10, x.size());
|
|
|
|
DefaultDiskWriter writer(fn);
|
|
int64_t offset = 10;
|
|
int64_t totalLength = 32_k + 8_k;
|
|
|
|
// we have to open file first.
|
|
writer.openExistingFile();
|
|
SingleFileAllocationIterator itr(&writer, offset, totalLength);
|
|
itr.init();
|
|
|
|
while (!itr.finished()) {
|
|
itr.allocateChunk();
|
|
}
|
|
File f(fn);
|
|
CPPUNIT_ASSERT_EQUAL((int64_t)40_k, f.size());
|
|
}
|
|
|
|
} // namespace aria2
|