@@ -724,11 +724,11 @@ mod tests {
724
724
} )
725
725
. unwrap ( ) ;
726
726
client. shutdown ( ) . await ;
727
- while telemetry_srv. hits_async ( ) . await == 0 {
727
+ while telemetry_srv. calls_async ( ) . await == 0 {
728
728
sleep ( Duration :: from_millis ( 10 ) ) . await ;
729
729
}
730
730
// One payload generate-metrics
731
- telemetry_srv. assert_hits_async ( 1 ) . await ;
731
+ telemetry_srv. assert_calls_async ( 1 ) . await ;
732
732
}
733
733
734
734
#[ cfg_attr( miri, ignore) ]
@@ -739,7 +739,7 @@ mod tests {
739
739
let telemetry_srv = server
740
740
. mock_async ( |when, then| {
741
741
when. method ( POST )
742
- . body_contains ( r#""application":{"service_name":"test_service","service_version":"test_version","env":"test_env","language_name":"test_language","language_version":"test_language_version","tracer_version":"test_tracer_version"}"# ) ;
742
+ . body_includes ( r#""application":{"service_name":"test_service","service_version":"test_version","env":"test_env","language_name":"test_language","language_version":"test_language_version","tracer_version":"test_tracer_version"}"# ) ;
743
743
then. status ( 200 ) . body ( "" ) ;
744
744
} )
745
745
. await ;
@@ -765,7 +765,12 @@ mod tests {
765
765
} )
766
766
. unwrap ( ) ;
767
767
client. shutdown ( ) . await ;
768
+ // Wait for the server to receive at least one call, but don't hang forever.
769
+ let start = std:: time:: Instant :: now ( ) ;
768
770
while telemetry_srv. calls_async ( ) . await == 0 {
771
+ if start. elapsed ( ) > Duration :: from_secs ( 180 ) {
772
+ panic ! ( "telemetry server did not receive calls within timeout" ) ;
773
+ }
769
774
sleep ( Duration :: from_millis ( 10 ) ) . await ;
770
775
}
771
776
// One payload generate-metrics
0 commit comments