diff options
author | 2016-09-17 17:09:05 +0100 | |
---|---|---|
committer | 2016-09-17 17:09:05 +0100 | |
commit | 30fd224504d7e6fcfa7da027d62d2105ecb6f0c5 (patch) | |
tree | d63f32750f56265e13d5a8c98ef840f1c343395f /middleware/chaos/chaos_test.go | |
parent | ed907d33278003072443029fef666232f16e1985 (diff) | |
download | coredns-30fd224504d7e6fcfa7da027d62d2105ecb6f0c5.tar.gz coredns-30fd224504d7e6fcfa7da027d62d2105ecb6f0c5.tar.zst coredns-30fd224504d7e6fcfa7da027d62d2105ecb6f0c5.zip |
middleware/whoami: add (#264)
Add a new middleware that tells you who you are; IP, port and transport
is echoed back.
Also some various cleanup and documentation improvements while at it:
* ResponseWriter: improve the documentation of these helper functions.
* And add an NextHandler for use in tests. Make chaos_test.go and
* whoam_test.go use it.
Diffstat (limited to 'middleware/chaos/chaos_test.go')
-rw-r--r-- | middleware/chaos/chaos_test.go | 12 |
1 files changed, 3 insertions, 9 deletions
diff --git a/middleware/chaos/chaos_test.go b/middleware/chaos/chaos_test.go index 6a3261754..333083187 100644 --- a/middleware/chaos/chaos_test.go +++ b/middleware/chaos/chaos_test.go @@ -26,21 +26,21 @@ func TestChaos(t *testing.T) { expectedErr error }{ { - next: genHandler(dns.RcodeSuccess, nil), + next: test.NextHandler(dns.RcodeSuccess, nil), qname: "version.bind", expectedCode: dns.RcodeSuccess, expectedReply: version, expectedErr: nil, }, { - next: genHandler(dns.RcodeSuccess, nil), + next: test.NextHandler(dns.RcodeSuccess, nil), qname: "authors.bind", expectedCode: dns.RcodeSuccess, expectedReply: "Miek Gieben", expectedErr: nil, }, { - next: genHandler(dns.RcodeSuccess, nil), + next: test.NextHandler(dns.RcodeSuccess, nil), qname: "authors.bind", qtype: dns.TypeSRV, expectedCode: dns.RcodeSuccess, @@ -77,10 +77,4 @@ func TestChaos(t *testing.T) { } } -func genHandler(rcode int, err error) middleware.Handler { - return middleware.HandlerFunc(func(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { - return rcode, err - }) -} - const version = "CoreDNS-001" |