isolation_main.c

Go to the documentation of this file.

/*-------------------------------------------------------------------------
 *
 * isolation_main --- pg_regress test launcher for isolation tests
 *
 * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
 * Portions Copyright (c) 1994, Regents of the University of California
 *
 * src/test/isolation/isolation_main.c
 *
 *-------------------------------------------------------------------------
 */
 
#include "postgres_fe.h"
 
#include "lib/stringinfo.h"
#include "pg_regress.h"
 
static char saved_argv0[MAXPGPATH];
static char isolation_exec[MAXPGPATH];
static bool looked_up_isolation_exec = false;
 
#define PG_ISOLATION_VERSIONSTR "isolationtester (PostgreSQL) " PG_VERSION "\n"
 
/*
 * start an isolation tester process for specified file (including
 * redirection), and return process ID
 */
static PID_TYPE
isolation_start_test(const char *testname,
                     _stringlist **resultfiles,
                     _stringlist **expectfiles,
                     _stringlist **tags)
{
    PID_TYPE    pid;
    char        infile[MAXPGPATH];
    char        outfile[MAXPGPATH];
    char        expectfile[MAXPGPATH];
    StringInfoData psql_cmd;
    char       *appnameenv;
 
    /* need to do the path lookup here, check isolation_init() for details */
    if (!looked_up_isolation_exec)
    {
        /* look for isolationtester binary */
        if (find_other_exec(saved_argv0, "isolationtester",
                            PG_ISOLATION_VERSIONSTR, isolation_exec) != 0)
        {
            fprintf(stderr, _("could not find proper isolationtester binary\n"));
            exit(2);
        }
        looked_up_isolation_exec = true;
    }
 
    /*
     * Look for files in the output dir first, consistent with a vpath search.
     * This is mainly to create more reasonable error messages if the file is
     * not found.  It also allows local test overrides when running pg_regress
     * outside of the source tree.
     */
    snprintf(infile, sizeof(infile), "%s/specs/%s.spec",
             outputdir, testname);
    if (!file_exists(infile))
        snprintf(infile, sizeof(infile), "%s/specs/%s.spec",
                 inputdir, testname);
 
    snprintf(outfile, sizeof(outfile), "%s/results/%s.out",
             outputdir, testname);
 
    snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
             outputdir, testname);
    if (!file_exists(expectfile))
        snprintf(expectfile, sizeof(expectfile), "%s/expected/%s.out",
                 inputdir, testname);
 
    add_stringlist_item(resultfiles, outfile);
    add_stringlist_item(expectfiles, expectfile);
 
    initStringInfo(&psql_cmd);
 
    if (launcher)
        appendStringInfo(&psql_cmd, "%s ", launcher);
 
    appendStringInfo(&psql_cmd,
                     "\"%s\" \"dbname=%s\" < \"%s\" > \"%s\" 2>&1",
                     isolation_exec,
                     dblist->str,
                     infile,
                     outfile);
 
    appnameenv = psprintf("isolation/%s", testname);
    setenv("PGAPPNAME", appnameenv, 1);
    free(appnameenv);
 
    pid = spawn_process(psql_cmd.data);
 
    if (pid == INVALID_PID)
    {
        fprintf(stderr, _("could not start process for test %s\n"),
                testname);
        exit(2);
    }
 
    unsetenv("PGAPPNAME");
 
    pfree(psql_cmd.data);
 
    return pid;
}
 
static void
isolation_init(int argc, char **argv)
{
    size_t      argv0_len;
 
    /*
     * We unfortunately cannot do the find_other_exec() lookup to find the
     * "isolationtester" binary here.  regression_main() calls the
     * initialization functions before parsing the commandline arguments and
     * thus hasn't changed the library search path at this point which in turn
     * can cause the "isolationtester -V" invocation that find_other_exec()
     * does to fail since it's linked to libpq.  So we instead copy argv[0]
     * and do the lookup the first time through isolation_start_test().
     */
    argv0_len = strlcpy(saved_argv0, argv[0], MAXPGPATH);
    if (argv0_len >= MAXPGPATH)
    {
        fprintf(stderr, _("path for isolationtester executable is longer than %d bytes\n"),
                (int) (MAXPGPATH - 1));
        exit(2);
    }
 
    /* set default regression database name */
    add_stringlist_item(&dblist, "isolation_regression");
}
 
int
main(int argc, char *argv[])
{
    return regression_main(argc, argv,
                           isolation_init,
                           isolation_start_test,
                           NULL /* no postfunc needed */ );
}