aboutsummaryrefslogtreecommitdiff
path: root/middleware/chaos/chaos_test.go
diff options
context:
space:
mode:
authorGravatar Miek Gieben <miek@miek.nl> 2016-09-17 17:09:05 +0100
committerGravatar GitHub <noreply@github.com> 2016-09-17 17:09:05 +0100
commit30fd224504d7e6fcfa7da027d62d2105ecb6f0c5 (patch)
treed63f32750f56265e13d5a8c98ef840f1c343395f /middleware/chaos/chaos_test.go
parented907d33278003072443029fef666232f16e1985 (diff)
downloadcoredns-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.go12
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"