admin/maxDcb-C2Core
1
0
Fork 0
mirror of https://github.com/maxDcb/C2Core synced 2026-06-08 15:48:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve module tests

Browse Source
This commit is contained in:
Maxime dcb
2025-07-30 21:18:43 +02:00
parent 02a7db29b4
commit 63b37f05a9
4 changed files with 176 additions and 239 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ModuleTemplate/tests/testsModuleTemplate.cpp
+37 -39
View File
@@ -29,49 +29,47 @@ bool testModuleTemplate()
{
std::unique_ptr<ModuleTemplate> moduleTemplate = std::make_unique<ModuleTemplate>();
bool ok = true;
// ----- correct argument -----
{
std::vector<std::string> splitedCmd;
splitedCmd.push_back("moduleTemplate");
splitedCmd.push_back("arg1");
C2Message c2Message;
C2Message c2RetMessage;
moduleTemplate->init(splitedCmd, c2Message);
moduleTemplate->process(c2Message, c2RetMessage);
std::string output = "output:\n";
output += c2RetMessage.returnvalue();
output += "\n";
std::cout << output << std::endl;
std::vector<std::string> cmd = {"moduleTemplate", "arg1"};
C2Message msg, ret;
moduleTemplate->init(cmd, msg);
moduleTemplate->process(msg, ret);
ok &= ret.returnvalue().find("return value") != std::string::npos;
}
// ----- wrong argument triggers error -----
{
std::vector<std::string> splitedCmd;
splitedCmd.push_back("moduleTemplate");
splitedCmd.push_back("notarg1");
C2Message c2Message;
C2Message c2RetMessage;
moduleTemplate->init(splitedCmd, c2Message);
moduleTemplate->process(c2Message, c2RetMessage);
std::string output = "output:\n";
output += c2RetMessage.returnvalue();
output += "\n";
std::cout << output << std::endl;
if (c2RetMessage.errorCode())
{
std::string errorMsg;
moduleTemplate->errorCodeToMsg(c2RetMessage, errorMsg);
std::cout << "[+] error: \n" << errorMsg << std::endl;
}
else
{
return false;
}
std::vector<std::string> cmd = {"moduleTemplate", "notarg1"};
C2Message msg, ret;
moduleTemplate->init(cmd, msg);
moduleTemplate->process(msg, ret);
std::string err;
moduleTemplate->errorCodeToMsg(ret, err);
#ifdef BUILD_TEAMSERVER
ok &= ret.errorCode() == 1 && !err.empty();
#else
ok &= ret.errorCode() == 1;
#endif
}
return true;
// ----- missing argument -----
{
std::vector<std::string> cmd = {"moduleTemplate"};
C2Message msg, ret;
ok &= moduleTemplate->init(cmd, msg) == -1;
ok &= !msg.returnvalue().empty();
}
// ----- followUp path -----
{
C2Message ret;
ret.set_errorCode(-1);
ret.set_args("dummy");
ok &= moduleTemplate->followUp(ret) == 0;
}
return ok;
}