Function runReplicationScript

pgserver/logrepl/replication_test.go:738–836

runReplicationScript runs the script given on the postgres connection provided

(
	ctx context.Context,
	t *testing.T,
	script ReplicationTest,
	server *pgserver.Server,
	replicaConn *pgx.Conn,
	primaryDns string,
)

// runReplicationScript runs the script given on the postgres connection provided
func runReplicationScript(
	ctx context.Context,
	t *testing.T,
	script ReplicationTest,
	server *pgserver.Server,
	replicaConn *pgx.Conn,
	primaryDns string,
) {
	r := newReplicator(server.NewInternalCtx(), t, primaryDns)
	defer r.Stop()

	if script.Skip {
		t.Skip("Skip has been set in the script")
	}

	connections := map[string]*pgx.Conn{
		"replica": replicaConn,
	}

	defer func() {
		for _, conn := range connections {
			if conn != nil {
				conn.Close(ctx)
			}
		}
	}()

	// Run the setup
	for _, query := range script.SetUpScript {
		// handle logic for special pseudo-queries
		if handlePseudoQuery(t, server, query, r) {
			continue
		}

		conn := connectionForQuery(t, query, connections, primaryDns)

		target, _ := clientSpecFromQueryComment(query)
		if target == "replica" && configuration.IsReplicationWithoutIndex() {
			// Remove the primary key index from the replica table
			query = strings.Replace(query, " primary key,", ",", 1)
		}

		log.Println("Running setup query:", query)
		_, err := conn.Exec(ctx, query)
		require.NoError(t, err)
	}

	// Run the assertions
	for _, assertion := range script.Assertions {
		t.Run(assertion.Query, func(t *testing.T) {
			if assertion.Skip {
				t.Skip("Skip has been set in the assertion")
			}

			// handle logic for special pseudo-queries
			if handlePseudoQuery(t, server, assertion.Query, r) {
				return
			}