38 lines
717 B
Go
38 lines
717 B
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/containous/traefik/testhelpers"
|
||
|
|
)
|
||
|
|
|
||
|
|
type staticTransport struct {
|
||
|
|
res *http.Response
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *staticTransport) RoundTrip(r *http.Request) (*http.Response, error) {
|
||
|
|
return t.res, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func BenchmarkProxy(b *testing.B) {
|
||
|
|
res := &http.Response{
|
||
|
|
StatusCode: 200,
|
||
|
|
Body: ioutil.NopCloser(strings.NewReader("")),
|
||
|
|
}
|
||
|
|
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
req := testhelpers.MustNewRequest(http.MethodGet, "http://foo.bar/", nil)
|
||
|
|
|
||
|
|
pool := newBufferPool()
|
||
|
|
handler, _ := buildProxy(false, nil, &staticTransport{res}, pool, nil)
|
||
|
|
|
||
|
|
b.ReportAllocs()
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
handler.ServeHTTP(w, req)
|
||
|
|
}
|
||
|
|
}
|