Method testMakeDiskWriterEntries

test/MultiFileAllocationIteratorTest.cc:35–134

CPPUNIT_TEST_SUITE_REGISTRATION(MultiFileAllocationIteratorTest);

void MultiFileAllocationIteratorTest::testMakeDiskWriterEntries()
{
  std::string storeDir =
      A2_TEST_OUT_DIR "/aria2_MultiFileAllocationIteratorTest"
                      "_testMakeDiskWriterEntries";

  auto fs = std::vector<std::shared_ptr<FileEntry>>{
      std::make_shared<FileEntry>(storeDir + "/file1", 1536, 0),
      std::make_shared<FileEntry>(storeDir + "/file2", 2048, 1536), // req no
      std::make_shared<FileEntry>(storeDir + "/file3", 1024, 3584),
      std::make_shared<FileEntry>(storeDir + "/file4", 1024, 4608), // req no
      std::make_shared<FileEntry>(storeDir + "/file5", 1024, 5632), // req no
      std::make_shared<FileEntry>(storeDir + "/file6", 1024, 6656), // req no
      std::make_shared<FileEntry>(storeDir + "/file7", 256, 7680),  // req no
      std::make_shared<FileEntry>(storeDir + "/file8", 255, 7936),
      std::make_shared<FileEntry>(storeDir + "/file9", 1025, 8191), // req no
      std::make_shared<FileEntry>(storeDir + "/fileA", 1024, 9216), // req no
      std::make_shared<FileEntry>(storeDir + "/fileB", 1024, 10240)};
  fs[1]->setRequested(false); // file2
  fs[3]->setRequested(false); // file4
  fs[4]->setRequested(false); // file5
  fs[5]->setRequested(false); // file6
  fs[6]->setRequested(false); // file7
  fs[8]->setRequested(false); // file9
  fs[9]->setRequested(false); // fileA

  for (auto& fe : fs) {
    File{fe->getPath()}.remove();
  }

  // create empty file4
  createFile(storeDir + std::string("/file4"), 0);

  MultiDiskAdaptor diskAdaptor;
  diskAdaptor.setFileEntries(std::begin(fs), std::end(fs));
  diskAdaptor.setPieceLength(1_k);
  diskAdaptor.openFile();

  auto allocitr = diskAdaptor.fileAllocationIterator();
  auto itr = dynamic_cast<MultiFileAllocationIterator*>(allocitr.get());

  auto& entries = itr->getDiskWriterEntries();
  CPPUNIT_ASSERT_EQUAL((size_t)11, entries.size());

  // file1
  CPPUNIT_ASSERT_EQUAL(storeDir + std::string("/file1"),
                       entries[0]->getFilePath());
  CPPUNIT_ASSERT(entries[0]->needsFileAllocation());
  CPPUNIT_ASSERT(entries[0]->getDiskWriter());
  // file2
  CPPUNIT_ASSERT_EQUAL(storeDir + std::string("/file2"),
                       entries[1]->getFilePath());
  CPPUNIT_ASSERT(entries[1]->needsFileAllocation());
  CPPUNIT_ASSERT(entries[1]->getDiskWriter());
  // file3
  CPPUNIT_ASSERT_EQUAL(storeDir + std::string("/file3"),
                       entries[2]->getFilePath());
  CPPUNIT_ASSERT(entries[2]->needsFileAllocation());