aboutsummaryrefslogtreecommitdiff
path: root/test/middleware_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'test/middleware_test.go')
-rw-r--r--test/middleware_test.go42
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)
+ }
+}