diff options
Diffstat (limited to 'test/middleware_test.go')
-rw-r--r-- | test/middleware_test.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/test/middleware_test.go b/test/middleware_test.go new file mode 100644 index 000000000..ec90f0d71 --- /dev/null +++ b/test/middleware_test.go @@ -0,0 +1,42 @@ +package test + +import ( + "io/ioutil" + "log" + "testing" + + "github.com/miekg/coredns/middleware/test" + + "github.com/miekg/dns" +) + +func BenchmarkLookupBalanceRewriteCache(b *testing.B) { + t := new(testing.T) + name, rm, err := test.TempFile(t, ".", exampleOrg) + if err != nil { + t.Fatalf("failed to created zone: %s", err) + } + defer rm() + + corefile := `example.org:0 { + file ` + name + ` + rewrite ANY HINFO + loadbalance +} +` + ex, _, udp, err := Server(t, corefile) + if err != nil { + t.Fatalf("Could get server: %s", err) + } + defer ex.Stop() + + log.SetOutput(ioutil.Discard) + c := new(dns.Client) + m := new(dns.Msg) + m.SetQuestion("example.org.", dns.TypeA) + + b.ResetTimer() + for i := 0; i < b.N; i++ { + c.Exchange(m, udp) + } +} |