mirror of
https://github.com/laomms/wxhelper.git
synced 2024-12-23 21:09:26 +08:00
170 lines
7.1 KiB
C++
170 lines
7.1 KiB
C++
|
/*
|
||
|
* This content is released under the MIT License as specified in
|
||
|
* https://raw.githubusercontent.com/gabime/spdlog/master/LICENSE
|
||
|
*/
|
||
|
#include "includes.h"
|
||
|
|
||
|
#define TEST_FILENAME "test_logs/file_helper_test.txt"
|
||
|
|
||
|
using spdlog::details::file_helper;
|
||
|
|
||
|
static void write_with_helper(file_helper &helper, size_t howmany) {
|
||
|
spdlog::memory_buf_t formatted;
|
||
|
spdlog::fmt_lib::format_to(std::back_inserter(formatted), "{}", std::string(howmany, '1'));
|
||
|
helper.write(formatted);
|
||
|
helper.flush();
|
||
|
}
|
||
|
|
||
|
TEST_CASE("file_helper_filename", "[file_helper::filename()]") {
|
||
|
prepare_logdir();
|
||
|
|
||
|
file_helper helper;
|
||
|
spdlog::filename_t target_filename = SPDLOG_FILENAME_T(TEST_FILENAME);
|
||
|
helper.open(target_filename);
|
||
|
REQUIRE(helper.filename() == target_filename);
|
||
|
}
|
||
|
|
||
|
TEST_CASE("file_helper_size", "[file_helper::size()]") {
|
||
|
prepare_logdir();
|
||
|
spdlog::filename_t target_filename = SPDLOG_FILENAME_T(TEST_FILENAME);
|
||
|
size_t expected_size = 123;
|
||
|
{
|
||
|
file_helper helper;
|
||
|
helper.open(target_filename);
|
||
|
write_with_helper(helper, expected_size);
|
||
|
REQUIRE(static_cast<size_t>(helper.size()) == expected_size);
|
||
|
}
|
||
|
REQUIRE(get_filesize(TEST_FILENAME) == expected_size);
|
||
|
}
|
||
|
|
||
|
TEST_CASE("file_helper_reopen", "[file_helper::reopen()]") {
|
||
|
prepare_logdir();
|
||
|
spdlog::filename_t target_filename = SPDLOG_FILENAME_T(TEST_FILENAME);
|
||
|
file_helper helper;
|
||
|
helper.open(target_filename);
|
||
|
write_with_helper(helper, 12);
|
||
|
REQUIRE(helper.size() == 12);
|
||
|
helper.reopen(true);
|
||
|
REQUIRE(helper.size() == 0);
|
||
|
}
|
||
|
|
||
|
TEST_CASE("file_helper_reopen2", "[file_helper::reopen(false)]") {
|
||
|
prepare_logdir();
|
||
|
spdlog::filename_t target_filename = SPDLOG_FILENAME_T(TEST_FILENAME);
|
||
|
size_t expected_size = 14;
|
||
|
file_helper helper;
|
||
|
helper.open(target_filename);
|
||
|
write_with_helper(helper, expected_size);
|
||
|
REQUIRE(helper.size() == expected_size);
|
||
|
helper.reopen(false);
|
||
|
REQUIRE(helper.size() == expected_size);
|
||
|
}
|
||
|
|
||
|
static void test_split_ext(const spdlog::filename_t::value_type *fname,
|
||
|
const spdlog::filename_t::value_type *expect_base,
|
||
|
const spdlog::filename_t::value_type *expect_ext) {
|
||
|
spdlog::filename_t filename(fname);
|
||
|
spdlog::filename_t expected_base(expect_base);
|
||
|
spdlog::filename_t expected_ext(expect_ext);
|
||
|
|
||
|
spdlog::filename_t basename;
|
||
|
spdlog::filename_t ext;
|
||
|
std::tie(basename, ext) = file_helper::split_by_extension(filename);
|
||
|
REQUIRE(basename == expected_base);
|
||
|
REQUIRE(ext == expected_ext);
|
||
|
}
|
||
|
|
||
|
TEST_CASE("file_helper_split_by_extension", "[file_helper::split_by_extension()]") {
|
||
|
test_split_ext(SPDLOG_FILENAME_T("mylog.txt"), SPDLOG_FILENAME_T("mylog"),
|
||
|
SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T(".mylog.txt"), SPDLOG_FILENAME_T(".mylog"),
|
||
|
SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T(".mylog"), SPDLOG_FILENAME_T(".mylog"), SPDLOG_FILENAME_T(""));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("/aaa/bb.d/mylog"), SPDLOG_FILENAME_T("/aaa/bb.d/mylog"),
|
||
|
SPDLOG_FILENAME_T(""));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("/aaa/bb.d/mylog.txt"), SPDLOG_FILENAME_T("/aaa/bb.d/mylog"),
|
||
|
SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("aaa/bbb/ccc/mylog.txt"),
|
||
|
SPDLOG_FILENAME_T("aaa/bbb/ccc/mylog"), SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("aaa/bbb/ccc/mylog."), SPDLOG_FILENAME_T("aaa/bbb/ccc/mylog."),
|
||
|
SPDLOG_FILENAME_T(""));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("aaa/bbb/ccc/.mylog.txt"),
|
||
|
SPDLOG_FILENAME_T("aaa/bbb/ccc/.mylog"), SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("/aaa/bbb/ccc/mylog.txt"),
|
||
|
SPDLOG_FILENAME_T("/aaa/bbb/ccc/mylog"), SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("/aaa/bbb/ccc/.mylog"),
|
||
|
SPDLOG_FILENAME_T("/aaa/bbb/ccc/.mylog"), SPDLOG_FILENAME_T(""));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("../mylog.txt"), SPDLOG_FILENAME_T("../mylog"),
|
||
|
SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T(".././mylog.txt"), SPDLOG_FILENAME_T(".././mylog"),
|
||
|
SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T(".././mylog.txt/xxx"), SPDLOG_FILENAME_T(".././mylog.txt/xxx"),
|
||
|
SPDLOG_FILENAME_T(""));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("/mylog.txt"), SPDLOG_FILENAME_T("/mylog"),
|
||
|
SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("//mylog.txt"), SPDLOG_FILENAME_T("//mylog"),
|
||
|
SPDLOG_FILENAME_T(".txt"));
|
||
|
test_split_ext(SPDLOG_FILENAME_T(""), SPDLOG_FILENAME_T(""), SPDLOG_FILENAME_T(""));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("."), SPDLOG_FILENAME_T("."), SPDLOG_FILENAME_T(""));
|
||
|
test_split_ext(SPDLOG_FILENAME_T("..txt"), SPDLOG_FILENAME_T("."), SPDLOG_FILENAME_T(".txt"));
|
||
|
}
|
||
|
|
||
|
TEST_CASE("file_event_handlers", "[file_helper]") {
|
||
|
enum class flags { before_open, after_open, before_close, after_close };
|
||
|
prepare_logdir();
|
||
|
|
||
|
spdlog::filename_t test_filename = SPDLOG_FILENAME_T(TEST_FILENAME);
|
||
|
// define event handles that update vector of flags when called
|
||
|
std::vector<flags> events;
|
||
|
spdlog::file_event_handlers handlers;
|
||
|
handlers.before_open = [&](spdlog::filename_t filename) {
|
||
|
REQUIRE(filename == test_filename);
|
||
|
events.push_back(flags::before_open);
|
||
|
};
|
||
|
handlers.after_open = [&](spdlog::filename_t filename, std::FILE *fstream) {
|
||
|
REQUIRE(filename == test_filename);
|
||
|
REQUIRE(fstream);
|
||
|
fputs("after_open\n", fstream);
|
||
|
events.push_back(flags::after_open);
|
||
|
};
|
||
|
handlers.before_close = [&](spdlog::filename_t filename, std::FILE *fstream) {
|
||
|
REQUIRE(filename == test_filename);
|
||
|
REQUIRE(fstream);
|
||
|
fputs("before_close\n", fstream);
|
||
|
events.push_back(flags::before_close);
|
||
|
};
|
||
|
handlers.after_close = [&](spdlog::filename_t filename) {
|
||
|
REQUIRE(filename == test_filename);
|
||
|
events.push_back(flags::after_close);
|
||
|
};
|
||
|
{
|
||
|
spdlog::details::file_helper helper{handlers};
|
||
|
REQUIRE(events.empty());
|
||
|
|
||
|
helper.open(test_filename);
|
||
|
REQUIRE(events == std::vector<flags>{flags::before_open, flags::after_open});
|
||
|
|
||
|
events.clear();
|
||
|
helper.close();
|
||
|
REQUIRE(events == std::vector<flags>{flags::before_close, flags::after_close});
|
||
|
REQUIRE(file_contents(TEST_FILENAME) == "after_open\nbefore_close\n");
|
||
|
|
||
|
helper.reopen(true);
|
||
|
events.clear();
|
||
|
}
|
||
|
// make sure that the file_helper destrcutor calls the close callbacks if needed
|
||
|
REQUIRE(events == std::vector<flags>{flags::before_close, flags::after_close});
|
||
|
REQUIRE(file_contents(TEST_FILENAME) == "after_open\nbefore_close\n");
|
||
|
}
|
||
|
|
||
|
TEST_CASE("file_helper_open", "[file_helper]") {
|
||
|
prepare_logdir();
|
||
|
spdlog::filename_t target_filename = SPDLOG_FILENAME_T(TEST_FILENAME);
|
||
|
file_helper helper;
|
||
|
helper.open(target_filename);
|
||
|
helper.close();
|
||
|
|
||
|
target_filename += SPDLOG_FILENAME_T("/invalid");
|
||
|
REQUIRE_THROWS_AS(helper.open(target_filename), spdlog::spdlog_ex);
|
||
|
}
|