初始化内容

project/dl_unit_test/test_tool.cpp

@@ -0,0 +1,146 @@
+/**
+* @file		test_tool.cpp
+* @brief	tool 模块单元测试（Archiver 归档器）
+*
+*
+* @version	1.0
+* @author	CodeBuddy
+* @date		26-09-03
+*
+* @note		
+*/
+#include <gtest/gtest.h>
+
+#include "tool/dl_tool_archiver.h"
+#include "io/dl_file.h"
+#include "render/dl_factory.h"
+
+namespace
+{
+//! 构造一个测试用的临时源目录（含 a.txt 与 sub/b.txt）
+bool CreateTestSrc(const std::string& dir_src)
+{
+	if (!dl::File::CreateFolder(dir_src))
+		return false;
+	if (!dl::File::CreateFolder(dir_src + "sub/"))
+		return false;
+	if (!dl::File::SaveString("hello world", dir_src + "a.txt"))
+		return false;
+	if (!dl::File::SaveString("second file", dir_src + "sub/b.txt"))
+		return false;
+	return true;
+}
+}
+
+TEST(ArchiverTest, CompressDecompress)
+{
+	std::string dir_src = dl::g_factory->GetPathTemp() + "dl_archiver_src/";
+	std::string dir_out = dl::g_factory->GetPathTemp() + "dl_archiver_out/";
+	std::string file_arc = dl::g_factory->GetPathTemp() + "dl_archiver.bin";
+
+	dl::File::Delete(dir_src, true);
+	dl::File::Delete(dir_out, true);
+	dl::File::Delete(file_arc);
+
+	ASSERT_TRUE(CreateTestSrc(dir_src));
+
+	dl::Tool::Archiver::Param p;
+	p._src = dir_src;
+	p._out = file_arc;
+	p._compressMode = dl::Memory::E_NONE;
+	p._bCrc = false;
+	dl::Tool::Archiver::Compress(p);
+	ASSERT_TRUE(dl::File::IsExist(file_arc));
+
+	ASSERT_TRUE(dl::Tool::Archiver::Decompress(file_arc, dir_out));
+	EXPECT_TRUE(dl::File::IsExist(dir_out + "a.txt"));
+	EXPECT_TRUE(dl::File::IsExist(dir_out + "sub/b.txt"));
+	EXPECT_EQ(dl::File::GetString(dir_out + "a.txt"), "hello world");
+	EXPECT_EQ(dl::File::GetString(dir_out + "sub/b.txt"), "second file");
+
+	dl::File::Delete(dir_src, true);
+	dl::File::Delete(dir_out, true);
+	dl::File::Delete(file_arc);
+}
+
+TEST(ArchiverTest, CompressDecompressLz4)
+{
+	std::string dir_src = dl::g_factory->GetPathTemp() + "dl_archiver_src/";
+	std::string dir_out = dl::g_factory->GetPathTemp() + "dl_archiver_out/";
+	std::string file_arc = dl::g_factory->GetPathTemp() + "dl_archiver.bin";
+
+	dl::File::Delete(dir_src, true);
+	dl::File::Delete(dir_out, true);
+	dl::File::Delete(file_arc);
+
+	ASSERT_TRUE(CreateTestSrc(dir_src));
+
+	dl::Tool::Archiver::Param p;
+	p._src = dir_src;
+	p._out = file_arc;
+	p._compressMode = dl::Memory::E_LZ4;
+	p._bCrc = false;
+	dl::Tool::Archiver::Compress(p);
+	ASSERT_TRUE(dl::File::IsExist(file_arc));
+
+	ASSERT_TRUE(dl::Tool::Archiver::Decompress(file_arc, dir_out));
+	EXPECT_TRUE(dl::File::IsExist(dir_out + "a.txt"));
+	EXPECT_TRUE(dl::File::IsExist(dir_out + "sub/b.txt"));
+	EXPECT_EQ(dl::File::GetString(dir_out + "a.txt"), "hello world");
+	EXPECT_EQ(dl::File::GetString(dir_out + "sub/b.txt"), "second file");
+
+	dl::File::Delete(dir_src, true);
+	dl::File::Delete(dir_out, true);
+	dl::File::Delete(file_arc);
+}
+
+TEST(ArchiverTest, GetFileList)
+{
+	std::string dir_src = dl::g_factory->GetPathTemp() + "dl_archiver_src/";
+	std::string file_arc = dl::g_factory->GetPathTemp() + "dl_archiver.bin";
+
+	dl::File::Delete(dir_src, true);
+	dl::File::Delete(file_arc);
+
+	ASSERT_TRUE(CreateTestSrc(dir_src));
+
+	dl::Tool::Archiver::Param p;
+	p._src = dir_src;
+	p._out = file_arc;
+	p._compressMode = dl::Memory::E_NONE;
+	p._bCrc = false;
+	dl::Tool::Archiver::Compress(p);
+
+	dl::Buffer buf = dl::File::GetBuffer(file_arc);
+	ASSERT_FALSE(buf.empty());
+
+	dl::Tool::Archiver::FileList list;
+	ASSERT_TRUE(dl::Tool::Archiver::GetFileList(buf, list));
+	EXPECT_EQ(list.size(), 2u);
+
+	bool found_a = false;
+	bool found_b = false;
+	for (auto& [name, info] : list)
+	{
+		if (name.find("a.txt") != std::string::npos)
+			found_a = true;
+		if (name.find("b.txt") != std::string::npos)
+			found_b = true;
+	}
+	EXPECT_TRUE(found_a);
+	EXPECT_TRUE(found_b);
+
+	dl::File::Delete(dir_src, true);
+	dl::File::Delete(file_arc);
+}
+
+TEST(ArchiverTest, DecompressBadType)
+{
+	std::string bad = "this is not an archiver file";
+	dl::Buffer buf = dl::CreateBuffer(dl::CreateBufferView(bad));
+
+	std::string dir_out = dl::g_factory->GetPathTemp() + "dl_archiver_bad/";
+	dl::File::Delete(dir_out, true);
+	EXPECT_FALSE(dl::Tool::Archiver::Decompress(buf, dir_out));
+	dl::File::Delete(dir_out, true);
+}