2024-07-05 00:04:55 -07:00

118 lines
5.0 KiB
C++

#include "GameDependencies.h"
#include "Character.h"
#include "Entity.h"
#include "tinyxml2.h"
#include "BuffComponent.h"
#include "CharacterComponent.h"
class SavingTest : public GameDependenciesTest {
protected:
std::unique_ptr<Entity> entity;
std::unique_ptr<Character> character;
tinyxml2::XMLDocument doc;
tinyxml2::XMLPrinter printer{ 0, true, 0 };
void SetUp() override {
SetUpDependencies();
Game::zoneManager->LoadZone(LWOZONEID(1800, 2, 0));
static_cast<dServerMock*>(Game::server)->SetZoneId(1800);
entity = std::make_unique<Entity>(1, GameDependenciesTest::info);
character = std::make_unique<Character>(1, nullptr);
doc.LoadFile("./test_xml_data.xml");
entity->SetCharacter(character.get());
character->SetEntity(entity.get());
doc.Print(&printer);
character->_setXmlData(printer.CStr());
printer.ClearBuffer();
character->_doQuickXMLDataParse();
character->LoadXmlRespawnCheckpoints();
entity->AddComponent<BuffComponent>()->LoadFromXml(entity->GetCharacter()->GetXMLDoc());
entity->AddComponent<CharacterComponent>(character.get(), UNASSIGNED_SYSTEM_ADDRESS)->LoadFromXml(entity->GetCharacter()->GetXMLDoc());
}
void TearDown() override {
entity->SetCharacter(nullptr);
entity.reset();
character.reset();
TearDownDependencies();
}
};
TEST_F(SavingTest, CharacterComponentTest) {
// Print the original XML data
// character->GetXMLDoc().Print(&printer);
// std::string xmlDataOriginal(printer.CStr());
// printer.ClearBuffer();
// std::ofstream oldXml("./test_xml_data_original.xml");
// oldXml << xmlDataOriginal;
auto* characterComponent = entity->GetComponent<CharacterComponent>();
auto statsPrev = characterComponent->StatisticsToString();
auto claimCodesPrev = characterComponent->GetClaimCodes();
auto eyebrowsPrev = characterComponent->m_Character->GetEyebrows();
auto eyesPrev = characterComponent->m_Character->GetEyes();
auto hairColorPrev = characterComponent->m_Character->GetHairColor();
auto hairStylePrev = characterComponent->m_Character->GetHairStyle();
auto pantsColorPrev = characterComponent->m_Character->GetPantsColor();
auto leftHandPrev = characterComponent->m_Character->GetLeftHand();
auto shirtStylePrev = characterComponent->m_Character->GetShirtStyle();
auto mouthPrev = characterComponent->m_Character->GetMouth();
auto rightHandPrev = characterComponent->m_Character->GetRightHand();
auto shirtColorPrev = characterComponent->m_Character->GetShirtColor();
// Update the xml document so its been run through the saver
character->SaveXMLToDatabase();
// Reload the component from the now updated xml data
characterComponent = entity->AddComponent<CharacterComponent>(character.get(), UNASSIGNED_SYSTEM_ADDRESS);
characterComponent->LoadFromXml(entity->GetCharacter()->GetXMLDoc());
// Check that the buff component is the same as before which means resaving data and loading it back in didn't change anything
ASSERT_EQ(statsPrev, characterComponent->StatisticsToString());
ASSERT_EQ(claimCodesPrev, characterComponent->GetClaimCodes());
ASSERT_EQ(eyebrowsPrev, characterComponent->m_Character->GetEyebrows());
ASSERT_EQ(eyesPrev, characterComponent->m_Character->GetEyes());
ASSERT_EQ(hairColorPrev, characterComponent->m_Character->GetHairColor());
ASSERT_EQ(hairStylePrev, characterComponent->m_Character->GetHairStyle());
ASSERT_EQ(pantsColorPrev, characterComponent->m_Character->GetPantsColor());
ASSERT_EQ(leftHandPrev, characterComponent->m_Character->GetLeftHand());
ASSERT_EQ(mouthPrev, characterComponent->m_Character->GetMouth());
ASSERT_EQ(rightHandPrev, characterComponent->m_Character->GetRightHand());
ASSERT_EQ(shirtStylePrev, characterComponent->m_Character->GetShirtStyle());
ASSERT_EQ(shirtColorPrev, characterComponent->m_Character->GetShirtColor());
// Check that no data was lost during the saving process.
ASSERT_EQ("32114;69;343;13;163;2;181;2;388;252;146;24451;25;9022;41898;42186;42524;4404;0;0;0;0;0;0;0;0;0;", characterComponent->StatisticsToString());
// need a variable because the macro does not support {}
const std::array<uint64_t, 4> correctCodes = { 1073741968, 0, 0, 0 };
ASSERT_EQ(correctCodes, characterComponent->GetClaimCodes());
ASSERT_EQ(1, characterComponent->m_Character->GetEyebrows());
ASSERT_EQ(2, characterComponent->m_Character->GetEyes());
ASSERT_EQ(9, characterComponent->m_Character->GetHairColor());
ASSERT_EQ(8, characterComponent->m_Character->GetHairStyle());
ASSERT_EQ(3, characterComponent->m_Character->GetPantsColor());
ASSERT_EQ(27634704, characterComponent->m_Character->GetLeftHand());
ASSERT_EQ(3, characterComponent->m_Character->GetMouth());
ASSERT_EQ(27187396, characterComponent->m_Character->GetRightHand());
ASSERT_EQ(13, characterComponent->m_Character->GetShirtColor());
// Fails currently due to not reading style from xml
// ASSERT_EQ(27, characterComponent->m_Character->GetShirtStyle());
// character->GetXMLDoc().Print(&printer);
// std::string xmlDataModified(printer.CStr());
// printer.ClearBuffer();
// std::ofstream newXml("./test_xml_data_new.xml");
// newXml << xmlDataModified;
}