mirror of
https://github.com/maxDcb/C2Core
synced 2026-06-08 15:48:01 +00:00
50 lines
1.1 KiB
C++
50 lines
1.1 KiB
C++
#include "../MkDir.hpp"
|
|
#include <filesystem>
|
|
|
|
bool testMkDir();
|
|
|
|
int main()
|
|
{
|
|
bool res;
|
|
std::cout << "[+] testMkDir" << std::endl;
|
|
res = testMkDir();
|
|
if (res)
|
|
std::cout << "[+] Sucess" << std::endl;
|
|
else
|
|
std::cout << "[-] Failed" << std::endl;
|
|
|
|
return !res;
|
|
}
|
|
|
|
bool testMkDir()
|
|
{
|
|
namespace fs = std::filesystem;
|
|
fs::path temp = fs::temp_directory_path() / "c2core_mkdir_test";
|
|
fs::remove_all(temp);
|
|
bool ok = true;
|
|
|
|
{
|
|
MkDir mk;
|
|
std::vector<std::string> cmd = {"mkDir", temp.string()};
|
|
C2Message msg, ret;
|
|
mk.init(cmd, msg);
|
|
msg.set_cmd(temp.string());
|
|
mk.process(msg, ret);
|
|
ok &= ret.returnvalue().find("Directory created") != std::string::npos;
|
|
ok &= fs::exists(temp);
|
|
}
|
|
|
|
{
|
|
MkDir mk;
|
|
std::vector<std::string> cmd = {"mkDir", temp.string()};
|
|
C2Message msg, ret;
|
|
mk.init(cmd, msg);
|
|
msg.set_cmd(temp.string());
|
|
mk.process(msg, ret);
|
|
ok &= ret.returnvalue().find("Already exists") != std::string::npos;
|
|
}
|
|
|
|
fs::remove_all(temp);
|
|
return ok;
|
|
}
|