mte/unikernel/duniverse/base/test/test_ref.ml
2025-11-11 02:07:51 +01:00

69 lines
1.6 KiB
OCaml

open! Import
open Ref
let%test_unit "[set_temporarily] without raise" =
let r = ref 0 in
[%test_result: int] ~expect:1 (set_temporarily r 1 ~f:(fun () -> !r));
[%test_result: int] ~expect:0 !r
;;
let%expect_test "[set_temporarily] with raise" =
let r = ref 0 in
require_does_raise [%here] (fun () ->
Nothing.unreachable_code (set_temporarily r 1 ~f:(fun () -> failwith "")));
[%expect {| (Failure "") |}];
require_equal [%here] (module Int) !r 0
;;
let%test_unit "[set_temporarily] where [f] sets the ref" =
let r = ref 0 in
set_temporarily r 1 ~f:(fun () -> r := 2);
[%test_result: int] ~expect:0 !r
;;
let%expect_test "[sets_temporarily] without raise" =
let r1 = ref 1 in
let r2 = ref 2 in
let test and_values =
let i1 = !r1 in
let i2 = !r2 in
sets_temporarily and_values ~f:(fun () ->
print_s [%message (r1 : int ref) (r2 : int ref)]);
require_equal
[%here]
(module struct
type t = int * int [@@deriving equal, sexp_of]
end)
(!r1, !r2)
(i1, i2)
in
test [];
[%expect {|
((r1 1)
(r2 2))
|}];
test [ T (r1, 13) ];
[%expect {|
((r1 13)
(r2 2))
|}];
test [ T (r1, 13); T (r1, 17) ];
[%expect {|
((r1 17)
(r2 2))
|}];
test [ T (r1, 13); T (r2, 17) ];
[%expect {|
((r1 13)
(r2 17))
|}]
;;
let%expect_test "[sets_temporarily] with raise" =
let r = ref 0 in
require_does_raise [%here] (fun () ->
Nothing.unreachable_code (sets_temporarily [ T (r, 1) ] ~f:(fun () -> failwith "")));
[%expect {| (Failure "") |}];
print_s [%message (r : int ref)];
[%expect {| (r 0) |}]
;;