Branch data     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                 :         246 : psql_start_test(const char *testname,
      30                 :             :                                 _stringlist **resultfiles,
      31                 :             :                                 _stringlist **expectfiles,
      32                 :             :                                 _stringlist **tags)
      33                 :             : {
      34                 :         246 :         PID_TYPE        pid;
      35                 :         246 :         char            infile[MAXPGPATH];
      36                 :         246 :         char            outfile[MAXPGPATH];
      37                 :         246 :         char            expectfile[MAXPGPATH];
      38                 :         246 :         StringInfoData psql_cmd;
      39                 :         246 :         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                 :         492 :         snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
      48                 :         246 :                          outputdir, testname);
      49         [ -  + ]:         246 :         if (!file_exists(infile))
      50                 :         492 :                 snprintf(infile, sizeof(infile), "%s/sql/%s.sql",
      51                 :         246 :                                  inputdir, testname);
      52                 :             : 
      53                 :         492 :         snprintf(outfile, sizeof(outfile), "%s/results/%s.out",
      54                 :         246 :                          outputdir, testname);
      55                 :             : 
      56                 :         492 :         snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
      57                 :         246 :                          expecteddir, testname);
      58         [ +  + ]:         246 :         if (!file_exists(expectfile))
      59                 :         474 :                 snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
      60                 :         237 :                                  inputdir, testname);
      61                 :             : 
      62                 :         246 :         add_stringlist_item(resultfiles, outfile);
      63                 :         246 :         add_stringlist_item(expectfiles, expectfile);
      64                 :             : 
      65                 :         246 :         initStringInfo(&psql_cmd);
      66                 :             : 
      67         [ +  - ]:         246 :         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                 :         246 :         appendStringInfo(&psql_cmd,
      75                 :             :                                          "\"%s%spsql\" -X -a -q -d \"%s\" %s < \"%s\" > \"%s\" 2>&1",
      76         [ -  + ]:         246 :                                          bindir ? bindir : "",
      77                 :         246 :                                          bindir ? "/" : "",
      78                 :         246 :                                          dblist->str,
      79                 :             :                                          "-v HIDE_TABLEAM=on -v HIDE_TOAST_COMPRESSION=on",
      80                 :         246 :                                          infile,
      81                 :         246 :                                          outfile);
      82                 :             : 
      83                 :         246 :         appnameenv = psprintf("pg_regress/%s", testname);
      84                 :         246 :         setenv("PGAPPNAME", appnameenv, 1);
      85                 :         246 :         free(appnameenv);
      86                 :             : 
      87                 :         246 :         pid = spawn_process(psql_cmd.data);
      88                 :             : 
      89         [ +  - ]:         246 :         if (pid == INVALID_PID)
      90                 :             :         {
      91                 :           0 :                 fprintf(stderr, _("could not start process for test %s\n"),
      92                 :           0 :                                 testname);
      93                 :           0 :                 exit(2);
      94                 :             :         }
      95                 :             : 
      96                 :         246 :         unsetenv("PGAPPNAME");
      97                 :             : 
      98                 :         246 :         pfree(psql_cmd.data);
      99                 :             : 
     100                 :         492 :         return pid;
     101                 :         246 : }
     102                 :             : 
     103                 :             : static void
     104                 :           3 : psql_init(int argc, char **argv)
     105                 :             : {
     106                 :             :         /* set default regression database name */
     107                 :           3 :         add_stringlist_item(&dblist, "regression");
     108                 :           3 : }
     109                 :             : 
     110                 :             : int
     111                 :           3 : main(int argc, char *argv[])
     112                 :             : {
     113                 :           3 :         return regression_main(argc, argv,
     114                 :             :                                                    psql_init,
     115                 :             :                                                    psql_start_test,
     116                 :             :                                                    NULL /* no postfunc needed */ );
     117                 :             : }