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
|
let to_string_tests =
let test ?suf expected =
Alcotest.(check string)
__LOC__ expected
(Yojson.Safe.to_string ?suf Fixtures.json_value)
in
[
( "to_string with default settings",
`Quick,
fun () -> test Fixtures.json_string );
( "to_string with newline",
`Quick,
fun () -> test ~suf:"\n" Fixtures.json_string_newline );
( "to_string without newline",
`Quick,
fun () -> test ~suf:"" Fixtures.json_string );
]
let to_file_tests =
let test ?suf expected =
let output_file = Filename.temp_file "test_yojson_to_file" ".json" in
Yojson.Safe.to_file ?suf output_file Fixtures.json_value;
let file_content =
let ic = open_in output_file in
let length = in_channel_length ic in
let s = really_input_string ic length in
close_in ic;
s
in
Sys.remove output_file;
Alcotest.(check string) __LOC__ expected file_content
in
[
( "to_file with default settings",
`Quick,
fun () -> test Fixtures.json_string_newline );
( "to_file with newline",
`Quick,
fun () -> test ~suf:"\n" Fixtures.json_string_newline );
( "to_file without newline",
`Quick,
fun () -> test ~suf:"" Fixtures.json_string );
]
(* List.to_seq is not available on old OCaml versions. *)
let rec list_to_seq = function
| [] -> fun () -> Seq.Nil
| x :: xs -> fun () -> Seq.Cons (x, list_to_seq xs)
let seq_to_file_tests =
let test ?suf () =
let output_file = Filename.temp_file "test_yojson_seq_to_file" ".json" in
let data = [ `String "foo"; `String "bar" ] in
Yojson.Safe.seq_to_file ?suf output_file (list_to_seq data);
let read_data =
let seq = Yojson.Safe.seq_from_file output_file in
let acc = ref [] in
Seq.iter (fun v -> acc := v :: !acc) seq;
List.rev !acc
in
Sys.remove output_file;
Alcotest.(check (list Testable.yojson))
"seq_{to,from}_file roundtrip" data read_data
in
[
("seq_to_file with default settings", `Quick, fun () -> test ());
("seq_to_file with newline", `Quick, fun () -> test ~suf:"\n" ());
("seq_to_file without newline", `Quick, fun () -> test ~suf:"" ());
]
let single_json =
List.flatten [ to_file_tests; to_string_tests; seq_to_file_tests ]
|
