mirror of
https://github.com/maxDcb/C2Core
synced 2026-06-08 15:48:01 +00:00
maxdcb
101 lines
2.8 KiB
C++
101 lines
2.8 KiB
C++
#include "../Tree.hpp"
|
|
|
|
#include <chrono>
|
|
#include <filesystem>
|
|
#include <functional>
|
|
#include <string>
|
|
#include <thread>
|
|
|
|
#ifdef __linux__
|
|
#elif _WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
bool testTree();
|
|
|
|
static std::filesystem::path uniqueTestTempPath(const char* prefix)
|
|
{
|
|
const auto suffix = std::to_string(std::chrono::steady_clock::now().time_since_epoch().count())
|
|
+ "_" + std::to_string(std::hash<std::thread::id>{}(std::this_thread::get_id()));
|
|
return std::filesystem::temp_directory_path() / (std::string(prefix) + "_" + suffix);
|
|
}
|
|
|
|
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 !res;
|
|
}
|
|
|
|
bool testTree()
|
|
{
|
|
namespace fs = std::filesystem;
|
|
fs::path temp = uniqueTestTempPath("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);
|
|
std::string errorMsg;
|
|
tree->errorCodeToMsg(ret, errorMsg);
|
|
ok &= ret.errorCode() > 0;
|
|
ok &= ret.returnvalue().find("Error:") == 0;
|
|
ok &= errorMsg == ret.returnvalue();
|
|
}
|
|
|
|
// ----- 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;
|
|
}
|