From e2fb26b7fa35078af59158d9b3af765ac61f8ccd Mon Sep 17 00:00:00 2001
From: Simon Cruanes <simon.cruanes.2007@m4x.org>
Date: Tue, 7 Apr 2026 22:08:35 -0400
Subject: [PATCH] test ambient

---
 test/dune                 |  7 ++++++
 test/t_ambient_context.ml | 53 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 60 insertions(+)
 create mode 100644 test/t_ambient_context.ml

diff --git a/test/dune b/test/dune
index 0e2a295c..5a4b7c99 100644
--- a/test/dune
+++ b/test/dune
@@ -21,6 +21,13 @@
   trace-tef
   trace))
 
+(test
+ (name t_ambient_context)
+ (package moonpool)
+ (enabled_if
+  (and %{lib-available:ambient-context} %{lib-available:hmap}))
+ (libraries moonpool moonpool.ambient-context ambient-context))
+
 (test
  (name t_fib_await_mem)
  (package moonpool)
diff --git a/test/t_ambient_context.ml b/test/t_ambient_context.ml
new file mode 100644
index 00000000..db2a7b81
--- /dev/null
+++ b/test/t_ambient_context.ml
@@ -0,0 +1,53 @@
+open Moonpool
+
+let () = Ambient_context.set_current_storage Moonpool_ambient_context.storage
+let key_a : string Ambient_context.Context.key = Ambient_context.new_key ()
+let key_b : int Ambient_context.Context.key = Ambient_context.new_key ()
+let get_a () = Ambient_context.get key_a
+let get_b () = Ambient_context.get key_b
+
+let () =
+  let pool = Ws_pool.create ~num_threads:4 () in
+
+  (* basic: values set in a task are visible within that task *)
+  let fut =
+    Fut.spawn ~on:pool (fun () ->
+        assert (get_a () = None);
+        Ambient_context.with_key_bound_to key_a "hello" (fun () ->
+            assert (get_a () = Some "hello");
+            (* nested: inner binding shadows outer *)
+            Ambient_context.with_key_bound_to key_a "world" (fun () ->
+                assert (get_a () = Some "world"));
+            (* restored after inner scope *)
+            assert (get_a () = Some "hello"));
+        assert (get_a () = None))
+  in
+  Fut.wait_block_exn fut;
+
+  (* two keys are independent *)
+  let fut =
+    Fut.spawn ~on:pool (fun () ->
+        Ambient_context.with_key_bound_to key_a "foo" (fun () ->
+            Ambient_context.with_key_bound_to key_b 42 (fun () ->
+                assert (get_a () = Some "foo");
+                assert (get_b () = Some 42));
+            assert (get_b () = None);
+            assert (get_a () = Some "foo")))
+  in
+  Fut.wait_block_exn fut;
+
+  (* child tasks do not inherit the parent's ambient context *)
+  let fut =
+    Fut.spawn ~on:pool (fun () ->
+        Ambient_context.with_key_bound_to key_a "parent" (fun () ->
+            let child =
+              Fut.spawn ~on:pool (fun () ->
+                  (* child starts with empty context *)
+                  assert (get_a () = None))
+            in
+            Fut.wait_block_exn child))
+  in
+  Fut.wait_block_exn fut;
+
+  Ws_pool.shutdown pool;
+  Printf.printf "all assertions passed\n%!"