summaryrefslogtreecommitdiffstats
path: root/third_party/rust/tracing-attributes/tests/parents.rs
blob: 7069b98ea5d71104e6be1db07018676ccc36e3c5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
use tracing::{subscriber::with_default, Id, Level};
use tracing_attributes::instrument;
use tracing_mock::*;

#[instrument]
fn with_default_parent() {}

#[instrument(parent = parent_span, skip(parent_span))]
fn with_explicit_parent<P>(parent_span: P)
where
    P: Into<Option<Id>>,
{
}

#[test]
fn default_parent_test() {
    let contextual_parent = span::mock().named("contextual_parent");
    let child = span::mock().named("with_default_parent");

    let (subscriber, handle) = subscriber::mock()
        .new_span(
            contextual_parent
                .clone()
                .with_contextual_parent(None)
                .with_explicit_parent(None),
        )
        .new_span(
            child
                .clone()
                .with_contextual_parent(Some("contextual_parent"))
                .with_explicit_parent(None),
        )
        .enter(child.clone())
        .exit(child.clone())
        .enter(contextual_parent.clone())
        .new_span(
            child
                .clone()
                .with_contextual_parent(Some("contextual_parent"))
                .with_explicit_parent(None),
        )
        .enter(child.clone())
        .exit(child)
        .exit(contextual_parent)
        .done()
        .run_with_handle();

    with_default(subscriber, || {
        let contextual_parent = tracing::span!(Level::TRACE, "contextual_parent");

        with_default_parent();

        contextual_parent.in_scope(|| {
            with_default_parent();
        });
    });

    handle.assert_finished();
}

#[test]
fn explicit_parent_test() {
    let contextual_parent = span::mock().named("contextual_parent");
    let explicit_parent = span::mock().named("explicit_parent");
    let child = span::mock().named("with_explicit_parent");

    let (subscriber, handle) = subscriber::mock()
        .new_span(
            contextual_parent
                .clone()
                .with_contextual_parent(None)
                .with_explicit_parent(None),
        )
        .new_span(
            explicit_parent
                .with_contextual_parent(None)
                .with_explicit_parent(None),
        )
        .enter(contextual_parent.clone())
        .new_span(
            child
                .clone()
                .with_contextual_parent(Some("contextual_parent"))
                .with_explicit_parent(Some("explicit_parent")),
        )
        .enter(child.clone())
        .exit(child)
        .exit(contextual_parent)
        .done()
        .run_with_handle();

    with_default(subscriber, || {
        let contextual_parent = tracing::span!(Level::INFO, "contextual_parent");
        let explicit_parent = tracing::span!(Level::INFO, "explicit_parent");

        contextual_parent.in_scope(|| {
            with_explicit_parent(&explicit_parent);
        });
    });

    handle.assert_finished();
}