follows_from.rs - mozsearch

Enable keyboard shortcuts

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();