Function newTestDatabaseConnection

testing/go/framework.go:469–510

newTestDatabaseConnection returns a Connection to the test |database| at |host|:|port|. If the |database| provided does not exist, it will be automatically created.

(t *testing.T, ctx context.Context, database, host string, port int)

// newTestDatabaseConnection returns a Connection to the test |database| at |host|:|port|. If the |database| provided
// does not exist, it will be automatically created.
func newTestDatabaseConnection(t *testing.T, ctx context.Context, database, host string, port int) *Connection {
	const connectionUrlFmt = "postgres://postgres:password@%s:%d/%s?DateStyle=ISO%%2C%%20MDY"
	func() {
		var conn *pgx.Conn
		var err error
		// Connections can happen before the server has a chance to grab the port so we retry.
		for range 3 {
			conn, err = pgx.Connect(ctx, fmt.Sprintf(connectionUrlFmt, host, port, ""))
			if err == nil {
				break
			}

			time.Sleep(time.Second)
		}
		require.NoError(t, err)

		_, err = conn.Exec(ctx, fmt.Sprintf("CREATE DATABASE IF NOT EXISTS %s", database))
		require.NoError(t, err)

		defer require.NoError(t, conn.Close(ctx))
	}()

	config, err := pgx.ParseConfig(fmt.Sprintf(connectionUrlFmt, host, port, database))
	require.NoError(t, err)
	config.OnNotice = func(conn *pgconn.PgConn, notice *pgconn.Notice) {
		receivedNotices = append(receivedNotices, notice)
	}
	// pgx v5.9.1+ skips Describe(portal) on statement-cache hits, so stale field descriptions
	// from before a DDL change (e.g. DROP COLUMN) cause a field-count/value-count mismatch and
	// panic in rows.Values(). DescribeExec re-describes every query, keeping schema info fresh.
	// It also preserves binary wire format (needed for record type decoding).
	config.DefaultQueryExecMode = pgx.QueryExecModeDescribeExec

	conn, err := pgx.ConnectConfig(ctx, config)
	require.NoError(t, err)
	// Ping tests that Doltgres can handle empty queries, and makes sure connection is alive.
	require.NoError(t, conn.Ping(ctx))
	return &Connection{
		Default: conn,
		Current: conn,
	}
}

// ReadRows reads all of the given rows into a slice, then closes the rows. If `normalizeRows` is true, then the rows
// will be normalized such that all integers are int64, etc. Normalization does not affect the raw returned bytes.