diff options
Diffstat (limited to 'lib/tests/test_opts.c')
| -rw-r--r-- | lib/tests/test_opts.c | 208 |
1 files changed, 208 insertions, 0 deletions
diff --git a/lib/tests/test_opts.c b/lib/tests/test_opts.c new file mode 100644 index 00000000..12726059 --- /dev/null +++ b/lib/tests/test_opts.c @@ -0,0 +1,208 @@ +/***************************************************************************** +* +* This program is free software: you can redistribute it and/or modify +* it under the terms of the GNU General Public License as published by +* the Free Software Foundation, either version 3 of the License, or +* (at your option) any later version. +* +* This program is distributed in the hope that it will be useful, +* but WITHOUT ANY WARRANTY; without even the implied warranty of +* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +* GNU General Public License for more details. +* +* You should have received a copy of the GNU General Public License +* along with this program. If not, see <http://www.gnu.org/licenses/>. +* +* $Id: test_ini.c 1951 2008-03-16 18:10:47Z dermoth $ +* +*****************************************************************************/ + +#include "common.h" +#include "extra_opts.h" +#include "utils_base.h" + +#include "tap.h" + +void my_free(int *argc, char **argv) { + int i; + printf (" Arg(%i): ", *argc); + for (i=1; i<*argc; i++) printf ("'%s' ", argv[i]); + printf ("\n"); + free(argv); + *argc=0; +} + +int array_diff(int i1, char **a1, int i2, char **a2) { + int i; + + if (i1 != i2) { + printf(" Argument count doesn't match!\n"); + return 0; + } + for (i=0; i<=i1; i++) { + if (a1[i]==NULL && a2[i]==NULL) continue; + if (a1[i]==NULL || a2[i]==NULL) { + printf(" Argument # %i null in one array!\n", i); + return 0; + } + if (strcmp(a1[i], a2[i])) { + printf(" Argument # %i doesn't match!\n", i); + return 0; + } + } + return 1; +} + +int +main (int argc, char **argv) +{ + char **argv_test=NULL, **argv_known=NULL; + int i, argc_test, argc_new; + + plan_tests(8); + + argv_test=(char **)malloc(2*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = NULL; + argc_test=1; + argv_known=(char **)realloc(argv_known, 2*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_disk", &argc_new); + ok(array_diff(argc_new, argv_test, 1, argv_known), "No opts, returns correct argv/argc"); + my_free(&argc_new, argv_test); + + argv_test=(char **)malloc(6*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = "arg1"; + argv_test[2] = "--arg2=val1"; + argv_test[3] = "--arg3"; + argv_test[4] = "val2"; + argv_test[5] = NULL; + argc_test=5; + argv_known=(char **)realloc(argv_known, 6*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = "arg1"; + argv_known[2] = "--arg2=val1"; + argv_known[3] = "--arg3"; + argv_known[4] = "val2"; + argv_known[5] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_disk", &argc_new); + ok(array_diff(argc_new, argv_test, 5, argv_known), "No extra opts, verbatim copy of argv"); + my_free(&argc_new,argv_test); + + argv_test=(char **)malloc(3*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = "--extra-opts=@./config-opts.ini"; + argv_test[2] = NULL; + argc_test=2; + argv_known=(char **)realloc(argv_known, 5*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = "--foo=Bar"; + argv_known[2] = "--this=Your Mother!"; + argv_known[3] = "--blank"; + argv_known[4] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_disk", &argc_new); + ok(array_diff(argc_new, argv_test, 4, argv_known), "Only extra opts using default section"); + my_free(&argc_new,argv_test); + + argv_test=(char **)malloc(5*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = "--extra-opts=sect1@./config-opts.ini"; + argv_test[2] = "--extra-opts"; + argv_test[3] = "sect2@./config-opts.ini"; + argv_test[4] = NULL; + argc_test=4; + argv_known=(char **)realloc(argv_known, 5*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = "--one=two"; + argv_known[2] = "--something else=oops"; + argv_known[3] = "--this=that"; + argv_known[4] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_disk", &argc_new); + ok(array_diff(argc_new, argv_test, 4, argv_known), "Only extra opts specified twice"); + my_free(&argc_new,argv_test); + + argv_test=(char **)malloc(7*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = "--arg1=val1"; + argv_test[2] = "--extra-opts=@./config-opts.ini"; + argv_test[3] = "--extra-opts"; + argv_test[4] = "sect1@./config-opts.ini"; + argv_test[5] = "--arg2"; + argv_test[6] = NULL; + argc_test=6; + argv_known=(char **)realloc(argv_known, 8*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = "--foo=Bar"; + argv_known[2] = "--this=Your Mother!"; + argv_known[3] = "--blank"; + argv_known[4] = "--one=two"; + argv_known[5] = "--arg1=val1"; + argv_known[6] = "--arg2"; + argv_known[7] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_disk", &argc_new); + ok(array_diff(argc_new, argv_test, 7, argv_known), "twice extra opts using two sections"); + my_free(&argc_new,argv_test); + + /* Next three checks should die according to N::P - for now they're useful + * to test code is working properly (i.e. no srash or unexpected behavior) + */ + argv_test=(char **)malloc(6*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = "arg1"; + argv_test[2] = "--extra-opts=missing@./config-opts.ini"; + argv_test[3] = "--arg3"; + argv_test[4] = "val2"; + argv_test[5] = NULL; + argc_test=5; + argv_known=(char **)realloc(argv_known, 5*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = "arg1"; + argv_known[2] = "--arg3"; + argv_known[3] = "val2"; + argv_known[4] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_missing", &argc_new); + ok(array_diff(argc_new, argv_test, 4, argv_known), "Missing section 1"); + my_free(&argc_new,argv_test); + + argv_test=(char **)malloc(7*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = "arg1"; + argv_test[2] = "--extra-opts"; + argv_test[3] = "missing@./config-opts.ini"; + argv_test[4] = "--arg3"; + argv_test[5] = "val2"; + argv_test[6] = NULL; + argc_test=6; + argv_known=(char **)realloc(argv_known, 5*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = "arg1"; + argv_known[2] = "--arg3"; + argv_known[3] = "val2"; + argv_known[4] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_missing", &argc_new); + ok(array_diff(argc_new, argv_test, 4, argv_known), "Missing section 2"); + my_free(&argc_new,argv_test); + + argv_test=(char **)malloc(6*sizeof(char **)); + argv_test[0] = "prog_name"; + argv_test[1] = "arg1"; + argv_test[2] = "--extra-opts"; + argv_test[3] = "--arg3"; + argv_test[4] = "val2"; + argv_test[5] = NULL; + argc_test=5; + argv_known=(char **)realloc(argv_known, 5*sizeof(char **)); + argv_known[0] = "prog_name"; + argv_known[1] = "arg1"; + argv_known[2] = "--arg3"; + argv_known[3] = "val2"; + argv_known[4] = NULL; + argv_test=np_extra_opts(argc_test, argv_test, "check_missing", &argc_new); + ok(array_diff(argc_new, argv_test, 4, argv_known), "Missing section 3"); + my_free(&argc_new,argv_test); + + return exit_status(); +} + |