101 lines
3.3 KiB
Rust
101 lines
3.3 KiB
Rust
use tracing::{subscriber::with_default, Id, Level, Span};
|
|
use tracing_attributes::instrument;
|
|
use tracing_mock::*;
|
|
|
|
#[instrument(follows_from = causes, skip(causes))]
|
|
fn with_follows_from_sync(causes: impl IntoIterator<Item = impl Into<Option<Id>>>) {}
|
|
|
|
#[instrument(follows_from = causes, skip(causes))]
|
|
async fn with_follows_from_async(causes: impl IntoIterator<Item = impl Into<Option<Id>>>) {}
|
|
|
|
#[instrument(follows_from = [&Span::current()])]
|
|
fn follows_from_current() {}
|
|
|
|
#[test]
|
|
fn follows_from_sync_test() {
|
|
let cause_a = span::mock().named("cause_a");
|
|
let cause_b = span::mock().named("cause_b");
|
|
let cause_c = span::mock().named("cause_c");
|
|
let consequence = span::mock().named("with_follows_from_sync");
|
|
|
|
let (subscriber, handle) = subscriber::mock()
|
|
.new_span(cause_a.clone())
|
|
.new_span(cause_b.clone())
|
|
.new_span(cause_c.clone())
|
|
.new_span(consequence.clone())
|
|
.follows_from(consequence.clone(), cause_a)
|
|
.follows_from(consequence.clone(), cause_b)
|
|
.follows_from(consequence.clone(), cause_c)
|
|
.enter(consequence.clone())
|
|
.exit(consequence)
|
|
.done()
|
|
.run_with_handle();
|
|
|
|
with_default(subscriber, || {
|
|
let cause_a = tracing::span!(Level::TRACE, "cause_a");
|
|
let cause_b = tracing::span!(Level::TRACE, "cause_b");
|
|
let cause_c = tracing::span!(Level::TRACE, "cause_c");
|
|
|
|
with_follows_from_sync(&[cause_a, cause_b, cause_c])
|
|
});
|
|
|
|
handle.assert_finished();
|
|
}
|
|
|
|
#[test]
|
|
fn follows_from_async_test() {
|
|
let cause_a = span::mock().named("cause_a");
|
|
let cause_b = span::mock().named("cause_b");
|
|
let cause_c = span::mock().named("cause_c");
|
|
let consequence = span::mock().named("with_follows_from_async");
|
|
|
|
let (subscriber, handle) = subscriber::mock()
|
|
.new_span(cause_a.clone())
|
|
.new_span(cause_b.clone())
|
|
.new_span(cause_c.clone())
|
|
.new_span(consequence.clone())
|
|
.follows_from(consequence.clone(), cause_a)
|
|
.follows_from(consequence.clone(), cause_b)
|
|
.follows_from(consequence.clone(), cause_c)
|
|
.enter(consequence.clone())
|
|
.exit(consequence.clone())
|
|
.enter(consequence.clone())
|
|
.exit(consequence)
|
|
.done()
|
|
.run_with_handle();
|
|
|
|
with_default(subscriber, || {
|
|
block_on_future(async {
|
|
let cause_a = tracing::span!(Level::TRACE, "cause_a");
|
|
let cause_b = tracing::span!(Level::TRACE, "cause_b");
|
|
let cause_c = tracing::span!(Level::TRACE, "cause_c");
|
|
|
|
with_follows_from_async(&[cause_a, cause_b, cause_c]).await
|
|
})
|
|
});
|
|
|
|
handle.assert_finished();
|
|
}
|
|
|
|
#[test]
|
|
fn follows_from_current_test() {
|
|
let cause = span::mock().named("cause");
|
|
let consequence = span::mock().named("follows_from_current");
|
|
|
|
let (subscriber, handle) = subscriber::mock()
|
|
.new_span(cause.clone())
|
|
.enter(cause.clone())
|
|
.new_span(consequence.clone())
|
|
.follows_from(consequence.clone(), cause.clone())
|
|
.enter(consequence.clone())
|
|
.exit(consequence)
|
|
.exit(cause)
|
|
.done()
|
|
.run_with_handle();
|
|
|
|
with_default(subscriber, || {
|
|
tracing::span!(Level::TRACE, "cause").in_scope(follows_from_current)
|
|
});
|
|
|
|
handle.assert_finished();
|
|
}
|