Merge branch 'jk/close-duped-fd-before-unlock-for-bundle'
[git] / 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 "test-tool.h"
12 #include "git-compat-util.h"
13 #include "run-command.h"
14 #include "argv-array.h"
15 #include "strbuf.h"
16 #include <string.h>
17 #include <errno.h>
18
19 static int number_callbacks;
20 static int parallel_next(struct child_process *cp,
21                          struct strbuf *err,
22                          void *cb,
23                          void **task_cb)
24 {
25         struct child_process *d = cb;
26         if (number_callbacks >= 4)
27                 return 0;
28
29         argv_array_pushv(&cp->args, d->argv);
30         strbuf_addstr(err, "preloaded output of a child\n");
31         number_callbacks++;
32         return 1;
33 }
34
35 static int no_job(struct child_process *cp,
36                   struct strbuf *err,
37                   void *cb,
38                   void **task_cb)
39 {
40         strbuf_addstr(err, "no further jobs available\n");
41         return 0;
42 }
43
44 static int task_finished(int result,
45                          struct strbuf *err,
46                          void *pp_cb,
47                          void *pp_task_cb)
48 {
49         strbuf_addstr(err, "asking for a quick stop\n");
50         return 1;
51 }
52
53 int cmd__run_command(int argc, const char **argv)
54 {
55         struct child_process proc = CHILD_PROCESS_INIT;
56         int jobs;
57
58         if (argc < 3)
59                 return 1;
60         while (!strcmp(argv[1], "env")) {
61                 if (!argv[2])
62                         die("env specifier without a value");
63                 argv_array_push(&proc.env_array, argv[2]);
64                 argv += 2;
65                 argc -= 2;
66         }
67         if (argc < 3)
68                 return 1;
69         proc.argv = (const char **)argv + 2;
70
71         if (!strcmp(argv[1], "start-command-ENOENT")) {
72                 if (start_command(&proc) < 0 && errno == ENOENT)
73                         return 0;
74                 fprintf(stderr, "FAIL %s\n", argv[1]);
75                 return 1;
76         }
77         if (!strcmp(argv[1], "run-command"))
78                 exit(run_command(&proc));
79
80         jobs = atoi(argv[2]);
81         proc.argv = (const char **)argv + 3;
82
83         if (!strcmp(argv[1], "run-command-parallel"))
84                 exit(run_processes_parallel(jobs, parallel_next,
85                                             NULL, NULL, &proc));
86
87         if (!strcmp(argv[1], "run-command-abort"))
88                 exit(run_processes_parallel(jobs, parallel_next,
89                                             NULL, task_finished, &proc));
90
91         if (!strcmp(argv[1], "run-command-no-jobs"))
92                 exit(run_processes_parallel(jobs, no_job,
93                                             NULL, task_finished, &proc));
94
95         fprintf(stderr, "check usage\n");
96         return 1;
97 }