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-2026, 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 "lib/stringinfo.h"
      22              : #include "pg_regress.h"
      23              : 
      24              : /*
      25              :  * start a psql test process for specified file (including redirection),
      26              :  * and return process ID
      27              :  */
      28              : static PID_TYPE
      29         1033 : psql_start_test(const char *testname,
      30              :                 _stringlist **resultfiles,
      31              :                 _stringlist **expectfiles,
      32              :                 _stringlist **tags)
      33              : {
      34              :     PID_TYPE    pid;
      35              :     char        infile[MAXPGPATH];
      36              :     char        outfile[MAXPGPATH];
      37              :     char        expectfile[MAXPGPATH];
      38              :     StringInfoData psql_cmd;
      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         1033 :     snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
      48              :              outputdir, testname);
      49         1033 :     if (!file_exists(infile))
      50          478 :         snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
      51              :                  inputdir, testname);
      52              : 
      53         1033 :     snprintf(outfile, sizeof(outfile), "%s/results/%s.out",
      54              :              outputdir, testname);
      55              : 
      56         1033 :     snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
      57              :              expecteddir, testname);
      58         1033 :     if (!file_exists(expectfile))
      59          478 :         snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
      60              :                  inputdir, testname);
      61              : 
      62         1033 :     add_stringlist_item(resultfiles, outfile);
      63         1033 :     add_stringlist_item(expectfiles, expectfile);
      64              : 
      65         1033 :     initStringInfo(&psql_cmd);
      66              : 
      67         1033 :     if (launcher)
      68            0 :         appendStringInfo(&psql_cmd, "%s ", launcher);
      69              : 
      70              :     /*
      71              :      * Use HIDE_TABLEAM to hide different AMs to allow to use regression tests
      72              :      * against different AMs without unnecessary differences.
      73              :      */
      74         2066 :     appendStringInfo(&psql_cmd,
      75              :                      "\"%s%spsql\" -X -a -q -d \"%s\" %s < \"%s\" > \"%s\" 2>&1",
      76         1033 :                      bindir ? bindir : "",
      77         1033 :                      bindir ? "/" : "",
      78         1033 :                      dblist->str,
      79              :                      "-v HIDE_TABLEAM=on -v HIDE_TOAST_COMPRESSION=on",
      80              :                      infile,
      81              :                      outfile);
      82              : 
      83         1033 :     appnameenv = psprintf("pg_regress/%s", testname);
      84         1033 :     setenv("PGAPPNAME", appnameenv, 1);
      85         1033 :     free(appnameenv);
      86              : 
      87         1033 :     pid = spawn_process(psql_cmd.data);
      88              : 
      89         1033 :     if (pid == INVALID_PID)
      90              :     {
      91            0 :         fprintf(stderr, _("could not start process for test %s\n"),
      92              :                 testname);
      93            0 :         exit(2);
      94              :     }
      95              : 
      96         1033 :     unsetenv("PGAPPNAME");
      97              : 
      98         1033 :     pfree(psql_cmd.data);
      99              : 
     100         1033 :     return pid;
     101              : }
     102              : 
     103              : static void
     104          427 : psql_init(int argc, char **argv)
     105              : {
     106              :     /* set default regression database name */
     107          427 :     add_stringlist_item(&dblist, "regression");
     108          427 : }
     109              : 
     110              : int
     111          427 : main(int argc, char *argv[])
     112              : {
     113          427 :     return regression_main(argc, argv,
     114              :                            psql_init,
     115              :                            psql_start_test,
     116              :                            NULL /* no postfunc needed */ );
     117              : }