open CCFormat

module T = (val Containers_testlib.make ~__FILE__ ())
include T;;

let to_string_test s = CCFormat.sprintf_no_color "@[<h>%a@]%!" s ();;

eq ~printer:(fun s->CCFormat.sprintf "%S" s) "a b" (to_string_test (return "a@ b"));;
eq ~printer:(fun s->CCFormat.sprintf "%S" s) ", " (to_string_test (return ",@ "));;
eq ~printer:(fun s->CCFormat.sprintf "%S" s) "and then" (to_string_test (return "@{<Red>and then@}@,"));;
eq ~printer:(fun s->CCFormat.sprintf "%S" s) "a b" (to_string_test (return "@[<h>a@ b@]"));;


eq ~printer:(fun s->CCFormat.sprintf "%S" s) "a\nb\nc"
  (sprintf_no_color "@[<v>%a@]%!" text "a b c");;
eq ~printer:(fun s->CCFormat.sprintf "%S" s) "a b\nc"
  (sprintf_no_color "@[<h>%a@]%!" text "a b\nc");;


eq ~printer:(fun s->CCFormat.sprintf "%S" s)
  "(a\n b\n c)" (sprintf_no_color "(@[<v>%a@])" string_lines "a\nb\nc");;


eq ~printer:(fun s -> CCFormat.sprintf "%S" s) "foobar"
  (to_string_test (append (return "foo") (return "bar")));;
eq ~printer:(fun s -> CCFormat.sprintf "%S" s) "bar"
  (to_string_test (append (return "") (return "bar")));;
eq ~printer:(fun s -> CCFormat.sprintf "%S" s) "foo"
  (to_string_test (append (return "foo") (return "")));;


eq ~printer:(fun s -> CCFormat.sprintf "%S" s) "" (to_string_test @@ append_l []);;
eq ~printer:(fun s -> CCFormat.sprintf "%S" s) "foobarbaz" (to_string_test @@ append_l (List.map return ["foo"; "bar"; "baz"]));;
eq ~printer:(fun s -> CCFormat.sprintf "%S" s) "3141" (to_string_test @@ append_l (List.map (const int) [3; 14; 1]));;

t @@ fun () ->
  let buf1 = Buffer.create 42 in
  let buf2 = Buffer.create 42 in
  let f1 = Format.formatter_of_buffer buf1 in
  let f2 = Format.formatter_of_buffer buf2 in
  let fmt = tee f1 f2 in
  Format.fprintf fmt "coucou@.";
  assert_equal ~printer:CCFun.id "coucou\n" (Buffer.contents buf1);
  assert_equal ~printer:CCFun.id "coucou\n" (Buffer.contents buf2);
  true;;

[@@@ifge 4.8]

t @@ fun () ->
  set_color_default true;
  let s = sprintf
    "what is your %a? %a! No, %a! Ahhhhhhh@."
    (styling [`FG `White; `Bold] string) "favorite color"
    (styling [`FG `Blue] string) "blue"
    (styling [`FG `Red] string) "red"
  in
  assert_equal ~printer:CCFun.id
    "what is your \027[37;1mfavorite color\027[0m? \027[34mblue\027[0m! No, \027[31mred\027[0m! Ahhhhhhh\n"
    s;
    true
;;

[@@@endif]

t @@ fun () ->
  set_color_default true;
  let s = sprintf
    "what is your @{<White>favorite color@}? @{<blue>blue@}! No, @{<red>red@}! Ahhhhhhh@."
  in
  assert_equal ~printer:CCFun.id
    "what is your \027[37;1mfavorite color\027[0m? \027[34mblue\027[0m! No, \027[31mred\027[0m! Ahhhhhhh\n"
    s;
    true;;

t @@ fun () -> sprintf "yolo %s %d" "a b" 42 = "yolo a b 42";;
t @@ fun () -> sprintf "%d " 0 = "0 ";;
t @@ fun () -> sprintf_no_color "%d " 0 = "0 ";;

t @@ fun () ->
  set_color_default true;
  assert_equal "\027[31myolo\027[0m" (sprintf "@{<red>yolo@}");
  assert_equal "yolo" (sprintf_no_color "@{<red>yolo@}");
  true;;

eq ~printer:CCFormat.(to_string (opt string))
  (Some "hello world")
    (ksprintf ~f:(fun s -> Some s) "hello %a" CCFormat.string "world");;


eq ~printer:(fun s->s) "[1;2;3]" (to_string Dump.(list int) [1;2;3]);;
eq ~printer:(fun s->s) "Some 1" (to_string Dump.(option int) (Some 1));;
eq ~printer:(fun s->s) "[None;Some \"a b\"]" (to_string Dump.(list (option string)) [None; Some "a b"]);;
eq ~printer:(fun s->s) "[(Ok \"a b c\");(Error \"nope\")]"
    (to_string Dump.(list (result string)) [Ok "a b c"; Error "nope"]);;

eq  ANSI_codes.reset "\x1b[0m";;