diff --git a/tests/per_logger_mixed_mode_test.cpp b/tests/per_logger_mixed_mode_test.cpp index 818ab56..40d612c 100644 --- a/tests/per_logger_mixed_mode_test.cpp +++ b/tests/per_logger_mixed_mode_test.cpp @@ -28,7 +28,7 @@ class CountingLogger : public logit::ILogger { } ~CountingLogger() override { - if (m_executor) m_executor->shutdown(); + shutdown(); } void log(const LogRecord& record, const std::string& message) override { @@ -56,6 +56,11 @@ class CountingLogger : public logit::ILogger { else logit::detail::TaskExecutor::get_instance().wait(); } + void shutdown() override { + wait(); + if (m_executor) m_executor->shutdown(); + } + std::size_t count() const { return m_count.load(std::memory_order_relaxed); } private: