Skip to content

Commit dc636df

Browse files
hdsdavidbarsky
authored andcommitted
mock: change helper functions to expect::<thing> (#2377)
* 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
1 parent 5048b86 commit dc636df

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

45 files changed

+873
-874
lines changed

tracing-attributes/tests/async_fn.rs

Lines changed: 38 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -84,15 +84,15 @@ fn repro_1831_2() -> impl Future<Output = Result<(), Infallible>> {
8484
#[test]
8585
fn async_fn_only_enters_for_polls() {
8686
let (subscriber, handle) = subscriber::mock()
87-
.new_span(span::expect().named("test_async_fn"))
88-
.enter(span::expect().named("test_async_fn"))
89-
.event(event::expect().with_fields(field::expect("awaiting").with_value(&true)))
90-
.exit(span::expect().named("test_async_fn"))
91-
.enter(span::expect().named("test_async_fn"))
92-
.exit(span::expect().named("test_async_fn"))
93-
.enter(span::expect().named("test_async_fn"))
94-
.exit(span::expect().named("test_async_fn"))
95-
.drop_span(span::expect().named("test_async_fn"))
87+
.new_span(expect::span().named("test_async_fn"))
88+
.enter(expect::span().named("test_async_fn"))
89+
.event(expect::event().with_fields(expect::field("awaiting").with_value(&true)))
90+
.exit(expect::span().named("test_async_fn"))
91+
.enter(expect::span().named("test_async_fn"))
92+
.exit(expect::span().named("test_async_fn"))
93+
.enter(expect::span().named("test_async_fn"))
94+
.exit(expect::span().named("test_async_fn"))
95+
.drop_span(expect::span().named("test_async_fn"))
9696
.only()
9797
.run_with_handle();
9898
with_default(subscriber, || {
@@ -113,14 +113,14 @@ fn async_fn_nested() {
113113
tracing::trace!(nested = true);
114114
}
115115

116-
let span = span::expect().named("test_async_fns_nested");
117-
let span2 = span::expect().named("test_async_fns_nested_other");
116+
let span = expect::span().named("test_async_fns_nested");
117+
let span2 = expect::span().named("test_async_fns_nested_other");
118118
let (subscriber, handle) = subscriber::mock()
119119
.new_span(span.clone())
120120
.enter(span.clone())
121121
.new_span(span2.clone())
122122
.enter(span2.clone())
123-
.event(event::expect().with_fields(field::expect("nested").with_value(&true)))
123+
.event(expect::event().with_fields(expect::field("nested").with_value(&true)))
124124
.exit(span2.clone())
125125
.enter(span2.clone())
126126
.exit(span2.clone())
@@ -191,26 +191,26 @@ fn async_fn_with_async_trait() {
191191
}
192192
}
193193

194-
let span = span::expect().named("foo");
195-
let span2 = span::expect().named("bar");
196-
let span3 = span::expect().named("baz");
194+
let span = expect::span().named("foo");
195+
let span2 = expect::span().named("bar");
196+
let span3 = expect::span().named("baz");
197197
let (subscriber, handle) = subscriber::mock()
198198
.new_span(
199199
span.clone()
200-
.with_field(field::expect("self"))
201-
.with_field(field::expect("v")),
200+
.with_field(expect::field("self"))
201+
.with_field(expect::field("v")),
202202
)
203203
.enter(span.clone())
204204
.new_span(span3.clone())
205205
.enter(span3.clone())
206-
.event(event::expect().with_fields(field::expect("val").with_value(&2u64)))
206+
.event(expect::event().with_fields(expect::field("val").with_value(&2u64)))
207207
.exit(span3.clone())
208208
.enter(span3.clone())
209209
.exit(span3.clone())
210210
.drop_span(span3)
211-
.new_span(span2.clone().with_field(field::expect("self")))
211+
.new_span(span2.clone().with_field(expect::field("self")))
212212
.enter(span2.clone())
213-
.event(event::expect().with_fields(field::expect("val").with_value(&5u64)))
213+
.event(expect::event().with_fields(expect::field("val").with_value(&5u64)))
214214
.exit(span2.clone())
215215
.enter(span2.clone())
216216
.exit(span2.clone())
@@ -255,15 +255,15 @@ fn async_fn_with_async_trait_and_fields_expressions() {
255255
async fn call(&mut self, _v: usize) {}
256256
}
257257

258-
let span = span::expect().named("call");
258+
let span = expect::span().named("call");
259259
let (subscriber, handle) = subscriber::mock()
260260
.new_span(
261261
span.clone().with_field(
262-
field::expect("_v")
262+
expect::field("_v")
263263
.with_value(&5usize)
264-
.and(field::expect("test").with_value(&tracing::field::debug(10)))
265-
.and(field::expect("val").with_value(&42u64))
266-
.and(field::expect("val2").with_value(&42u64)),
264+
.and(expect::field("test").with_value(&tracing::field::debug(10)))
265+
.and(expect::field("val").with_value(&42u64))
266+
.and(expect::field("val2").with_value(&42u64)),
267267
),
268268
)
269269
.enter(span.clone())
@@ -322,27 +322,27 @@ fn async_fn_with_async_trait_and_fields_expressions_with_generic_parameter() {
322322
async fn call_with_mut_self(&mut self) {}
323323
}
324324

325-
//let span = span::expect().named("call");
326-
let span2 = span::expect().named("call_with_self");
327-
let span3 = span::expect().named("call_with_mut_self");
328-
let span4 = span::expect().named("sync_fun");
325+
//let span = span::mock().named("call");
326+
let span2 = expect::span().named("call_with_self");
327+
let span3 = expect::span().named("call_with_mut_self");
328+
let span4 = expect::span().named("sync_fun");
329329
let (subscriber, handle) = subscriber::mock()
330330
/*.new_span(span.clone()
331331
.with_field(
332-
field::expect("Self").with_value(&"TestImpler")))
332+
expect::field("Self").with_value(&"TestImpler")))
333333
.enter(span.clone())
334334
.exit(span.clone())
335335
.drop_span(span)*/
336336
.new_span(
337337
span2
338338
.clone()
339-
.with_field(field::expect("Self").with_value(&std::any::type_name::<TestImpl>())),
339+
.with_field(expect::field("Self").with_value(&std::any::type_name::<TestImpl>())),
340340
)
341341
.enter(span2.clone())
342342
.new_span(
343343
span4
344344
.clone()
345-
.with_field(field::expect("Self").with_value(&std::any::type_name::<TestImpl>())),
345+
.with_field(expect::field("Self").with_value(&std::any::type_name::<TestImpl>())),
346346
)
347347
.enter(span4.clone())
348348
.exit(span4.clone())
@@ -355,7 +355,7 @@ fn async_fn_with_async_trait_and_fields_expressions_with_generic_parameter() {
355355
.new_span(
356356
span3
357357
.clone()
358-
.with_field(field::expect("Self").with_value(&std::any::type_name::<TestImpl>())),
358+
.with_field(expect::field("Self").with_value(&std::any::type_name::<TestImpl>())),
359359
)
360360
.enter(span3.clone())
361361
.exit(span3.clone())
@@ -397,7 +397,7 @@ fn out_of_scope_fields() {
397397
}
398398
}
399399

400-
let span = span::expect().named("call");
400+
let span = expect::span().named("call");
401401
let (subscriber, handle) = subscriber::mock()
402402
.new_span(span.clone())
403403
.enter(span.clone())
@@ -430,8 +430,8 @@ fn manual_impl_future() {
430430
}
431431
}
432432

433-
let span = span::expect().named("manual_impl_future");
434-
let poll_event = || event::expect().with_fields(field::expect("poll").with_value(&true));
433+
let span = expect::span().named("manual_impl_future");
434+
let poll_event = || expect::event().with_fields(expect::field("poll").with_value(&true));
435435

436436
let (subscriber, handle) = subscriber::mock()
437437
// await manual_impl_future
@@ -463,8 +463,8 @@ fn manual_box_pin() {
463463
})
464464
}
465465

466-
let span = span::expect().named("manual_box_pin");
467-
let poll_event = || event::expect().with_fields(field::expect("poll").with_value(&true));
466+
let span = expect::span().named("manual_box_pin");
467+
let poll_event = || expect::event().with_fields(expect::field("poll").with_value(&true));
468468

469469
let (subscriber, handle) = subscriber::mock()
470470
// await manual_box_pin

tracing-attributes/tests/destructuring.rs

Lines changed: 21 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,14 @@ fn destructure_tuples() {
77
#[instrument]
88
fn my_fn((arg1, arg2): (usize, usize)) {}
99

10-
let span = span::expect().named("my_fn");
10+
let span = expect::span().named("my_fn");
1111

1212
let (subscriber, handle) = subscriber::mock()
1313
.new_span(
1414
span.clone().with_field(
15-
field::expect("arg1")
15+
expect::field("arg1")
1616
.with_value(&format_args!("1"))
17-
.and(field::expect("arg2").with_value(&format_args!("2")))
17+
.and(expect::field("arg2").with_value(&format_args!("2")))
1818
.only(),
1919
),
2020
)
@@ -36,16 +36,16 @@ fn destructure_nested_tuples() {
3636
#[instrument]
3737
fn my_fn(((arg1, arg2), (arg3, arg4)): ((usize, usize), (usize, usize))) {}
3838

39-
let span = span::expect().named("my_fn");
39+
let span = expect::span().named("my_fn");
4040

4141
let (subscriber, handle) = subscriber::mock()
4242
.new_span(
4343
span.clone().with_field(
44-
field::expect("arg1")
44+
expect::field("arg1")
4545
.with_value(&format_args!("1"))
46-
.and(field::expect("arg2").with_value(&format_args!("2")))
47-
.and(field::expect("arg3").with_value(&format_args!("3")))
48-
.and(field::expect("arg4").with_value(&format_args!("4")))
46+
.and(expect::field("arg2").with_value(&format_args!("2")))
47+
.and(expect::field("arg3").with_value(&format_args!("3")))
48+
.and(expect::field("arg4").with_value(&format_args!("4")))
4949
.only(),
5050
),
5151
)
@@ -67,12 +67,12 @@ fn destructure_refs() {
6767
#[instrument]
6868
fn my_fn(&arg1: &usize) {}
6969

70-
let span = span::expect().named("my_fn");
70+
let span = expect::span().named("my_fn");
7171

7272
let (subscriber, handle) = subscriber::mock()
7373
.new_span(
7474
span.clone()
75-
.with_field(field::expect("arg1").with_value(&1usize).only()),
75+
.with_field(expect::field("arg1").with_value(&1usize).only()),
7676
)
7777
.enter(span.clone())
7878
.exit(span.clone())
@@ -94,14 +94,14 @@ fn destructure_tuple_structs() {
9494
#[instrument]
9595
fn my_fn(Foo(arg1, arg2): Foo) {}
9696

97-
let span = span::expect().named("my_fn");
97+
let span = expect::span().named("my_fn");
9898

9999
let (subscriber, handle) = subscriber::mock()
100100
.new_span(
101101
span.clone().with_field(
102-
field::expect("arg1")
102+
expect::field("arg1")
103103
.with_value(&format_args!("1"))
104-
.and(field::expect("arg2").with_value(&format_args!("2")))
104+
.and(expect::field("arg2").with_value(&format_args!("2")))
105105
.only(),
106106
),
107107
)
@@ -135,14 +135,14 @@ fn destructure_structs() {
135135
let _ = (arg1, arg2);
136136
}
137137

138-
let span = span::expect().named("my_fn");
138+
let span = expect::span().named("my_fn");
139139

140140
let (subscriber, handle) = subscriber::mock()
141141
.new_span(
142142
span.clone().with_field(
143-
field::expect("arg1")
143+
expect::field("arg1")
144144
.with_value(&format_args!("1"))
145-
.and(field::expect("arg2").with_value(&format_args!("2")))
145+
.and(expect::field("arg2").with_value(&format_args!("2")))
146146
.only(),
147147
),
148148
)
@@ -180,16 +180,16 @@ fn destructure_everything() {
180180
let _ = (arg1, arg2, arg3, arg4);
181181
}
182182

183-
let span = span::expect().named("my_fn");
183+
let span = expect::span().named("my_fn");
184184

185185
let (subscriber, handle) = subscriber::mock()
186186
.new_span(
187187
span.clone().with_field(
188-
field::expect("arg1")
188+
expect::field("arg1")
189189
.with_value(&format_args!("1"))
190-
.and(field::expect("arg2").with_value(&format_args!("2")))
191-
.and(field::expect("arg3").with_value(&format_args!("3")))
192-
.and(field::expect("arg4").with_value(&format_args!("4")))
190+
.and(expect::field("arg2").with_value(&format_args!("2")))
191+
.and(expect::field("arg3").with_value(&format_args!("3")))
192+
.and(expect::field("arg4").with_value(&format_args!("4")))
193193
.only(),
194194
),
195195
)

0 commit comments

Comments
 (0)