coredns/core/setup/chaos_test.go
Miek Gieben 5d70567f1c Implement chaos middleware
This allows for CH TXT queries that return some information about
the server and/or the authors (or whatever you put in there).
2016-03-25 08:36:32 +00:00

61 lines
1.6 KiB
Go

package setup
import (
"fmt"
"strings"
"testing"
)
func TestChaos(t *testing.T) {
tests := []struct {
input string
shouldErr bool
expectedVersion string // expected veresion.
expectedAuthor string // expected author (string, although we get a map).
expectedErrContent string // substring from the expected error. Empty for positive cases.
}{
// positive
{
`chaos`, false, defaultVersion, "", "",
},
{
`chaos v2`, false, "v2", "", "",
},
{
`chaos v3 "Miek Gieben"`, false, "v3", "Miek Gieben", "",
},
{
fmt.Sprintf(`chaos {
%s
}`, defaultVersion), false, defaultVersion, "", "",
},
}
for i, test := range tests {
c := NewTestController(test.input)
version, authors, err := chaosParse(c)
if test.shouldErr && err == nil {
t.Errorf("Test %d: Expected error but found %s for input %s", i, err, test.input)
}
if err != nil {
if !test.shouldErr {
t.Errorf("Test %d: Expected no error but found one for input %s. Error was: %v", i, test.input, err)
}
if !strings.Contains(err.Error(), test.expectedErrContent) {
t.Errorf("Test %d: Expected error to contain: %v, found error: %v, input: %s", i, test.expectedErrContent, err, test.input)
}
}
if !test.shouldErr && version != test.expectedVersion {
t.Errorf("Chaos not correctly set for input %s. Expected: %s, actual: %s", test.input, test.expectedVersion, version)
}
if !test.shouldErr && authors != nil {
if _, ok := authors[test.expectedAuthor]; !ok {
t.Errorf("Chaos not correctly set for input %s. Expected: '%s', actual: '%s'", test.input, test.expectedAuthor, "Miek Gieben")
}
}
}
}