coredns/middleware/whoami/setup_test.go
Miek Gieben 30fd224504 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.
2016-09-17 17:09:05 +01:00

19 lines
396 B
Go

package whoami
import (
"testing"
"github.com/mholt/caddy"
)
func TestSetupWhoami(t *testing.T) {
c := caddy.NewTestController("dns", `whoami`)
if err := setupWhoami(c); err != nil {
t.Fatalf("Expected no errors, but got: %v", err)
}
c = caddy.NewTestController("dns", `whoami example.org`)
if err := setupWhoami(c); err == nil {
t.Fatalf("Expected errors, but got: %v", err)
}
}