Code/UnitTests/FoundationTest/IO/ArchiveTest.cpp - dev - redgpu/ezEngine

Code/UnitTests/FoundationTest/IO/ArchiveTest.cpp
161 строка5 KB
Jan Krassnigg
Some ezFoundation changes (#1317)
04 июн 2024, 13:15
04 июн 2024, 13:15ded6365
#include <FoundationTest/FoundationTestPCH.h>

#include <Foundation/IO/Archive/Archive.h>
#include <Foundation/IO/Archive/DataDirTypeArchive.h>
#include <Foundation/IO/FileSystem/DataDirTypeFolder.h>
#include <Foundation/IO/FileSystem/FileReader.h>
#include <Foundation/IO/FileSystem/FileSystem.h>
#include <Foundation/IO/FileSystem/FileWriter.h>
#include <Foundation/System/Process.h>
#include <Foundation/Utilities/CommandLineUtils.h>

#if (EZ_ENABLED(EZ_SUPPORTS_FILE_ITERATORS) && EZ_ENABLED(EZ_SUPPORTS_FILE_STATS) && defined(BUILDSYSTEM_HAS_ARCHIVE_TOOL))

EZ_CREATE_SIMPLE_TEST(IO, Archive)
{
  ezStringBuilder sOutputFolder = ezTestFramework::GetInstance()->GetAbsOutputPath();
  sOutputFolder.AppendPath("ArchiveTest");
  sOutputFolder.MakeCleanPath();

  // make sure it is empty
  ezOSFile::DeleteFolder(sOutputFolder).IgnoreResult();
  ezOSFile::CreateDirectoryStructure(sOutputFolder).IgnoreResult();

  if (!EZ_TEST_BOOL(ezFileSystem::AddDataDirectory(sOutputFolder, "Clear", "output", ezDataDirUsage::AllowWrites).Succeeded()))
    return;

  const char* szTestData = "TestData";
  const char* szUnpackedData = "Unpacked";

  // write a couple of files for packaging
  const char* szFileList[] = {
    "File1.txt",
    "FolderA/File2.jpg",         // should get stored uncompressed
    "FolderB/File3.txt",
    "FolderA/FolderC/File4.zip", // should get stored uncompressed
    "FolderA/FolderD/File5.txt",
    "File6.txt",
  };

  const ezUInt32 uiMinFileSize = 1024 * 128;


  EZ_TEST_BLOCK(ezTestBlock::Enabled, "Generate Data")
  {
    ezUInt64 uiValue = 0;

    ezStringBuilder fileName;

    for (ezUInt32 uiFileIdx = 0; uiFileIdx < EZ_ARRAY_SIZE(szFileList); ++uiFileIdx)
    {
      fileName.Set(":output/", szTestData, "/", szFileList[uiFileIdx]);

      ezFileWriter file;
      if (!EZ_TEST_BOOL(file.Open(fileName).Succeeded()))
        return;

      for (ezUInt32 i = 0; i < uiMinFileSize * uiFileIdx; ++i)
      {
        file << uiValue;
        ++uiValue;
      }
    }
  }

  const ezStringBuilder sArchiveFolder(sOutputFolder, "/", szTestData);
  const ezStringBuilder sUnpackFolder(sOutputFolder, "/", szUnpackedData);
  const ezStringBuilder sArchiveFile(sOutputFolder, "/", szTestData, ".ezArchive");

  ezStringBuilder pathToArchiveTool = ezCommandLineUtils::GetGlobalInstance()->GetParameter(0);
  pathToArchiveTool.PathParentDirectory();
  pathToArchiveTool.AppendPath("ezArchiveTool");
#  if EZ_ENABLED(EZ_PLATFORM_WINDOWS)
  pathToArchiveTool.Append(".exe");
#  endif
  EZ_TEST_BLOCK(ezTestBlock::Enabled, "Create a Package")
  {

    ezProcessOptions opt;
    opt.m_sProcess = pathToArchiveTool;
    opt.m_Arguments.PushBack(sArchiveFolder);

    ezInt32 iReturnValue = 1;

    ezProcess ArchiveToolProc;
    if (!EZ_TEST_BOOL(ArchiveToolProc.Execute(opt, &iReturnValue).Succeeded()))
      return;

    EZ_TEST_INT(iReturnValue, 0);
  }

  EZ_TEST_BLOCK(ezTestBlock::Enabled, "Unpack the Package")
  {

    ezProcessOptions opt;
    opt.m_sProcess = pathToArchiveTool;
    opt.m_Arguments.PushBack("-unpack");
    opt.m_Arguments.PushBack(sArchiveFile);
    opt.m_Arguments.PushBack("-out");
    opt.m_Arguments.PushBack(sUnpackFolder);

    ezInt32 iReturnValue = 1;

    ezProcess ArchiveToolProc;
    if (!EZ_TEST_BOOL(ArchiveToolProc.Execute(opt, &iReturnValue).Succeeded()))
      return;

    EZ_TEST_INT(iReturnValue, 0);
  }

  EZ_TEST_BLOCK(ezTestBlock::Enabled, "Compare unpacked data")
  {
    ezUInt64 uiValue = 0;

    ezStringBuilder sFileSrc;
    ezStringBuilder sFileDst;

    for (ezUInt32 uiFileIdx = 0; uiFileIdx < EZ_ARRAY_SIZE(szFileList); ++uiFileIdx)
    {
      sFileSrc.Set(sOutputFolder, "/", szTestData, "/", szFileList[uiFileIdx]);
      sFileDst.Set(sOutputFolder, "/", szUnpackedData, "/", szFileList[uiFileIdx]);

      EZ_TEST_FILES(sFileSrc, sFileDst, "Unpacked file should be identical");
    }
  }

  EZ_TEST_BLOCK(ezTestBlock::Enabled, "Mount as Data Dir")
  {
    if (!EZ_TEST_BOOL(ezFileSystem::AddDataDirectory(sArchiveFile, "Clear", "archive", ezDataDirUsage::ReadOnly) == EZ_SUCCESS))
      return;

    ezStringBuilder sFileSrc;
    ezStringBuilder sFileDst;

    // test opening multiple files in parallel and keeping them open
    ezFileReader readers[EZ_ARRAY_SIZE(szFileList)];
    for (ezUInt32 uiFileIdx = 0; uiFileIdx < EZ_ARRAY_SIZE(szFileList); ++uiFileIdx)
    {
      sFileDst.Set(":archive/", szFileList[uiFileIdx]);
      EZ_TEST_BOOL(readers[uiFileIdx].Open(sFileDst).Succeeded());

      // advance the reader a bit
      EZ_TEST_INT(readers[uiFileIdx].SkipBytes(uiMinFileSize * uiFileIdx), uiMinFileSize * uiFileIdx);
    }

    for (ezUInt32 uiFileIdx = 0; uiFileIdx < EZ_ARRAY_SIZE(szFileList); ++uiFileIdx)
    {
      sFileSrc.Set(":output/", szTestData, "/", szFileList[uiFileIdx]);
      sFileDst.Set(":archive/", szFileList[uiFileIdx]);

      EZ_TEST_FILES(sFileSrc, sFileDst, "Unpacked file should be identical");
    }

    // mount a second time
    if (!EZ_TEST_BOOL(ezFileSystem::AddDataDirectory(sArchiveFile, "Clear", "archive2", ezDataDirUsage::ReadOnly) == EZ_SUCCESS))
      return;
  }

  ezFileSystem::RemoveDataDirectoryGroup("Clear");
}

#endif