From e9a9e470eb0177c1ba222938e5cff02fb5b01729 Mon Sep 17 00:00:00 2001
From: Zero Fanker <zcyfksn@vip.qq.com>
Date: Sat, 13 Apr 2024 11:20:51 -0400
Subject: [PATCH] enriched test cases for ini .

---
 UnitTest/CIni_Test.cpp | 136 +++++++++++++++++++++++++++++++++++++----
 1 file changed, 125 insertions(+), 11 deletions(-)

diff --git a/UnitTest/CIni_Test.cpp b/UnitTest/CIni_Test.cpp
index 2ca6754..8c261e9 100644
--- a/UnitTest/CIni_Test.cpp
+++ b/UnitTest/CIni_Test.cpp
@@ -1,25 +1,139 @@
 #include "stdafx.h"
 #include "../MissionEditor/IniFile.h"
 
+class IniTestHelper
+{
+	std::string m_fileName;
+
+	void writeDownContent(const char* pContent) {
+		std::ofstream iniFile(m_fileName.c_str());
+		ASSERT(iniFile.is_open() == true);
+		iniFile << pContent;
+		iniFile.flush();
+		iniFile.close();
+	}
+
+	public :
+		IniTestHelper(std::string&& name, const char* pContent) :
+			m_fileName(std::move(name))
+		{
+			ASSERT(!m_fileName.empty());
+			ASSERT(pContent != nullptr);
+			writeDownContent(pContent);
+		}
+		~IniTestHelper() {
+			remove(m_fileName.c_str());
+		}
+
+};
+
 TEST(CIniFileClass, LoadFileTest) {
-	auto const iniContent = R"(
+	auto const fileName = "test.ini";
+	IniTestHelper helper(fileName, R"(
 [Debug]
 ;DisplayAllOverlay=Yes ; Doesn´t cripple the overlay list in any way
 ;EnableTrackLogic=Yes ; Enables Track Logic
 ;IgnoreSHPImageHeadUnused=Yes ; Use this *carefully* to make SHP graphics of some mods work that incorrectly have the shadow flag set
 AllowTunnels=yes
 AllowUnidirectionalTunnels=yes
-)";
 
-	auto const testIni = "test.ini";
-	std::ofstream iniFile(testIni);
-	iniFile << iniContent;
-	iniFile.flush();
-	iniFile.close();
+[BuildingVoxelTurretsRA2OLD]
+GTGCANX=00
+GTGCANY=44;-6
+)");
+
 
 	CIniFile file;
-	ASSERT_EQ(file.LoadFile(std::string(testIni)), 0);
-	remove(testIni);
+	ASSERT_EQ(file.LoadFile(std::string(fileName)), 0);
+
+	EXPECT_NE(file.Size(), 0);
+	EXPECT_EQ(true, file.GetBool("Debug", "AllowTunnels"));
+	EXPECT_EQ(false, file.GetBool("Debug", "DisplayAllOverlay"));
+	EXPECT_EQ("00", file.GetString("BuildingVoxelTurretsRA2OLD", "GTGCANX"));
+}
+
+
+TEST(CIniFileClass, IniSequenceTest) {
+	auto const fileName = "test.ini";
+	IniTestHelper helper(fileName, R"(
+[SlopeSetPiecesDirections]
+Count=10
+0=Right_1
+1=Left_1
+2=Top_1
+3=Bottom_1
+4=Right_2
+5=Left_2
+6=Left_2
+7=Bottom_2
+8=Top_2
+9=Top_2
+)");
+
+	CIniFile file;
+	ASSERT_EQ(file.LoadFile(std::string(fileName)), 0);
+	EXPECT_EQ(11, file["SlopeSetPiecesDirections"].Size());
+	EXPECT_EQ("10", file["SlopeSetPiecesDirections"].Nth(0).second);
+
+}
+
+TEST(CIniFileClass, IniSetValueTest) {
+	auto const fileName = "test.ini";
+	IniTestHelper helper(fileName, R"(
+[Debug]
+;DisplayAllOverlay=Yes ; Doesn´t cripple the overlay list in any way
+;EnableTrackLogic=Yes ; Enables Track Logic
+;IgnoreSHPImageHeadUnused=Yes ; Use this *carefully* to make SHP graphics of some mods work that incorrectly have the shadow flag set
+AllowTunnels=yes
+AllowUnidirectionalTunnels=yes
+)");
+
+	CIniFile file;
+	ASSERT_EQ(file.LoadFile(std::string(fileName)), 0);
+	// Test value not exists
+	EXPECT_EQ(false, file.GetBool("Debug", "DisplayAllOverlay"));
+	file.SetBool("Debug", "DisplayAllOverlay", true);
+	EXPECT_EQ(true, file.GetBool("Debug", "DisplayAllOverlay"));
+	// Test existed value and override
+	EXPECT_EQ(true, file.GetBool("Debug", "AllowTunnels"));
+	file.SetBool("Debug", "AllowTunnels", false);
+	EXPECT_EQ(false, file.GetBool("Debug", "AllowTunnels"));
+	// Test section not exists
+	EXPECT_EQ("", file.GetString("SlopeSetPiecesDirections", "0"));
+	file.SetString("SlopeSetPiecesDirections", "0", "Right_1");
+	EXPECT_EQ("Right_1", file.GetString("SlopeSetPiecesDirections", "0"));
+}
+
+TEST(CIniFileClass, IniSetSectionTest) {
+	auto const fileName = "test.ini";
+	IniTestHelper helper(fileName, R"(
+[SlopeSetPiecesDirections]
+Count=10
+0=Right_1
+1=Left_1
+2=Top_1
+3=Bottom_1
+4=Right_2
+5=Left_2
+6=Left_2
+7=Bottom_2
+8=Top_2
+9=Top_2
+)");
+
+	CIniFile file;
+	ASSERT_EQ(file.LoadFile(std::string(fileName)), 0);
+
+	CIniFile anotherIni;
+	anotherIni.AddSection("Debug");
+	auto pDebugSecAnother = anotherIni.TryGetSection("Debug");
+	ASSERT_NE(pDebugSecAnother, nullptr);
+	pDebugSecAnother->SetBool("DisplayAllOverlay", false);
+	pDebugSecAnother->SetBool("AllowTunnels", true);
+
+
+	file.SetSection("Debug", anotherIni["Debug"]);
+	EXPECT_EQ(true, file.GetBool("Debug", "AllowTunnels"));
+	EXPECT_EQ(false, file.GetBool("Debug", "DisplayAllOverlay"));
+}
 
-	EXPECT_EQ(true, file.GetBool("Debug","AllowTunnels"));
-}
\ No newline at end of file