Function runReplicationScript

testing/go/replication_test.go:585–676

runReplicationScript runs the script given on the postgres connection provided. This does not handle assertions that use ExpectedRaw.

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

// runReplicationScript runs the script given on the postgres connection provided. This does not handle assertions that
// use ExpectedRaw.
func runReplicationScript(
	ctx context.Context,
	t *testing.T,
	script ReplicationTest,
	replicaConn *pgx.Conn,
	primaryDns string,
) {
	walFile := fmt.Sprintf("%s/%s", t.TempDir(), "wal")
	r := newReplicator(t, walFile, replicaConn, 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, query, r) {
			continue
		}

		conn := connectionForQuery(t, query, connections, primaryDns)
		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, assertion.Query, r) {
				return
			}

			target, _ := clientSpecFromQueryComment(assertion.Query)
			enableRetries := target == "replica"
			conn := connectionForQuery(t, assertion.Query, connections, primaryDns)

			numRetries := 3
			for retries := 0; retries < numRetries; retries++ {