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

module P = Make(struct let max_size = 30 end)
module P2 = Make(struct let max_size = 15 end)
module Fut = P.Fut
module Fut2 = P2.Fut;;

t @@ fun () ->
  List.iter
    (fun n ->
      let l = Iter.(1 -- n) |> Iter.to_list in
      let l = List.rev_map (fun _i ->
        Fut.make
          (fun () ->
            Thread.delay 0.01;
            1
        )) l in
      let l' = List.map Fut.get l in
      assert_equal n (List.fold_left (+) 0 l');
    )
    [ 10; 300; ];
  true;;

t @@ fun () ->
  List.iter
    (fun n ->
      let l = Iter.(1 -- n) |> Iter.to_list in
      let l = List.rev_map (fun _i ->
        Fut2.make
          (fun () ->
            Thread.delay 0.01;
            1
        )) l in
      let l' = List.map Fut2.get l in
      assert_equal n (List.fold_left (+) 0 l');
    )
    [ 10; 300; ];
  true;;

t @@ fun () ->
  let a = Fut.make (fun () -> 1) in
  let b = Fut.return 42 in
  let c = Fut.monoid_product CCPair.make a b in
  assert_equal (1,42) (Fut.get c);
  true;;

t @@ fun () ->
  let a = Fut.make (fun () -> 1) in
  let b = Fut.make (fun () -> 42) in
  let c = Fut.monoid_product CCPair.make a b in
  assert_equal (1,42) (Fut.get c);
  true;;

t @@ fun () ->
  let a = Fut.make (fun () -> 1) in
  let b = Fut.map succ @@ Fut.make (fun () -> 41) in
  let c = Fut.monoid_product CCPair.make a b in
  assert_equal (1,42) (Fut.get c);
  true;;

eq [2;3] (Fut.get @@ Fut.map_l (fun x -> Fut.return (x+1)) [1;2]);;
eq [] (Fut.get @@ Fut.map_l (fun x -> Fut.return (x+1)) []);;

t @@ fun () ->
  let l = CCList.(1 -- 50) in
  let l' = l
    |> List.map
      (fun x -> Fut.make (fun () -> Thread.delay 0.1; x*10))
    |> Fut.sequence_l
    |> Fut.map (List.fold_left (+) 0)
  in
  let expected = List.fold_left (fun acc x -> acc + 10 * x) 0 l in
  assert_equal expected (Fut.get l');
  true;;

t @@ fun () ->
  let l = CCList.(1 -- 100_000) in
  let l' = l
    |> CCList.map
      (fun _x -> Fut.make (fun () -> 1))
    |> Fut.sequence_l
    |> Fut.map (List.fold_left (+) 0)
  in
  let expected = 100_000 in
  assert_equal expected (Fut.get l');
  true;;

t @@ fun () ->
  let l = CCList.(1 -- 50) in
  let l' = l
    |> List.map
      (fun x -> Fut.make (fun () -> Thread.delay 0.1; if x = 5 then raise Exit; x))
    |> Fut.sequence_l
    |> Fut.map (List.fold_left (+) 0)
  in
  assert_raises ((=)Exit) (fun () -> Fut.get l');
  true;;

t @@ fun () ->
  let rec fib x = if x<2 then 1 else fib (x-1)+fib(x-2) in
  let l =
    CCList.(1--10_000)
    |> List.rev_map
      (fun x-> Fut.make (fun () -> Thread.yield(); fib (x mod 20)))
    |> Fut.(map_l (fun x->x>|= fun x->x+1))
  in
  assert (Fut.state l = Waiting);
  let l' = Fut.get l in
  assert_equal 10_000 (List.length l');
  true;;

t @@ fun () ->
  let l = CCList.(1 -- 50) in
  let l' = l
    |> List.map
      (fun x -> Fut2.make (fun () -> Thread.delay 0.1; x*10))
    |> Fut2.sequence_l
    |> Fut2.map (List.fold_left (+) 0)
  in
  let expected = List.fold_left (fun acc x -> acc + 10 * x) 0 l in
  assert_equal expected (Fut2.get l');
  true;;

t @@ fun () ->
  let l = CCList.(1 -- 50) in
  let l' = l
    |> List.map
      (fun x -> Fut2.make (fun () -> Thread.delay 0.1; if x = 5 then raise Exit; x))
    |> Fut2.sequence_l
    |> Fut2.map (List.fold_left (+) 0)
  in
  assert_raises ((=)Exit) (fun () -> Fut2.get l');
  true;;

t @@ fun () ->
  let rec fib x = if x<2 then 1 else fib (x-1)+fib(x-2) in
  let l =
    CCList.(1--10_000)
    |> List.rev_map
      (fun x-> Fut2.make (fun () -> Thread.yield(); fib (x mod 20)))
    |> Fut2.(map_l (fun x->x>|= fun x->x+1))
  in
  assert (Fut2.state l = Waiting);
  let l' = Fut2.get l in
  assert_equal 10_000 (List.length l');
  true;;

t @@ fun () ->
  let start = Unix.gettimeofday () in
  let pause = 0.2 and n = 10 in
  let l = CCList.(1 -- n)
    |> List.map (fun _ -> Fut.make (fun () -> Thread.delay pause))
  in
  List.iter Fut.get l;
  let stop = Unix.gettimeofday () in
  assert (stop -. start < float_of_int n *. pause);
  true;;

t @@ fun () ->
  let start = Unix.gettimeofday () in
  let pause = 0.2 and n = 10 in
  let l = CCList.(1 -- n)
    |> List.map (fun _ -> Fut2.make (fun () -> Thread.delay pause))
  in
  List.iter Fut2.get l;
  let stop = Unix.gettimeofday () in
  assert (stop -. start < float_of_int n *. pause);
  true;;