mirror of
https://github.com/CachyOS/New-Cli-Installer.git
synced 2025-02-02 22:07:13 +08:00
150 lines
3.5 KiB
C++
150 lines
3.5 KiB
C++
#include "gucc/file_utils.hpp"
|
|
#include "gucc/initcpio.hpp"
|
|
#include "gucc/logger.hpp"
|
|
|
|
#include <cassert>
|
|
#include <filesystem>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <string_view>
|
|
#include <vector>
|
|
|
|
#include <spdlog/sinks/callback_sink.h>
|
|
#include <spdlog/spdlog.h>
|
|
|
|
namespace fs = std::filesystem;
|
|
|
|
static constexpr auto MKINITCPIO_STR = R"(
|
|
# MODULES
|
|
MODULES=()
|
|
|
|
# BINARIES
|
|
BINARIES=()
|
|
|
|
# FILES
|
|
FILES=()
|
|
|
|
# HOOKS
|
|
HOOKS=(base udev autodetect modconf block filesystems keyboard fsck)
|
|
)";
|
|
|
|
static constexpr auto MKINITCPIO_INVALID_TEST = R"(
|
|
# MODULES
|
|
MODULES=()
|
|
|
|
(
|
|
# BINARIES
|
|
BINARIES=()
|
|
)
|
|
# FILES
|
|
FILES=()
|
|
()
|
|
# HOOKS
|
|
HOOKS=(base udev autodetect modconf block filesystems keyboard fsck
|
|
|
|
)";
|
|
|
|
static constexpr auto MKINITCPIO_TEST = R"(
|
|
# MODULES
|
|
MODULES=(radeon crc32c-intel)
|
|
|
|
# BINARIES
|
|
BINARIES=()
|
|
|
|
# FILES
|
|
FILES=()
|
|
|
|
# HOOKS
|
|
HOOKS=(base udev autodetect modconf block filesystems keyboard fsck btrfs usr lvm2 zfs)
|
|
)";
|
|
|
|
int main() {
|
|
auto callback_sink = std::make_shared<spdlog::sinks::callback_sink_mt>([](const spdlog::details::log_msg&) {
|
|
// noop
|
|
});
|
|
auto logger = std::make_shared<spdlog::logger>("default", callback_sink);
|
|
spdlog::set_default_logger(logger);
|
|
gucc::logger::set_logger(logger);
|
|
|
|
using namespace gucc; // NOLINT
|
|
|
|
static constexpr std::string_view filename{"/tmp/mkinitcpio.conf"};
|
|
|
|
// Open mkinitcpio file for writing.
|
|
std::ofstream mkinitcpio_file{filename.data()};
|
|
assert(mkinitcpio_file.is_open());
|
|
|
|
// Setup mkinitcpio file.
|
|
mkinitcpio_file << MKINITCPIO_STR;
|
|
mkinitcpio_file.close();
|
|
|
|
auto initcpio = detail::Initcpio{filename};
|
|
|
|
// Insert data.
|
|
initcpio.append_module("radeon");
|
|
initcpio.append_hook("btrfs");
|
|
initcpio.append_module("crc32c-intel");
|
|
initcpio.append_hooks({"usr", "lvm2", "zfs"});
|
|
|
|
// Write data.
|
|
assert(initcpio.write());
|
|
|
|
// Checking insertion of equal items
|
|
assert(!initcpio.append_module("radeon"));
|
|
assert(!initcpio.append_hook("btrfs"));
|
|
assert(!initcpio.append_module("crc32c-intel"));
|
|
assert(!initcpio.insert_hook("btrfs", {"usr", "lvm2", "zfs"}));
|
|
|
|
// Write data.
|
|
assert(initcpio.write());
|
|
|
|
// Check if file is equal to test data.
|
|
// "\n# MODULES\nMODULES=(crc32c-intel)\n\n# BINARIES\nBINARIES=()\n\n# FILES\nFILES=()\n\n# HOOKS\nHOOKS=(base usr lvm2 zfs)"\n
|
|
const auto& file_content = file_utils::read_whole_file(filename);
|
|
assert(file_content == MKINITCPIO_TEST);
|
|
|
|
// Cleanup.
|
|
fs::remove(filename);
|
|
|
|
// check invalid file
|
|
mkinitcpio_file = std::ofstream{filename.data()};
|
|
assert(mkinitcpio_file.is_open());
|
|
|
|
mkinitcpio_file << MKINITCPIO_INVALID_TEST;
|
|
mkinitcpio_file.close();
|
|
|
|
initcpio = detail::Initcpio{filename};
|
|
assert(initcpio.parse_file());
|
|
|
|
assert(initcpio.modules.empty());
|
|
assert(initcpio.files.empty());
|
|
assert(initcpio.hooks.empty());
|
|
|
|
// Cleanup.
|
|
fs::remove(filename);
|
|
|
|
// check empty file
|
|
mkinitcpio_file = std::ofstream{filename.data()};
|
|
assert(mkinitcpio_file.is_open());
|
|
|
|
mkinitcpio_file << "";
|
|
mkinitcpio_file.close();
|
|
|
|
initcpio = detail::Initcpio{filename};
|
|
assert(!initcpio.parse_file());
|
|
assert(!initcpio.write());
|
|
|
|
// Cleanup.
|
|
fs::remove(filename);
|
|
|
|
// check write to nonexistent file
|
|
initcpio = detail::Initcpio{"/path/to/non/exist_file.conf"};
|
|
assert(!initcpio.parse_file());
|
|
assert(!initcpio.write());
|
|
|
|
// check write to root file
|
|
initcpio = detail::Initcpio{"/etc/mtab"};
|
|
assert(!initcpio.parse_file());
|
|
assert(!initcpio.write());
|
|
}
|