module Test = (val Containers_testlib.make ~__FILE__ ())
open Test
open CCLock;;

t @@ fun () ->
let l = create 0 in
let try_incr l =
  update l (fun x ->
      Thread.yield ();
      x + 1)
in
for _i = 1 to 10 do
  ignore (Thread.create try_incr l)
done;
Thread.delay 0.10;
assert_equal 10 (get l);
true
;;

t @@ fun () ->
let l = create 0 in
let test_it l =
  with_lock_as_ref l ~f:(fun r ->
      (* increment and decrement *)
      for j = 0 to 100 do
        let x = LockRef.get r in
        LockRef.set r (x + 10);
        if j mod 5 = 0 then Thread.yield ();
        let y = LockRef.get r in
        LockRef.set r (y - 10)
      done)
in
for _i = 1 to 100 do
  ignore (Thread.create test_it l)
done;
Thread.delay 0.10;
0 = get l
;;

t @@ fun () ->
let l = create 5 in
update l (fun x -> x + 1);
get l = 6
;;

t @@ fun () ->
let l = create 5 in
update_map l (fun x -> x + 1, string_of_int x) = "5" && get l = 6
;;

t @@ fun () ->
let l = create 0 in
set l 4;
get l = 4
;;

t @@ fun () ->
let l = create 0 in
set l 4;
set l 5;
get l = 5
;;

t @@ fun () ->
let l = create 0 in
let a = Array.init 100 (fun _ -> Thread.create (fun _ -> incr l) ()) in
Array.iter Thread.join a;
assert_equal ~printer:CCInt.to_string 100 (get l);
true
;;

t @@ fun () ->
let l = create 0 in
incr l;
get l = 1
;;

t @@ fun () ->
let l = create 0 in
decr l;
get l = ~-1
;;

t @@ fun () ->
let l = create 0 in
1 = incr_then_get l && 1 = get l
;;

t @@ fun () ->
let l = create 0 in
0 = get_then_incr l && 1 = get l
;;

t @@ fun () ->
let l = create 10 in
9 = decr_then_get l && 9 = get l
;;

t @@ fun () ->
let l = create 10 in
10 = get_then_decr l && 9 = get l