t/helper/test-parse-options.c

   1 #include "test-tool.h"
   2 #include "cache.h"
   3 #include "parse-options.h"
   4 #include "string-list.h"
   5
   6 static int boolean = 0;
   7 static int integer = 0;
   8 static unsigned long magnitude = 0;
   9 static timestamp_t timestamp;
  10 static int abbrev = 7;
  11 static int verbose = -1; /* unspecified */
  12 static int dry_run = 0, quiet = 0;
  13 static char *string = NULL;
  14 static char *file = NULL;
  15 static int ambiguous;
  16 static struct string_list list = STRING_LIST_INIT_NODUP;
  17
  18 static struct {
  19         int called;
  20         const char *arg;
  21         int unset;
  22 } length_cb;
  23
  24 static int length_callback(const struct option *opt, const char *arg, int unset)
  25 {
  26         length_cb.called = 1;
  27         length_cb.arg = arg;
  28         length_cb.unset = unset;
  29
  30         if (unset)
  31                 return 1; /* do not support unset */
  32
  33         *(int *)opt->value = strlen(arg);
  34         return 0;
  35 }
  36
  37 static int number_callback(const struct option *opt, const char *arg, int unset)
  38 {
  39         *(int *)opt->value = strtol(arg, NULL, 10);
  40         return 0;
  41 }
  42
  43 static int collect_expect(const struct option *opt, const char *arg, int unset)
  44 {
  45         struct string_list *expect;
  46         struct string_list_item *item;
  47         struct strbuf label = STRBUF_INIT;
  48         const char *colon;
  49
  50         if (!arg || unset)
  51                 die("malformed --expect option");
  52
  53         expect = (struct string_list *)opt->value;
  54         colon = strchr(arg, ':');
  55         if (!colon)
  56                 die("malformed --expect option, lacking a colon");
  57         strbuf_add(&label, arg, colon - arg);
  58         item = string_list_insert(expect, strbuf_detach(&label, NULL));
  59         if (item->util)
  60                 die("malformed --expect option, duplicate %s", label.buf);
  61         item->util = (void *)arg;
  62         return 0;
  63 }
  64
  65 __attribute__((format (printf,3,4)))
  66 static void show(struct string_list *expect, int *status, const char *fmt, ...)
  67 {
  68         struct string_list_item *item;
  69         struct strbuf buf = STRBUF_INIT;
  70         va_list args;
  71
  72         va_start(args, fmt);
  73         strbuf_vaddf(&buf, fmt, args);
  74         va_end(args);
  75
  76         if (!expect->nr)
  77                 printf("%s\n", buf.buf);
  78         else {
  79                 char *colon = strchr(buf.buf, ':');
  80                 if (!colon)
  81                         die("malformed output format, output lacking colon: %s", fmt);
  82                 *colon = '\0';
  83                 item = string_list_lookup(expect, buf.buf);
  84                 *colon = ':';
  85                 if (!item)
  86                         ; /* not among entries being checked */
  87                 else {
  88                         if (strcmp((const char *)item->util, buf.buf)) {
  89                                 printf("-%s\n", (char *)item->util);
  90                                 printf("+%s\n", buf.buf);
  91                                 *status = 1;
  92                         }
  93                 }
  94         }
  95         strbuf_release(&buf);
  96 }
  97
  98 int cmd__parse_options(int argc, const char **argv)
  99 {
 100         const char *prefix = "prefix/";
 101         const char *usage[] = {
 102                 "test-tool parse-options <options>",
 103                 "",
 104                 "A helper function for the parse-options API.",
 105                 NULL
 106         };
 107         struct string_list expect = STRING_LIST_INIT_NODUP;
 108         struct option options[] = {
 109                 OPT_BOOL(0, "yes", &boolean, "get a boolean"),
 110                 OPT_BOOL('D', "no-doubt", &boolean, "begins with 'no-'"),
 111                 { OPTION_SET_INT, 'B', "no-fear", &boolean, NULL,
 112                   "be brave", PARSE_OPT_NOARG | PARSE_OPT_NONEG, NULL, 1 },
 113                 OPT_COUNTUP('b', "boolean", &boolean, "increment by one"),
 114                 OPT_BIT('4', "or4", &boolean,
 115                         "bitwise-or boolean with ...0100", 4),
 116                 OPT_NEGBIT(0, "neg-or4", &boolean, "same as --no-or4", 4),
 117                 OPT_GROUP(""),
 118                 OPT_INTEGER('i', "integer", &integer, "get a integer"),
 119                 OPT_INTEGER('j', NULL, &integer, "get a integer, too"),
 120                 OPT_MAGNITUDE('m', "magnitude", &magnitude, "get a magnitude"),
 121                 OPT_SET_INT(0, "set23", &integer, "set integer to 23", 23),
 122                 OPT_CALLBACK('L', "length", &integer, "str",
 123                         "get length of <str>", length_callback),
 124                 OPT_FILENAME('F', "file", &file, "set file to <file>"),
 125                 OPT_GROUP("String options"),
 126                 OPT_STRING('s', "string", &string, "string", "get a string"),
 127                 OPT_STRING(0, "string2", &string, "str", "get another string"),
 128                 OPT_STRING(0, "st", &string, "st", "get another string (pervert ordering)"),
 129                 OPT_STRING('o', NULL, &string, "str", "get another string"),
 130                 OPT_NOOP_NOARG(0, "obsolete"),
 131                 OPT_STRING_LIST(0, "list", &list, "str", "add str to list"),
 132                 OPT_GROUP("Magic arguments"),
 133                 OPT_ARGUMENT("quux", "means --quux"),
 134                 OPT_NUMBER_CALLBACK(&integer, "set integer to NUM",
 135                         number_callback),
 136                 { OPTION_COUNTUP, '+', NULL, &boolean, NULL, "same as -b",
 137                   PARSE_OPT_NOARG | PARSE_OPT_NONEG | PARSE_OPT_NODASH },
 138                 { OPTION_COUNTUP, 0, "ambiguous", &ambiguous, NULL,
 139                   "positive ambiguity", PARSE_OPT_NOARG | PARSE_OPT_NONEG },
 140                 { OPTION_COUNTUP, 0, "no-ambiguous", &ambiguous, NULL,
 141                   "negative ambiguity", PARSE_OPT_NOARG | PARSE_OPT_NONEG },
 142                 OPT_GROUP("Standard options"),
 143                 OPT__ABBREV(&abbrev),
 144                 OPT__VERBOSE(&verbose, "be verbose"),
 145                 OPT__DRY_RUN(&dry_run, "dry run"),
 146                 OPT__QUIET(&quiet, "be quiet"),
 147                 OPT_CALLBACK(0, "expect", &expect, "string",
 148                              "expected output in the variable dump",
 149                              collect_expect),
 150                 OPT_END(),
 151         };
 152         int i;
 153         int ret = 0;
 154
 155         argc = parse_options(argc, (const char **)argv, prefix, options, usage, 0);
 156
 157         if (length_cb.called) {
 158                 const char *arg = length_cb.arg;
 159                 int unset = length_cb.unset;
 160                 show(&expect, &ret, "Callback: \"%s\", %d",
 161                      (arg ? arg : "not set"), unset);
 162         }
 163         show(&expect, &ret, "boolean: %d", boolean);
 164         show(&expect, &ret, "integer: %d", integer);
 165         show(&expect, &ret, "magnitude: %lu", magnitude);
 166         show(&expect, &ret, "timestamp: %"PRItime, timestamp);
 167         show(&expect, &ret, "string: %s", string ? string : "(not set)");
 168         show(&expect, &ret, "abbrev: %d", abbrev);
 169         show(&expect, &ret, "verbose: %d", verbose);
 170         show(&expect, &ret, "quiet: %d", quiet);
 171         show(&expect, &ret, "dry run: %s", dry_run ? "yes" : "no");
 172         show(&expect, &ret, "file: %s", file ? file : "(not set)");
 173
 174         for (i = 0; i < list.nr; i++)
 175                 show(&expect, &ret, "list: %s", list.items[i].string);
 176
 177         for (i = 0; i < argc; i++)
 178                 show(&expect, &ret, "arg %02d: %s", i, argv[i]);
 179
 180         return ret;
 181 }