1 /*-------------------------------------------------------------------------
2 *
3 * pg_regress_main --- regression test for the main backend
4 *
5 * This is a C implementation of the previous shell script for running
6 * the regression tests, and should be mostly compatible with it.
7 * Initial author of C translation: Magnus Hagander
8 *
9 * This code is released under the terms of the PostgreSQL License.
10 *
11 * Portions Copyright (c) 1996-2018, PostgreSQL Global Development Group
12 * Portions Copyright (c) 1994, Regents of the University of California
13 *
14 * src/test/regress/pg_regress_main.c
15 *
16 *-------------------------------------------------------------------------
17 */
18
19 #include "postgres_fe.h"
20
21 #include "pg_regress.h"
22
23 /*
24 * start a psql test process for specified file (including redirection),
25 * and return process ID
26 */
27 static PID_TYPE
psql_start_test(const char * testname,_stringlist ** resultfiles,_stringlist ** expectfiles,_stringlist ** tags)28 psql_start_test(const char *testname,
29 _stringlist **resultfiles,
30 _stringlist **expectfiles,
31 _stringlist **tags)
32 {
33 PID_TYPE pid;
34 char infile[MAXPGPATH];
35 char outfile[MAXPGPATH];
36 char expectfile[MAXPGPATH];
37 char psql_cmd[MAXPGPATH * 3];
38 size_t offset = 0;
39 char *appnameenv;
40
41 /*
42 * Look for files in the output dir first, consistent with a vpath search.
43 * This is mainly to create more reasonable error messages if the file is
44 * not found. It also allows local test overrides when running pg_regress
45 * outside of the source tree.
46 */
47 snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
48 outputdir, testname);
49 if (!file_exists(infile))
50 snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
51 inputdir, testname);
52
53 snprintf(outfile, sizeof(outfile), "%s/results/%s.out",
54 outputdir, testname);
55
56 snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
57 outputdir, testname);
58 if (!file_exists(expectfile))
59 snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
60 inputdir, testname);
61
62 add_stringlist_item(resultfiles, outfile);
63 add_stringlist_item(expectfiles, expectfile);
64
65 if (launcher)
66 {
67 offset += snprintf(psql_cmd + offset, sizeof(psql_cmd) - offset,
68 "%s ", launcher);
69 if (offset >= sizeof(psql_cmd))
70 {
71 fprintf(stderr, _("command too long\n"));
72 exit(2);
73 }
74 }
75
76 offset += snprintf(psql_cmd + offset, sizeof(psql_cmd) - offset,
77 "\"%s%spsql\" -X -a -q -d \"%s\" < \"%s\" > \"%s\" 2>&1",
78 bindir ? bindir : "",
79 bindir ? "/" : "",
80 dblist->str,
81 infile,
82 outfile);
83 if (offset >= sizeof(psql_cmd))
84 {
85 fprintf(stderr, _("command too long\n"));
86 exit(2);
87 }
88
89 appnameenv = psprintf("PGAPPNAME=pg_regress/%s", testname);
90 putenv(appnameenv);
91
92 pid = spawn_process(psql_cmd);
93
94 if (pid == INVALID_PID)
95 {
96 fprintf(stderr, _("could not start process for test %s\n"),
97 testname);
98 exit(2);
99 }
100
101 unsetenv("PGAPPNAME");
102 free(appnameenv);
103
104 return pid;
105 }
106
107 static void
psql_init(int argc,char ** argv)108 psql_init(int argc, char **argv)
109 {
110 /* set default regression database name */
111 add_stringlist_item(&dblist, "regression");
112 }
113
114 int
main(int argc,char * argv[])115 main(int argc, char *argv[])
116 {
117 return regression_main(argc, argv, psql_init, psql_start_test);
118 }
119