Update test_eventlog.cpp

This commit is contained in:
gabime 2020-02-10 00:37:48 +02:00
parent 640921cd3f
commit 90454a93b2

View File

@ -1,20 +1,17 @@
#include "includes.h" #include "includes.h"
#include "test_sink.h" #include "test_sink.h"
#include "spdlog/fmt/bin_to_hex.h"
#if _WIN32 #if _WIN32
#include "spdlog/sinks/win_eventlog_sink.h" #include "spdlog/sinks/win_eventlog_sink.h"
static const LPCSTR TEST_LOG = "my log"; static const LPCSTR TEST_SOURCE = "spdlog_test";
static const LPCSTR TEST_SOURCE = "my source";
static void test_single_print(std::function<void(std::string const&)> do_print, std::string const& expectedContents, WORD expectedEventType) static void test_single_print(std::function<void(std::string const&)> do_log, std::string const& expected_contents, WORD expected_ev_type)
{ {
do_print(expectedContents);
using namespace std::chrono; using namespace std::chrono;
const auto expectedTimeGenerated = duration_cast<seconds>(system_clock::now().time_since_epoch()).count(); do_log(expected_contents);
const auto expected_time_generated = duration_cast<seconds>(system_clock::now().time_since_epoch()).count();
struct handle_t struct handle_t
{ {
@ -23,29 +20,31 @@ static void test_single_print(std::function<void(std::string const&)> do_print,
~handle_t() ~handle_t()
{ {
if (handle_) if (handle_)
{
REQUIRE(CloseEventLog(handle_)); REQUIRE(CloseEventLog(handle_));
} }
} eventLog {OpenEventLog(nullptr, TEST_SOURCE)}; }
} event_log {::OpenEventLog(nullptr, TEST_SOURCE)};
REQUIRE(eventLog.handle_); REQUIRE(event_log.handle_);
DWORD readBytes {}, sizeNeeded {}; DWORD read_bytes {}, size_needed{};
auto ok = ReadEventLog(eventLog.handle_, EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ, 0, &readBytes, 0, &readBytes, &sizeNeeded); auto ok = ::ReadEventLog(event_log.handle_, EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ, 0, &read_bytes, 0, &read_bytes, &size_needed);
REQUIRE(!ok); REQUIRE(!ok);
REQUIRE(GetLastError() == ERROR_INSUFFICIENT_BUFFER); REQUIRE(::GetLastError() == ERROR_INSUFFICIENT_BUFFER);
std::vector<char> recordBuffer(sizeNeeded); std::vector<char> record_buffer(size_needed);
PEVENTLOGRECORD record = (PEVENTLOGRECORD)recordBuffer.data(); PEVENTLOGRECORD record = (PEVENTLOGRECORD)record_buffer.data();
ok = ReadEventLog(eventLog.handle_, EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ, 0, record, sizeNeeded, &readBytes, &sizeNeeded); ok = ::ReadEventLog(event_log.handle_, EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ, 0, record, size_needed, &read_bytes, &size_needed);
REQUIRE(ok); REQUIRE(ok);
REQUIRE(record->NumStrings == 1); REQUIRE(record->NumStrings == 1);
REQUIRE(record->EventType == expectedEventType); REQUIRE(record->EventType == expected_ev_type);
REQUIRE(record->TimeGenerated == expectedTimeGenerated); REQUIRE(record->TimeGenerated == expected_time_generated);
std::string message_in_log(((char*) record + record->StringOffset)); std::string message_in_log(((char*) record + record->StringOffset));
REQUIRE(message_in_log == expectedContents + "\r\n"); REQUIRE(message_in_log == expected_contents + spdlog::details::os::default_eol);
} }
TEST_CASE("eventlog", "[eventlog]") TEST_CASE("eventlog", "[eventlog]")
@ -59,6 +58,7 @@ TEST_CASE("eventlog", "[eventlog]")
test_sink->set_pattern("%v"); test_sink->set_pattern("%v");
test_single_print([&test_logger] (std::string const& msg) { test_logger.trace(msg); }, "my trace message", EVENTLOG_SUCCESS);
test_single_print([&test_logger] (std::string const& msg) { test_logger.debug(msg); }, "my debug message", EVENTLOG_SUCCESS); test_single_print([&test_logger] (std::string const& msg) { test_logger.debug(msg); }, "my debug message", EVENTLOG_SUCCESS);
test_single_print([&test_logger] (std::string const& msg) { test_logger.info(msg); }, "my info message", EVENTLOG_INFORMATION_TYPE); test_single_print([&test_logger] (std::string const& msg) { test_logger.info(msg); }, "my info message", EVENTLOG_INFORMATION_TYPE);
test_single_print([&test_logger] (std::string const& msg) { test_logger.warn(msg); }, "my warn message", EVENTLOG_WARNING_TYPE); test_single_print([&test_logger] (std::string const& msg) { test_logger.warn(msg); }, "my warn message", EVENTLOG_WARNING_TYPE);