Function newBufferedConn

server/query_source_test.go:80–97

newBufferedConn builds a clientConn that writes to an in-memory buffer, for asserting the exact backend messages a simple query produces. The returned *bytes.Buffer holds the flushed output after handleQuery.

(exec QueryExecutor)

		}
		var coded interface{ SQLState() string }
		if !errors.As(err, &coded) || coded.SQLState() != "22023" {
			t.Fatalf("applyStartupQuerySource(%.40q) error = %v, want SQLSTATE 22023", raw, err)
		}
		if !strings.Contains(err.Error(), `"standard"`) || !strings.Contains(err.Error(), `"endpoints"`) {
			t.Fatalf("rejection must name the valid values, got %q", err.Error())
		}
		if got := c.QuerySource(); got != "standard" {
			t.Fatalf("after rejected startup option, QuerySource() = %q, want default %q", got, "standard")
		}
	}
}

// wireMsg is a decoded backend protocol message: a type byte and its body.
type wireMsg struct {
	typ  byte
	body []byte
}

// parseWireMsgs decodes the sequence of length-prefixed backend messages a
// clientConn wrote to its buffer (type byte + 4-byte length incl. length).
func parseWireMsgs(t *testing.T, buf []byte) []wireMsg {