diff --git a/util/src/lib.rs b/util/src/lib.rs index 6b078ac7..e2e7ace6 100644 --- a/util/src/lib.rs +++ b/util/src/lib.rs @@ -35,7 +35,7 @@ extern crate serde_derive; // Logging related pub mod logger; -pub use logger::{LOGGER, init_logger}; +pub use logger::{LOGGER, init_logger, init_test_logger}; pub mod types; pub use types::LoggingConfig; diff --git a/util/src/logger.rs b/util/src/logger.rs index 1a38d173..51d2764b 100644 --- a/util/src/logger.rs +++ b/util/src/logger.rs @@ -89,3 +89,15 @@ pub fn init_logger(config: Option) { *was_init_ref = true; } } + +/// Initializes the logger for unit and integration tests +pub fn init_test_logger() { + let mut was_init_ref = WAS_INIT.lock().unwrap(); + if was_init_ref { + return; + } + let mut config_ref = LOGGING_CONFIG.lock().unwrap(); + *config_ref = LoggingConfig::default(); + *was_init_ref = true; +} +