Function RunScriptN

testing/go/prepared_statement_test.go:1498–1562

RunScriptN runs the assertions of the given script n times using the same connection

(t *testing.T, script ScriptTest, n int)

// RunScriptN runs the assertions of the given script n times using the same connection
func RunScriptN(t *testing.T, script ScriptTest, n int) {
	scriptDatabase := script.Database
	if len(scriptDatabase) == 0 {
		scriptDatabase = "postgres"
	}
	ctx, conn, controller := CreateServer(t, scriptDatabase)
	defer func() {
		conn.Close(ctx)
		controller.Stop()
		err := controller.WaitForStop()
		require.NoError(t, err)
	}()

	// Run the setup
	for _, query := range script.SetUpScript {
		rows, err := conn.Query(ctx, query)
		require.NoError(t, err)
		_, _, err = ReadRows(rows, true)
		assert.NoError(t, err)
	}

	for i := 0; i < n; i++ {
		t.Run(script.Name, func(t *testing.T) {
			// 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")
					}
					rows, err := conn.Query(ctx, assertion.Query)
					if err == nil {
						defer rows.Close()
					}

					var errorSeen string

					if assertion.ExpectedErr == "" {
						require.NoError(t, err)
					} else if err != nil {
						errorSeen = err.Error()
					}

					if errorSeen == "" {
						foundRows, foundRawRows, err := ReadRows(rows, true)
						if assertion.ExpectedErr == "" {
							require.NoError(t, err)
							if assertion.ExpectedRaw != nil {
								assert.Equal(t, assertion.ExpectedRaw, foundRawRows)
							} else {
								assert.Equal(t, NormalizeExpectedRow(rows.FieldDescriptions(), assertion.Expected), foundRows)
							}
						} else if err != nil {
							errorSeen = err.Error()
						}
					}

					if assertion.ExpectedErr != "" {
						require.False(t, errorSeen == "", "Expected error but got none")