22c1ed9657
* mock: change helper functions to `expect::<thing>` The current format of test expectations in `tracing-mock` isn't ideal. The format `span::expect` requires importing `tracing_mock::<thing>` which may conflict with imports from other tracing crates, especially `tracing-core`. So we change the order and move the functions into a module called `expect` so that: * `event::expect` becomes `expect::event` * `span::expect` becomes `expect::span` * `field::expect` becomes `expect::field` This format has two advantages. 1. It reads as natural English, e.g "expect span" 2. It is no longer common to import the modules directly. Regarding point (2), the following format was previously common: ```rust use tracing_mock::field; field::expect(); ``` This import of the `field` module may then conflict with importing the same from `tracing_core`, making it necessary to rename one of the imports. The same code would now be written: ```rust use tracing_mock::expect; expect::field(); ``` Which is less likely to conflict. This change also fixes an unused warning on `MockHandle::new` when the `tracing-subscriber` feature is not enabled. Refs: #539
109 lines
2.6 KiB
Rust
109 lines
2.6 KiB
Rust
use tracing::collect::with_default;
|
|
use tracing_attributes::instrument;
|
|
use tracing_mock::*;
|
|
|
|
#[instrument]
|
|
fn default_target() {}
|
|
|
|
#[instrument(target = "my_target")]
|
|
fn custom_target() {}
|
|
|
|
mod my_mod {
|
|
use tracing_attributes::instrument;
|
|
|
|
pub const MODULE_PATH: &str = module_path!();
|
|
|
|
#[instrument]
|
|
pub fn default_target() {}
|
|
|
|
#[instrument(target = "my_other_target")]
|
|
pub fn custom_target() {}
|
|
}
|
|
|
|
#[test]
|
|
fn default_targets() {
|
|
let (collector, handle) = collector::mock()
|
|
.new_span(
|
|
expect::span()
|
|
.named("default_target")
|
|
.with_target(module_path!()),
|
|
)
|
|
.enter(
|
|
expect::span()
|
|
.named("default_target")
|
|
.with_target(module_path!()),
|
|
)
|
|
.exit(
|
|
expect::span()
|
|
.named("default_target")
|
|
.with_target(module_path!()),
|
|
)
|
|
.new_span(
|
|
expect::span()
|
|
.named("default_target")
|
|
.with_target(my_mod::MODULE_PATH),
|
|
)
|
|
.enter(
|
|
expect::span()
|
|
.named("default_target")
|
|
.with_target(my_mod::MODULE_PATH),
|
|
)
|
|
.exit(
|
|
expect::span()
|
|
.named("default_target")
|
|
.with_target(my_mod::MODULE_PATH),
|
|
)
|
|
.only()
|
|
.run_with_handle();
|
|
|
|
with_default(collector, || {
|
|
default_target();
|
|
my_mod::default_target();
|
|
});
|
|
|
|
handle.assert_finished();
|
|
}
|
|
|
|
#[test]
|
|
fn custom_targets() {
|
|
let (collector, handle) = collector::mock()
|
|
.new_span(
|
|
expect::span()
|
|
.named("custom_target")
|
|
.with_target("my_target"),
|
|
)
|
|
.enter(
|
|
expect::span()
|
|
.named("custom_target")
|
|
.with_target("my_target"),
|
|
)
|
|
.exit(
|
|
expect::span()
|
|
.named("custom_target")
|
|
.with_target("my_target"),
|
|
)
|
|
.new_span(
|
|
expect::span()
|
|
.named("custom_target")
|
|
.with_target("my_other_target"),
|
|
)
|
|
.enter(
|
|
expect::span()
|
|
.named("custom_target")
|
|
.with_target("my_other_target"),
|
|
)
|
|
.exit(
|
|
expect::span()
|
|
.named("custom_target")
|
|
.with_target("my_other_target"),
|
|
)
|
|
.only()
|
|
.run_with_handle();
|
|
|
|
with_default(collector, || {
|
|
custom_target();
|
|
my_mod::custom_target();
|
|
});
|
|
|
|
handle.assert_finished();
|
|
}
|