mirror of
https://github.com/maxDcb/C2Core
synced 2026-06-08 15:48:01 +00:00
Maxime dcb
86 lines
2.2 KiB
C++
86 lines
2.2 KiB
C++
#include "../Tree.hpp"
|
|
|
|
#include <filesystem>
|
|
|
|
#ifdef __linux__
|
|
#elif _WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
bool testTree();
|
|
|
|
int main()
|
|
{
|
|
bool res;
|
|
|
|
std::cout << "[+] testTree" << std::endl;
|
|
res = testTree();
|
|
if (res)
|
|
std::cout << "[+] Sucess" << std::endl;
|
|
else
|
|
std::cout << "[-] Failed" << std::endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
bool testTree()
|
|
{
|
|
namespace fs = std::filesystem;
|
|
fs::path temp = fs::temp_directory_path() / "c2core_tree_test";
|
|
fs::remove_all(temp);
|
|
fs::create_directories(temp / "sub");
|
|
std::ofstream(temp / "file.txt") << "data";
|
|
std::ofstream(temp / "sub" / "inner.txt") << "data";
|
|
|
|
std::unique_ptr<Tree> tree = std::make_unique<Tree>();
|
|
bool ok = true;
|
|
|
|
// ----- explicit path -----
|
|
{
|
|
std::vector<std::string> cmd = {"tree", temp.string()};
|
|
C2Message msg, ret;
|
|
tree->init(cmd, msg);
|
|
msg.set_cmd(temp.string());
|
|
tree->process(msg, ret);
|
|
std::string out = ret.returnvalue();
|
|
ok &= out.find((temp / "file.txt").string()) != std::string::npos;
|
|
ok &= out.find((temp / "sub").string() + "\\") != std::string::npos;
|
|
}
|
|
|
|
// ----- path with spaces using split tokens -----
|
|
fs::path spacedDir = temp / "dir space";
|
|
fs::create_directories(spacedDir);
|
|
std::ofstream(spacedDir / "dummy.txt") << "a";
|
|
{
|
|
std::vector<std::string> cmd = {"tree", (temp.string()+"/dir").c_str(), "space"};
|
|
C2Message msg, ret;
|
|
tree->init(cmd, msg);
|
|
msg.set_cmd(spacedDir.string());
|
|
tree->process(msg, ret);
|
|
ok &= ret.returnvalue().find((spacedDir/"dummy.txt").string()) != std::string::npos;
|
|
}
|
|
|
|
// ----- invalid path -----
|
|
{
|
|
fs::path invalid = temp / "does_not_exist";
|
|
std::vector<std::string> cmd = {"tree", invalid.string()};
|
|
C2Message msg, ret;
|
|
tree->init(cmd, msg);
|
|
msg.set_cmd(invalid.string());
|
|
tree->process(msg, ret);
|
|
ok &= ret.returnvalue().empty();
|
|
}
|
|
|
|
// ----- no argument -----
|
|
{
|
|
std::vector<std::string> cmd = {"tree"};
|
|
C2Message msg, ret;
|
|
tree->init(cmd, msg);
|
|
tree->process(msg, ret);
|
|
ok &= !ret.returnvalue().empty();
|
|
}
|
|
|
|
fs::remove_all(temp);
|
|
return ok;
|
|
}
|