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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
|
open OUnit2
open Posix_getopt
let sysname = Posix_uname.((uname ()).sysname)
module Process = struct
let read_stdout cmd args =
let cmd = (cmd, Array.append [| cmd |] args) in
Lwt_process.pread_lines cmd |> Lwt_stream.to_list |> Lwt_main.run
end
let is_alpine =
try
Process.read_stdout "/bin/sh"
[| "-c"; "cat /etc/os-release | grep alpine" |]
<> []
with _ -> false
let is_s390x_debian =
try
List.hd (Process.read_stdout "dpkg" [| "--print-architecture" |]) = "s390x"
with _ -> false
let test_short_flag _ =
reset ();
let argv = [| "progname"; "-b" |] in
let test = ref false in
let opt = { name = 'b'; arg = `None (fun () -> test := true) } in
let ret = getopt argv [opt] in
assert !test;
assert (Array.length ret = 0)
let test_short_required _ =
reset ();
let argv = [| "progname"; "-b"; "arg" |] in
let test = ref "" in
let opt = { name = 'b'; arg = `Required (fun arg -> test := arg) } in
let ret = getopt argv [opt] in
assert_equal "arg" !test;
assert (Array.length ret = 0)
let test_long_required_long _ =
skip_if (not has_getopt_long) "Doesn't support getopt_long";
reset ();
let argv = [| "progname"; "--bla"; "arg" |] in
let test = ref "" in
let opt = { name = ("bla", 'b'); arg = `Required (fun arg -> test := arg) } in
let ret = getopt_long argv [opt] in
assert_equal "arg" !test;
assert (Array.length ret = 0)
let test_long_required_short _ =
skip_if (not has_getopt_long) "Doesn't support getopt_long";
reset ();
let argv = [| "progname"; "-b"; "arg" |] in
let test = ref "" in
let opt = { name = ("bla", 'b'); arg = `Required (fun arg -> test := arg) } in
let ret = getopt_long argv [opt] in
assert_equal "arg" !test;
assert (Array.length ret = 0)
let test_short_required_no_arg _ =
skip_if is_s390x_debian "Not supported";
reset ();
let argv = [| "progname"; "-b" |] in
let opt = { name = 'b'; arg = `Required (fun _ -> ()) } in
let test = ref false in
begin try ignore (getopt argv [opt])
with Missing_argument 'b' -> test := true
end;
assert !test
let test_long_required_no_long_arg _ =
skip_if (not has_getopt_long) "Doesn't support getopt_long";
skip_if (sysname = "Darwin") "Doesn't work on OSX";
skip_if is_alpine "Doesn't work in Alpine";
skip_if is_s390x_debian "Not supported";
reset ();
let argv = [| "progname"; "--bla" |] in
let opt = { name = ("bla", 'b'); arg = `Required (fun _ -> ()) } in
let test = ref false in
begin try ignore (getopt_long argv [opt])
with Missing_argument 'b' -> test := true
end;
assert !test
let test_long_required_no_short_arg _ =
skip_if (not has_getopt_long) "Doesn't support getopt_long";
skip_if is_s390x_debian "Not supported";
reset ();
let argv = [| "progname"; "-b" |] in
let opt = { name = ("bla", 'b'); arg = `Required (fun _ -> ()) } in
let test = ref false in
begin try ignore (getopt_long argv [opt])
with Missing_argument 'b' -> test := true
end;
assert !test
let test_short_optional_no_arg _ =
skip_if is_s390x_debian "Not supported";
reset ();
let argv = [| "progname"; "-b" |] in
let test = ref false in
let opt = { name = 'b'; arg = `Optional (fun arg -> test := arg = None) } in
let ret = getopt argv [opt] in
assert !test;
assert (Array.length ret = 0)
let test_short_optional_arg _ =
reset ();
let argv = [| "progname"; "-b"; "yup" |] in
let test = ref false in
let opt =
{ name = 'b'; arg = `Optional (fun arg -> test := arg = Some "yup") }
in
let ret = getopt argv [opt] in
assert !test;
assert (Array.length ret = 0)
let test_remaining_arg _ =
reset ();
let argv = [| "progname"; "--"; "-b" |] in
let opt = { name = 'b'; arg = `None (fun () -> ()) } in
let ret = getopt argv [opt] in
assert_equal [| "-b" |] ret
let test_permuted_remaining_arg _ =
skip_if (sysname = "Darwin") "Doesn't work on OSX";
skip_if is_alpine "Doesn't work in Alpine";
reset ();
let argv = [| "progname"; "bla"; "-b" |] in
let opt = { name = 'b'; arg = `None (fun () -> ()) } in
let ret = getopt argv [opt] in
assert_equal [| "bla" |] ret
let test_complex_scenario _ =
skip_if (not has_getopt_long) "Doesn't support getopt_long";
reset ();
let argv =
[|
"progname";
"-a";
"-cret";
"-bopt";
"--gno";
"gna";
"--foo";
"--foo";
"--";
"bla";
|]
in
let a = ref false in
let b = ref "" in
let c = ref (Some "value") in
let gno = ref "" in
let foo = ref false in
let opts =
[
{ name = ("aa", 'a'); arg = `None (fun () -> a := true) };
{ name = ("bb", 'b'); arg = `Required (fun v -> b := v) };
{ name = ("cc", 'c'); arg = `Optional (fun v -> c := v) };
{ name = ("gno", 'g'); arg = `Required (fun v -> gno := v) };
{ name = ("foo", 'f'); arg = `None (fun () -> foo := true) };
]
in
let ret = getopt_long argv opts in
assert !a;
assert_equal "opt" !b;
assert_equal (Some "ret") !c;
assert_equal "gna" !gno;
assert_equal true !foo;
assert_equal [| "bla" |] ret
let test_short_unknown_before _ =
reset ();
let argv = [| "progname"; "-u"; "-b" |] in
let test = ref true in
let opt = { name = 'b'; arg = `None (fun () -> test := false) } in
try
ignore (getopt argv [opt]);
assert false
with Unknown_option "-u" -> assert !test
let test_short_unknown_after _ =
skip_if is_s390x_debian "Not supported";
reset ();
let argv = [| "progname"; "-b"; "--unknown" |] in
let test = ref false in
let opt = { name = 'b'; arg = `None (fun () -> test := true) } in
try
ignore (getopt argv [opt]);
assert false
with Unknown_option "--" -> assert !test
let test_long_unknown_before _ =
skip_if (not has_getopt_long) "Doesn't support getopt_long";
reset ();
let argv = [| "progname"; "--unknown"; "--bla" |] in
let test = ref true in
let opt = { name = ("bla", 'b'); arg = `None (fun () -> test := false) } in
try
ignore (getopt_long argv [opt]);
assert false
with Unknown_option "--unknown" -> assert !test
let test_long_unknown_after _ =
skip_if (not has_getopt_long) "Doesn't support getopt_long";
reset ();
let argv = [| "progname"; "--bla"; "--unknown" |] in
let test = ref false in
let opt = { name = ("bla", 'b'); arg = `None (fun () -> test := true) } in
try
ignore (getopt_long argv [opt]);
assert false
with Unknown_option "--unknown" -> assert !test
let suite =
"getopt tests"
>::: [
"test_short_flag" >:: test_short_flag;
"test_short_required" >:: test_short_required;
"test_long_required_long" >:: test_long_required_long;
"test_long_required_short" >:: test_long_required_short;
"test_short_required_no_arg" >:: test_short_required_no_arg;
"test_long_required_no_short_arg" >:: test_long_required_no_short_arg;
"test_long_required_no_long_arg" >:: test_long_required_no_long_arg;
"test_short_optional_no_arg" >:: test_short_optional_no_arg;
"test_short_optional_arg" >:: test_short_optional_arg;
"test_remaining_arg" >:: test_remaining_arg;
"test_permuted_remaining_arg" >:: test_permuted_remaining_arg;
"test_complex_scenario" >:: test_complex_scenario;
"test_short_unknown_before" >:: test_short_unknown_before;
"test_short_unknown_after" >:: test_short_unknown_after;
"test_long_unknown_before" >:: test_long_unknown_before;
"test_long_unknown_after" >:: test_long_unknown_after;
]
let () = run_test_tt_main suite
|
