Age Owner TLA Line data Source code
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
5780 magnus 28 CBC 916 : 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];
4459 rhaas 38 916 : 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 : */
5780 magnus 47 916 : snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
48 : outputdir, testname);
5303 peter_e 49 916 : if (!file_exists(infile))
50 916 : snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
51 : inputdir, testname);
52 :
5780 magnus 53 916 : snprintf(outfile, sizeof(outfile), "%s/results/%s.out",
54 : outputdir, testname);
55 :
56 916 : snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
57 : outputdir, testname);
5303 peter_e 58 916 : if (!file_exists(expectfile))
59 916 : snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
60 : inputdir, testname);
61 :
5780 magnus 62 916 : add_stringlist_item(resultfiles, outfile);
63 916 : add_stringlist_item(expectfiles, expectfile);
64 :
4459 rhaas 65 916 : if (launcher)
66 : {
4459 rhaas 67 UBC 0 : offset += snprintf(psql_cmd + offset, sizeof(psql_cmd) - offset,
68 : "%s ", launcher);
1698 tgl 69 0 : if (offset >= sizeof(psql_cmd))
70 : {
71 0 : fprintf(stderr, _("command too long\n"));
72 0 : 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 : */
1698 tgl 80 CBC 1832 : offset += snprintf(psql_cmd + offset, sizeof(psql_cmd) - offset,
81 : "\"%s%spsql\" -X -a -q -d \"%s\" %s < \"%s\" > \"%s\" 2>&1",
82 916 : bindir ? bindir : "",
83 916 : bindir ? "/" : "",
84 916 : dblist->str,
85 : "-v HIDE_TABLEAM=on -v HIDE_TOAST_COMPRESSION=on",
86 : infile,
87 : outfile);
88 916 : if (offset >= sizeof(psql_cmd))
89 : {
1698 tgl 90 UBC 0 : fprintf(stderr, _("command too long\n"));
91 0 : exit(2);
92 : }
93 :
830 tgl 94 CBC 916 : appnameenv = psprintf("pg_regress/%s", testname);
95 916 : setenv("PGAPPNAME", appnameenv, 1);
96 916 : free(appnameenv);
97 :
5780 magnus 98 916 : pid = spawn_process(psql_cmd);
99 :
100 916 : if (pid == INVALID_PID)
101 : {
5780 magnus 102 UBC 0 : fprintf(stderr, _("could not start process for test %s\n"),
103 : testname);
4115 peter_e 104 0 : exit(2);
105 : }
106 :
2382 peter_e 107 CBC 916 : unsetenv("PGAPPNAME");
108 :
5780 magnus 109 916 : return pid;
110 : }
111 :
112 : static void
3439 rhaas 113 295 : psql_init(int argc, char **argv)
114 : {
115 : /* set default regression database name */
5780 magnus 116 295 : add_stringlist_item(&dblist, "regression");
117 295 : }
118 :
119 : int
120 295 : main(int argc, char *argv[])
121 : {
818 tgl 122 295 : return regression_main(argc, argv,
123 : psql_init,
124 : psql_start_test,
125 : NULL /* no postfunc needed */ );
126 : }
|