TrueCloudLab/coredns
16
0
Fork 2
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

core: add ServeDNS benchmark (#1158)

Browse source 
* core: add ServeDNS benchmark

Add benchmark function so we can perf test future additions to the
servers' ServeDNS function.

* naming
This commit is contained in:
Miek Gieben 2017-10-21 09:30:59 +01:00 • committed by GitHub
parent d64b684831
commit 0af3fbab4f
2 changed files with 49 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

51
core/dnsserver/server_test.go
View file

@ -2,25 +2,66 @@ package dnsserver
import (
"testing"
"github.com/coredns/coredns/plugin"
"github.com/coredns/coredns/plugin/test"
"github.com/miekg/dns"
"golang.org/x/net/context"
)
func makeConfig(transport string) *Config {
return &Config{Zone: "example.com", Transport: transport, ListenHost: "127.0.0.1", Port: "53"}
type testPlugin struct{}
func (tp testPlugin) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
return 0, nil
}
func (tp testPlugin) Name() string { return "testplugin" }
func testConfig(transport string) *Config {
c := &Config{
Zone: "example.com.",
Transport: transport,
ListenHost: "127.0.0.1",
Port: "53",
Debug: false,
}
c.AddPlugin(func(next plugin.Handler) plugin.Handler { return testPlugin{} })
return c
}
func TestNewServer(t *testing.T) {
_, err := NewServer("127.0.0.1:53", []*Config{makeConfig("dns")})
_, err := NewServer("127.0.0.1:53", []*Config{testConfig("dns")})
if err != nil {
t.Errorf("Expected no error for NewServer, got %s.", err)
}
_, err = NewServergRPC("127.0.0.1:53", []*Config{makeConfig("grpc")})
_, err = NewServergRPC("127.0.0.1:53", []*Config{testConfig("grpc")})
if err != nil {
t.Errorf("Expected no error for NewServergRPC, got %s.", err)
}
_, err = NewServerTLS("127.0.0.1:53", []*Config{makeConfig("tls")})
_, err = NewServerTLS("127.0.0.1:53", []*Config{testConfig("tls")})
if err != nil {
t.Errorf("Expected no error for NewServerTLS, got %s.", err)
}
}
func BenchmarkCoreServeDNS(b *testing.B) {
s, err := NewServer("127.0.0.1:53", []*Config{testConfig("dns")})
if err != nil {
b.Errorf("Expected no error for NewServer, got %s.", err)
}
ctx := context.TODO()
w := &test.ResponseWriter{}
m := new(dns.Msg)
m.SetQuestion("aaa.example.com.", dns.TypeTXT)
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
s.ServeDNS(ctx, w, m)
}
}