mirror of
https://github.com/golang/go
synced 2024-11-23 14:00:03 -07:00
a040ebeb98
Update references missed in CL 263142.
For #41190
Change-Id: I778760a6a69bd0440fec0848bdef539c9ccb4ee1
GitHub-Last-Rev: dda42b09ff
GitHub-Pull-Request: golang/go#42874
Reviewed-on: https://go-review.googlesource.com/c/go/+/273946
Run-TryBot: Ian Lance Taylor <iant@golang.org>
TryBot-Result: Go Bot <gobot@golang.org>
Reviewed-by: Ian Lance Taylor <iant@golang.org>
Trust: Cherry Zhang <cherryyz@google.com>
46 lines
949 B
Go
46 lines
949 B
Go
// Copyright 2013 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package go1
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
// BenchmarkHTTPClientServer benchmarks both the HTTP client and the HTTP server,
|
|
// on small requests.
|
|
func BenchmarkHTTPClientServer(b *testing.B) {
|
|
msg := []byte("Hello world.\n")
|
|
ts := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
|
rw.Write(msg)
|
|
}))
|
|
defer ts.Close()
|
|
|
|
tr := &http.Transport{}
|
|
defer tr.CloseIdleConnections()
|
|
cl := &http.Client{
|
|
Transport: tr,
|
|
}
|
|
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
res, err := cl.Get(ts.URL)
|
|
if err != nil {
|
|
b.Fatal("Get:", err)
|
|
}
|
|
all, err := io.ReadAll(res.Body)
|
|
if err != nil {
|
|
b.Fatal("ReadAll:", err)
|
|
}
|
|
if !bytes.Equal(all, msg) {
|
|
b.Fatalf("Got body %q; want %q", all, msg)
|
|
}
|
|
}
|
|
}
|