test_eventlog.cpp 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. #if _WIN32
  2. #include "includes.h"
  3. #include "test_sink.h"
  4. #include "spdlog/sinks/win_eventlog_sink.h"
  5. static const LPCSTR TEST_SOURCE = "spdlog_test";
  6. static void test_single_print(std::function<void(std::string const &)> do_log,
  7. std::string const &expected_contents,
  8. WORD expected_ev_type) {
  9. using namespace std::chrono;
  10. do_log(expected_contents);
  11. const auto expected_time_generated =
  12. duration_cast<seconds>(system_clock::now().time_since_epoch()).count();
  13. struct handle_t {
  14. HANDLE handle_;
  15. ~handle_t() {
  16. if (handle_) {
  17. REQUIRE(CloseEventLog(handle_));
  18. }
  19. }
  20. } event_log{::OpenEventLogA(nullptr, TEST_SOURCE)};
  21. REQUIRE(event_log.handle_);
  22. DWORD read_bytes{}, size_needed{};
  23. auto ok = ::ReadEventLogA(event_log.handle_, EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ,
  24. 0, &read_bytes, 0, &read_bytes, &size_needed);
  25. REQUIRE(!ok);
  26. REQUIRE(::GetLastError() == ERROR_INSUFFICIENT_BUFFER);
  27. std::vector<char> record_buffer(size_needed);
  28. PEVENTLOGRECORD record = (PEVENTLOGRECORD)record_buffer.data();
  29. ok = ::ReadEventLogA(event_log.handle_, EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ, 0,
  30. record, size_needed, &read_bytes, &size_needed);
  31. REQUIRE(ok);
  32. REQUIRE(record->NumStrings == 1);
  33. REQUIRE(record->EventType == expected_ev_type);
  34. REQUIRE((expected_time_generated - record->TimeGenerated) <= 3u);
  35. std::string message_in_log(((char *)record + record->StringOffset));
  36. REQUIRE(message_in_log == expected_contents + spdlog::details::os::default_eol);
  37. }
  38. TEST_CASE("eventlog", "[eventlog]") {
  39. using namespace spdlog;
  40. auto test_sink = std::make_shared<sinks::win_eventlog_sink_mt>(TEST_SOURCE);
  41. spdlog::logger test_logger("eventlog", test_sink);
  42. test_logger.set_level(level::trace);
  43. test_sink->set_pattern("%v");
  44. test_single_print([&test_logger](std::string const &msg) { test_logger.trace(msg); },
  45. "my trace message", EVENTLOG_SUCCESS);
  46. test_single_print([&test_logger](std::string const &msg) { test_logger.debug(msg); },
  47. "my debug message", EVENTLOG_SUCCESS);
  48. test_single_print([&test_logger](std::string const &msg) { test_logger.info(msg); },
  49. "my info message", EVENTLOG_INFORMATION_TYPE);
  50. test_single_print([&test_logger](std::string const &msg) { test_logger.warn(msg); },
  51. "my warn message", EVENTLOG_WARNING_TYPE);
  52. test_single_print([&test_logger](std::string const &msg) { test_logger.error(msg); },
  53. "my error message", EVENTLOG_ERROR_TYPE);
  54. test_single_print([&test_logger](std::string const &msg) { test_logger.critical(msg); },
  55. "my critical message", EVENTLOG_ERROR_TYPE);
  56. }
  57. #endif //_WIN32