From 8005926bfc306f1ec8214a50d094b6a94668b946 Mon Sep 17 00:00:00 2001 From: Simon Cruanes Date: Tue, 21 Apr 2026 16:56:55 -0400 Subject: [PATCH] fix lwt test by providing a local ambient span provider --- test/lwt/t_lwt.real.ml | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/test/lwt/t_lwt.real.ml b/test/lwt/t_lwt.real.ml index 452b03c..dd863e7 100644 --- a/test/lwt/t_lwt.real.ml +++ b/test/lwt/t_lwt.real.ml @@ -3,10 +3,21 @@ open Test_trace_common let ( let@ ) = ( @@ ) +let k_ambient_span : Trace_core.span Lwt.key = Lwt.new_key () + +let ambient_span_provider : Trace_core.Ambient_span_provider.t = + ASP_some + ( (), + { + get_current_span = (fun () -> Lwt.get k_ambient_span); + with_current_span_set_to = + (fun () span f -> + Lwt.with_value k_ambient_span (Some span) (fun () -> f span)); + } ) let () = print_endline "=== ambient span (Lwt) ==="; - Trace_core.set_ambient_context_provider Trace_lwt.ambient_span_provider; + Trace_core.set_ambient_context_provider ambient_span_provider; let coll = make_recorder () in let@ () = Trace_core.with_setup_collector coll in