granicus.if.org Git - postgresql/blob - src/test/regress/pg_regress_main.c

   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-2019, 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\" -v %s < \"%s\" > \"%s\" 2>&1",
  82                                            bindir ? bindir : "",
  83                                            bindir ? "/" : "",
  84                                            dblist->str,
  85                                            "HIDE_TABLEAM=\"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("PGAPPNAME=pg_regress/%s", testname);
  95         putenv(appnameenv);
  96
  97         pid = spawn_process(psql_cmd);
  98
  99         if (pid == INVALID_PID)
 100         {
 101                 fprintf(stderr, _("could not start process for test %s\n"),
 102                                 testname);
 103                 exit(2);
 104         }
 105
 106         unsetenv("PGAPPNAME");
 107         free(appnameenv);
 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, psql_init, psql_start_test);
 123 }