1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
open Base
open Ocamlformat_lib
let eval_fmt term =
let buffer = Buffer.create 0 in
let ppf = Format_.formatter_of_buffer buffer in
Fmt.eval ppf term ;
Format_.pp_print_flush ppf () ;
Buffer.contents buffer
let tests_lazy =
[ ( "lazy_: not using lazy"
, `Quick
, fun () ->
let r = ref None in
let pp s =
r := Some s ;
Fmt.str s
in
let term = Fmt.fmt_if false (pp "hello") in
let expected = "" in
let expected_r = Some "hello" in
let got = eval_fmt term in
let got_r = !r in
Alcotest.check Alcotest.(string) Stdlib.__LOC__ expected got ;
Alcotest.check
Alcotest.(option string)
Stdlib.__LOC__ expected_r got_r )
; ( "lazy_: using lazy"
, `Quick
, fun () ->
let r = ref None in
let pp s =
Fmt.lazy_ (fun () ->
r := Some s ;
Fmt.str s )
in
let term = Fmt.fmt_if false (pp "hello") in
let expected = "" in
let expected_r = None in
let got = eval_fmt term in
let got_r = !r in
Alcotest.check Alcotest.(string) Stdlib.__LOC__ expected got ;
Alcotest.check
Alcotest.(option string)
Stdlib.__LOC__ expected_r got_r ) ]
let tests_list_pn =
let test name ~expected ~expected_calls f =
( "list_pn: " ^ name
, `Quick
, fun () ->
let calls = ref [] in
let record_call (prev, x, next) =
let opt_to_string so = Option.value so ~default:"-" in
let call_str = opt_to_string prev ^ x ^ opt_to_string next in
calls := call_str :: !calls
in
let pp_spy ~prev x ~next =
record_call (prev, x, next) ;
Fmt.str x
in
let term = f pp_spy in
let got = eval_fmt term in
Alcotest.check Alcotest.(string) Stdlib.__LOC__ expected got ;
let got_calls = List.rev !calls in
Alcotest.check
Alcotest.(list string)
Stdlib.__LOC__ expected_calls got_calls )
in
[ test "evaluation order" ~expected:"abcde"
~expected_calls:["-ab"; "abc"; "bcd"; "cde"; "de-"] (fun pp_spy ->
let l = ["a"; "b"; "c"; "d"; "e"] in
Fmt.list_pn l pp_spy )
; test "does not call pp if not formatting" ~expected:"" ~expected_calls:[]
(fun pp_spy ->
let l = ["a"; "b"; "c"; "d"; "e"] in
Fmt.fmt_if false (Fmt.list_pn l pp_spy) ) ]
let tests_list_k =
let test name ~expected ~expected_calls f =
( "list_k: " ^ name
, `Quick
, fun () ->
let calls = ref [] in
let record_call x = calls := x :: !calls in
let pp_spy x = record_call x ; Fmt.str x in
let term = f pp_spy in
let got = eval_fmt term in
Alcotest.check Alcotest.(string) Stdlib.__LOC__ expected got ;
let got_calls = List.rev !calls in
Alcotest.check
Alcotest.(list string)
Stdlib.__LOC__ expected_calls got_calls )
in
[ test "evaluation order" ~expected:"a b c d e"
~expected_calls:["a"; "b"; "c"; "d"; "e"] (fun pp_spy ->
let l = ["a"; "b"; "c"; "d"; "e"] in
Fmt.list l (Fmt.str " ") pp_spy )
; test "does not call pp if not formatting" ~expected:"" ~expected_calls:[]
(fun pp_spy ->
let l = ["a"; "b"; "c"; "d"; "e"] in
Fmt.fmt_if false (Fmt.list l (Fmt.str " ") pp_spy) ) ]
let tests_sequence =
let test name term ~expected =
( "sequence: " ^ name
, `Quick
, fun () ->
let got = eval_fmt term in
Alcotest.check Alcotest.string Stdlib.__LOC__ expected got )
in
[ test "1 element" (Fmt.sequence [Fmt.char 'c']) ~expected:"c"
; test "empty list" (Fmt.sequence []) ~expected:""
; test "list"
(Fmt.sequence [Fmt.str "a"; Fmt.str "b"; Fmt.str "c"; Fmt.str "d"])
~expected:"abcd"
; test "long list"
(Fmt.sequence (List.init 300_000 ~f:(fun _ -> Fmt.noop)))
~expected:"" ]
let tests = tests_lazy @ tests_list_pn @ tests_list_k @ tests_sequence
|
