Andrew's git - gitweb.git/blob - t/helper/test-run-command.c

   1/*
   2 * test-run-command.c: test run command API.
   3 *
   4 * (C) 2009 Ilari Liusvaara <ilari.liusvaara@elisanet.fi>
   5 *
   6 * This code is free software; you can redistribute it and/or modify
   7 * it under the terms of the GNU General Public License version 2 as
   8 * published by the Free Software Foundation.
   9 */
  10
  11#include "git-compat-util.h"
  12#include "run-command.h"
  13#include "argv-array.h"
  14#include "strbuf.h"
  15#include <string.h>
  16#include <errno.h>
  17
  18static int number_callbacks;
  19static int parallel_next(struct child_process *cp,
  20                         struct strbuf *err,
  21                         void *cb,
  22                         void **task_cb)
  23{
  24        struct child_process *d = cb;
  25        if (number_callbacks >= 4)
  26                return 0;
  27
  28        argv_array_pushv(&cp->args, d->argv);
  29        strbuf_addstr(err, "preloaded output of a child\n");
  30        number_callbacks++;
  31        return 1;
  32}
  33
  34static int no_job(struct child_process *cp,
  35                  struct strbuf *err,
  36                  void *cb,
  37                  void **task_cb)
  38{
  39        strbuf_addstr(err, "no further jobs available\n");
  40        return 0;
  41}
  42
  43static int task_finished(int result,
  44                         struct strbuf *err,
  45                         void *pp_cb,
  46                         void *pp_task_cb)
  47{
  48        strbuf_addstr(err, "asking for a quick stop\n");
  49        return 1;
  50}
  51
  52int cmd_main(int argc, const char **argv)
  53{
  54        struct child_process proc = CHILD_PROCESS_INIT;
  55        int jobs;
  56
  57        if (argc < 3)
  58                return 1;
  59        proc.argv = (const char **)argv + 2;
  60
  61        if (!strcmp(argv[1], "start-command-ENOENT")) {
  62                if (start_command(&proc) < 0 && errno == ENOENT)
  63                        return 0;
  64                fprintf(stderr, "FAIL %s\n", argv[1]);
  65                return 1;
  66        }
  67        if (!strcmp(argv[1], "run-command"))
  68                exit(run_command(&proc));
  69
  70        jobs = atoi(argv[2]);
  71        proc.argv = (const char **)argv + 3;
  72
  73        if (!strcmp(argv[1], "run-command-parallel"))
  74                exit(run_processes_parallel(jobs, parallel_next,
  75                                            NULL, NULL, &proc));
  76
  77        if (!strcmp(argv[1], "run-command-abort"))
  78                exit(run_processes_parallel(jobs, parallel_next,
  79                                            NULL, task_finished, &proc));
  80
  81        if (!strcmp(argv[1], "run-command-no-jobs"))
  82                exit(run_processes_parallel(jobs, no_job,
  83                                            NULL, task_finished, &proc));
  84
  85        fprintf(stderr, "check usage\n");
  86        return 1;
  87}