PostgreSQL Source Code  git master
pg_regress_main.c
Go to the documentation of this file.
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-2023, 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
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  /*
77  * Use HIDE_TABLEAM to hide different AMs to allow to use regression tests
78  * against different AMs without unnecessary differences.
79  */
80  offset += snprintf(psql_cmd + offset, sizeof(psql_cmd) - offset,
81  "\"%s%spsql\" -X -a -q -d \"%s\" %s < \"%s\" > \"%s\" 2>&1",
82  bindir ? bindir : "",
83  bindir ? "/" : "",
84  dblist->str,
85  "-v HIDE_TABLEAM=on -v HIDE_TOAST_COMPRESSION=on",
86  infile,
87  outfile);
88  if (offset >= sizeof(psql_cmd))
89  {
90  fprintf(stderr, _("command too long\n"));
91  exit(2);
92  }
93 
94  appnameenv = psprintf("pg_regress/%s", testname);
95  setenv("PGAPPNAME", appnameenv, 1);
96  free(appnameenv);
97 
98  pid = spawn_process(psql_cmd);
99 
100  if (pid == INVALID_PID)
101  {
102  fprintf(stderr, _("could not start process for test %s\n"),
103  testname);
104  exit(2);
105  }
106 
107  unsetenv("PGAPPNAME");
108 
109  return pid;
110 }
111 
112 static void
113 psql_init(int argc, char **argv)
114 {
115  /* set default regression database name */
116  add_stringlist_item(&dblist, "regression");
117 }
118 
119 int
120 main(int argc, char *argv[])
121 {
122  return regression_main(argc, argv,
123  psql_init,
125  NULL /* no postfunc needed */ );
126 }
#define _(x)
Definition: elog.c:91
#define free(a)
Definition: header.h:65
static void add_stringlist_item(_stringlist **listhead, const char *str)
Definition: initdb.c:387
static bool file_exists(const char *name)
Definition: jit.c:194
exit(1)
#define MAXPGPATH
static char * outfile
char * outputdir
Definition: pg_regress.c:94
char * launcher
Definition: pg_regress.c:97
char * inputdir
Definition: pg_regress.c:93
int regression_main(int argc, char *argv[], init_function ifunc, test_start_function startfunc, postprocess_result_function postfunc)
Definition: pg_regress.c:2073
_stringlist * dblist
Definition: pg_regress.c:91
PID_TYPE spawn_process(const char *cmdline)
Definition: pg_regress.c:1189
char * bindir
Definition: pg_regress.c:96
#define PID_TYPE
Definition: pg_regress.h:14
#define INVALID_PID
Definition: pg_regress.h:15
int main(int argc, char *argv[])
static void psql_init(int argc, char **argv)
static PID_TYPE psql_start_test(const char *testname, _stringlist **resultfiles, _stringlist **expectfiles, _stringlist **tags)
#define snprintf
Definition: port.h:238
#define fprintf
Definition: port.h:242
char * psprintf(const char *fmt,...)
Definition: psprintf.c:46
char * str
Definition: initdb.c:91
#define unsetenv(x)
Definition: win32_port.h:548
#define setenv(x, y, z)
Definition: win32_port.h:547
static void infile(const char *name)
Definition: zic.c:1243