71 lines
3.1 KiB
C++
71 lines
3.1 KiB
C++
// Integration harness preflight: ensure daemon operations are test-safe.
|
|
#pragma once
|
|
#include <cstdlib>
|
|
#include <filesystem>
|
|
#include <string>
|
|
#include <yams/compat/unistd.h>
|
|
|
|
namespace yams::test::harnesses {
|
|
struct DaemonPreflightConfig {
|
|
std::filesystem::path runtime_dir; // directory for pid/socket
|
|
std::string socket_name_prefix{"yams-test-sock-"};
|
|
bool kill_others{false}; // never kill user/system daemons in tests
|
|
};
|
|
|
|
struct DaemonPreflight {
|
|
static void ensure_environment(const DaemonPreflightConfig& cfg) {
|
|
// 1) Prevent tests from stopping user/system daemons.
|
|
::setenv("YAMS_DAEMON_KILL_OTHERS", cfg.kill_others ? "1" : "0", 1);
|
|
|
|
// 2) Ensure a unique runtime dir per test process. Keep AF_UNIX path short.
|
|
std::error_code ec;
|
|
auto pid = static_cast<unsigned long>(::getpid());
|
|
auto runtime_root = cfg.runtime_dir;
|
|
auto unique_dir = runtime_root / (std::string("run-") + std::to_string(pid));
|
|
std::filesystem::create_directories(unique_dir, ec);
|
|
|
|
auto make_paths = [&](const std::filesystem::path& dir) {
|
|
auto sock = (dir / (cfg.socket_name_prefix + std::to_string(pid) + ".sock")).string();
|
|
auto pidf = (dir / "yamsd.pid").string();
|
|
return std::pair<std::string, std::string>(sock, pidf);
|
|
};
|
|
auto [socket_path, pid_path] = make_paths(unique_dir);
|
|
|
|
#if defined(__APPLE__) || defined(__unix__)
|
|
// macOS/Linux AF_UNIX sun_path is typically 104-108 bytes. If too long, fall back to /tmp.
|
|
const size_t kUnixMax = 100; // keep margin
|
|
if (socket_path.size() > kUnixMax) {
|
|
runtime_root = std::filesystem::path("/tmp");
|
|
unique_dir = runtime_root / (std::string("yamsr-") + std::to_string(pid));
|
|
std::filesystem::create_directories(unique_dir, ec);
|
|
auto p = make_paths(unique_dir);
|
|
socket_path = p.first;
|
|
pid_path = p.second;
|
|
}
|
|
#elif defined(_WIN32)
|
|
// Windows AF_UNIX sun_path is also limited to ~108 bytes. Deeply-nested
|
|
// per-test temp dirs easily exceed this (bind() then fails with a cryptic
|
|
// "Name component" error), so fall back to a short path directly under the
|
|
// system temp root when the computed socket path is too long.
|
|
const size_t kUnixMax = 100; // keep margin
|
|
if (socket_path.size() > kUnixMax) {
|
|
std::error_code tec;
|
|
runtime_root = std::filesystem::temp_directory_path(tec);
|
|
unique_dir = runtime_root / (std::string("yr") + std::to_string(pid));
|
|
std::filesystem::create_directories(unique_dir, ec);
|
|
auto p = make_paths(unique_dir);
|
|
socket_path = p.first;
|
|
pid_path = p.second;
|
|
}
|
|
#endif
|
|
::setenv("YAMS_RUNTIME_DIR", unique_dir.string().c_str(), 1);
|
|
::setenv("YAMS_SOCKET_PATH", socket_path.c_str(), 1);
|
|
::setenv("YAMS_PID_FILE", pid_path.c_str(), 1);
|
|
}
|
|
|
|
static void post_test_cleanup(const std::filesystem::path& runtime_root) {
|
|
std::error_code ec;
|
|
std::filesystem::remove_all(runtime_root, ec);
|
|
}
|
|
};
|
|
} // namespace yams::test::harnesses
|