opt: fix opt_unregister.
[ccan] / ccan / opt / test / run-unregister.c
1 #include <ccan/tap/tap.h>
2 #include <stdlib.h>
3 #include <ccan/opt/opt.c>
4 #include <ccan/opt/usage.c>
5 #include <ccan/opt/helpers.c>
6 #include <ccan/opt/parse.c>
7 #include "utils.h"
8
9 int main(int argc, char *argv[])
10 {
11         const char *myname = argv[0];
12
13         plan_tests(15);
14
15         opt_register_noarg("--aaa|-a", test_noarg, NULL, "AAAAAAll");
16         opt_register_arg("-b", test_arg, NULL, "bbb", "b");
17
18         /* We can't unregister wrong ones, but can unregister correct one */
19         ok1(!opt_unregister("--aaa"));
20         ok1(!opt_unregister("-a"));
21         ok1(opt_unregister("--aaa|-a"));
22
23         /* Arg parsing works as if we'd never registered it */
24         ok1(parse_args(&argc, &argv, "-bbbb", NULL));
25         ok1(argc == 1);
26         ok1(argv[0] == myname);
27         ok1(argv[1] == NULL);
28         ok1(test_cb_called == 1);
29
30         ok1(!parse_args(&argc, &argv, "--aaa", NULL));
31
32         /* We can still add another one OK. */
33         opt_register_noarg("-c", test_noarg, NULL, "AAAAAAll");
34         ok1(parse_args(&argc, &argv, "-c", NULL));
35         ok1(argc == 1);
36         ok1(argv[0] == myname);
37         ok1(argv[1] == NULL);
38         ok1(test_cb_called == 2);
39
40         /* parse_args allocates argv */
41         free(argv);
42         return exit_status();
43 }