93 lines
3.1 KiB
C++
93 lines
3.1 KiB
C++
#include <td/protocol/command/CommandSerializer.h>
|
|
|
|
#include <td/misc/Test.h>
|
|
#include <td/protocol/command/CommandFactory.h>
|
|
|
|
namespace tp = td::protocol;
|
|
|
|
class Comparator {
|
|
public:
|
|
bool operator()(const tp::cdata::End& left, const tp::cdata::End& right) {
|
|
return true;
|
|
}
|
|
bool operator()(const tp::cdata::PlaceTower& left, const tp::cdata::PlaceTower& right) {
|
|
return left.m_Placer == right.m_Placer && left.m_Position.x == right.m_Position.x && left.m_Position.y == right.m_Position.y &&
|
|
left.m_Type == right.m_Type;
|
|
}
|
|
bool operator()(const tp::cdata::PlayerJoin& left, const tp::cdata::PlayerJoin& right) {
|
|
return left.m_ID == right.m_ID && left.m_Name == right.m_Name;
|
|
}
|
|
bool operator()(const tp::cdata::SpawnTroop& left, const tp::cdata::SpawnTroop& right) {
|
|
return left.m_Level == right.m_Level && left.m_Position.x == right.m_Position.x && left.m_Position.y == right.m_Position.y &&
|
|
left.m_Sender == right.m_Sender && left.m_Type == right.m_Type;
|
|
}
|
|
bool operator()(const tp::cdata::TeamChange& left, const tp::cdata::TeamChange& right) {
|
|
return left.m_NewTeam == right.m_NewTeam && left.m_Player == right.m_Player;
|
|
}
|
|
bool operator()(const tp::cdata::UpgradeTower& left, const tp::cdata::UpgradeTower& right) {
|
|
return left.m_Tower == right.m_Tower && left.m_Upgrade == right.m_Upgrade;
|
|
}
|
|
bool operator()(const tp::cdata::UseItem& left, const tp::cdata::UseItem& right) {
|
|
return left.m_Item == right.m_Item && left.m_Position.x == right.m_Position.x && left.m_Position.y == right.m_Position.y &&
|
|
left.m_User == right.m_User;
|
|
}
|
|
};
|
|
|
|
template <typename Command_T, typename Command_Data_T = typename Command_T::CommandDataType>
|
|
static int TestCommand(const Command_T& command) {
|
|
|
|
td::DataBuffer buffer = tp::CommandSerializer::Serialize(command);
|
|
|
|
auto abstractCommand = tp::CommandSerializer::Deserialize(buffer);
|
|
|
|
td_test_assert(abstractCommand);
|
|
|
|
Command_T* command2 = dynamic_cast<Command_T*>(abstractCommand.get());
|
|
|
|
td_test_assert(command2);
|
|
td_test_assert(command.GetType() == command2->GetType());
|
|
|
|
return Comparator{}(command.m_Data, command2->m_Data);
|
|
}
|
|
|
|
#define DeclareCommand(Command, ...) td_test_assert(TestCommand<tp::commands::Command>({}));
|
|
|
|
static int TestAllCommands() {
|
|
td_test_assert(TestCommand<tp::commands::End>({}));
|
|
td_test_assert(TestCommand<tp::commands::PlaceTower>({tp::cdata::PlaceTower{
|
|
td::TowerType::Necromancer,
|
|
8,
|
|
td::TowerCoords{-50, 69},
|
|
}}));
|
|
td_test_assert(TestCommand<tp::commands::PlayerJoin>({tp::cdata::PlayerJoin{
|
|
4,
|
|
"Persson",
|
|
}}));
|
|
td_test_assert(TestCommand<tp::commands::SpawnTroop>({tp::cdata::SpawnTroop{
|
|
td::EntityType::Blaze,
|
|
4,
|
|
td::EntityCoords{td::FpFloat{54}, td::FpFloat{58}},
|
|
2,
|
|
}}));
|
|
td_test_assert(TestCommand<tp::commands::TeamChange>({tp::cdata::TeamChange{
|
|
7,
|
|
td::Team::Red,
|
|
}}));
|
|
td_test_assert(TestCommand<tp::commands::UpgradeTower>({tp::cdata::UpgradeTower{
|
|
69,
|
|
3,
|
|
}}));
|
|
td_test_assert(TestCommand<tp::commands::UseItem>({tp::cdata::UseItem{
|
|
td::ShopItem::Freeze,
|
|
5,
|
|
td::EntityCoords{td::FpFloat{24}, td::FpFloat{-69}},
|
|
}}));
|
|
|
|
DeclareAllCommand()
|
|
|
|
return TD_TEST_SUCCESSFUL;
|
|
}
|
|
|
|
int main() {
|
|
return TestAllCommands();
|
|
} |