mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-26 03:25:35 +00:00
36 lines
906 B
Go
36 lines
906 B
Go
// Copyright (c) 2022 Tailscale Inc & 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 main
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkHandleBootstrapDNS(b *testing.B) {
|
|
prev := *bootstrapDNS
|
|
*bootstrapDNS = "log.tailscale.io,login.tailscale.com,controlplane.tailscale.com,login.us.tailscale.com"
|
|
defer func() {
|
|
*bootstrapDNS = prev
|
|
}()
|
|
refreshBootstrapDNS()
|
|
w := new(bitbucketResponseWriter)
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
b.RunParallel(func(b *testing.PB) {
|
|
for b.Next() {
|
|
handleBootstrapDNS(w, nil)
|
|
}
|
|
})
|
|
}
|
|
|
|
type bitbucketResponseWriter struct{}
|
|
|
|
func (b *bitbucketResponseWriter) Header() http.Header { return make(http.Header) }
|
|
|
|
func (b *bitbucketResponseWriter) Write(p []byte) (int, error) { return len(p), nil }
|
|
|
|
func (b *bitbucketResponseWriter) WriteHeader(statusCode int) {}
|