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
  18 static int number_callbacks;
  19 static 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
  34 static 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
  43 static 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
  52 int 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 }