Revision control
Copy as Markdown
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
use std::sync::Once;
use tracing_subscriber::{fmt, prelude::*, EnvFilter};
static TESTING_SUBSCRIBER: Once = Once::new();
/// Initialize a logging environment suitable for testing. Logging can be configured in the
/// environment (eg, via the `RUST_LOG_LEVEL` variable), and if not so configured, will
/// default to the `Level::Error` level.
pub fn init_for_tests() {
// This is intended to be equivalent to `env_logger::try_init().ok();`
// `debug!()` output is seen. We could maybe add logging for `#[tracing::instrument]`?
TESTING_SUBSCRIBER.call_once(|| {
tracing_subscriber::registry()
.with(fmt::layer())
.with(EnvFilter::from_default_env())
.init();
});
}
/// Like `init_for_tests` but uses the specified `level` is logging is not configured in the environment.
pub fn init_for_tests_with_level(level: crate::Level) {
// This is intended to be equivalent to `env_logger::try_init().ok();`
// `debug!()` output is seen. We could maybe add logging for `#[tracing::instrument]`?
let level: tracing::Level = level.into();
TESTING_SUBSCRIBER.call_once(|| {
tracing_subscriber::registry()
.with(fmt::layer())
.with(
EnvFilter::builder()
.with_default_directive(level.into())
.from_env_lossy(),
)
.init();
});
}