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