Function TestMultipleStatements

testing/go/multi_statement_test.go:30–148

TestMultipleStatements is a test for: https://github.com/dolthub/doltgresql/issues/2175

(t *testing.T)

// TestMultipleStatements is a test for: https://github.com/dolthub/doltgresql/issues/2175
func TestMultipleStatements(t *testing.T) {
	ctx := context.Background()
	var conn *Connection
	if runOnPostgres {
		pgxConn, err := pgx.Connect(ctx, "postgres://postgres:password@127.0.0.1:5432/postgres?sslmode=disable")
		require.NoError(t, err)
		conn = &Connection{
			Default: pgxConn,
			Current: pgxConn,
		}
		require.NoError(t, pgxConn.Ping(ctx))
		defer func() {
			conn.Close(ctx)
		}()
	} else {
		var controller *svcs.Controller
		ctx, conn, controller = CreateServer(t, "postgres")
		defer func() {
			conn.Close(ctx)
			controller.Stop()
			err := controller.WaitForStop()
			require.NoError(t, err)
		}()
	}
	queries := []string{
		`BEGIN;`,
		`DROP TABLE IF EXISTS migrations;`,
		`DROP TABLE IF EXISTS animals;`,
		`CREATE TABLE IF NOT EXISTS migrations (file_name TEXT NOT NULL, file_hash TEXT NOT NULL);`,
		`CREATE TABLE IF NOT EXISTS animals (id SERIAL PRIMARY KEY NOT NULL, name TEXT NOT NULL);`,
		`;`, // This should be ignored in the output
		`INSERT INTO migrations (file_name, file_hash) VALUES ('2021-09-07T154500-create-animals-table.sql', '42331f4277227d09e9bb32eeaf7e04d9c7fe320160e05372ed0ef010cfbf666b');`,
		`INSERT INTO animals(name) VALUES('Alpaca');`,
		`INSERT INTO animals(name) VALUES('Highland cow');`,
		`INSERT INTO animals(name) VALUES('Aardvark');`,
		`INSERT INTO migrations (file_name, file_hash) VALUES ('2021-09-07T154700-insert-animals.sql', '3223d0deb6fb7fb2accf6abffc0667ebe4503379987c472d10a585a553f9b3b6');`,
		`SELECT * FROM migrations ORDER BY file_name;`,
		`SELECT * FROM animals ORDER BY id;`,
		`COMMIT;`,
	}
	combinedQueries := ""
	for _, query := range queries {
		// We do this just to homogenize the queries, even though we're adding the delimiter right back
		query = sql.RemoveSpaceAndDelimiter(query, ';')
		combinedQueries += query + ";"
	}
	// First we'll check all invalid modes that fail immediately
	invalidModes := []pgx.QueryExecMode{
		pgx.QueryExecModeCacheStatement,
		pgx.QueryExecModeCacheDescribe,
		pgx.QueryExecModeDescribeExec,
		pgx.QueryExecModeExec,
	}
	for _, mode := range invalidModes {
		rows, err := conn.Current.Query(ctx, combinedQueries, mode)
		if mode == pgx.QueryExecModeExec {
			// This mode requires reading from the returned rows to find the error, rather than erroring immediately
			require.NoError(t, err)