818d2b10ad
* cidr everywhere: check all middleware Add tests for cidr in only that middleware that already tests for this. Check the other ones manually (and put reverse in the tests cases anyway). Make etcd setup_test run without +build etcd tag - it is not needed for this test - move rest of the code to lookup_test.go. Cleanup proxy test a bit and remove TempDir as there is test.TempFile that does the same thing. Fixes #909 * coredns package * Fix test compile
100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package auto
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mholt/caddy"
|
|
)
|
|
|
|
func TestAutoParse(t *testing.T) {
|
|
tests := []struct {
|
|
inputFileRules string
|
|
shouldErr bool
|
|
expectedDirectory string
|
|
expectedTempl string
|
|
expectedRe string
|
|
expectedTo []string
|
|
}{
|
|
{
|
|
`auto example.org {
|
|
directory /tmp
|
|
transfer to 127.0.0.1
|
|
}`,
|
|
false, "/tmp", "${1}", `db\.(.*)`, []string{"127.0.0.1:53"},
|
|
},
|
|
{
|
|
`auto 10.0.0.0/24 {
|
|
directory /tmp
|
|
}`,
|
|
false, "/tmp", "${1}", `db\.(.*)`, nil,
|
|
},
|
|
{
|
|
`auto {
|
|
directory /tmp
|
|
}`,
|
|
false, "/tmp", "${1}", `db\.(.*)`, nil,
|
|
},
|
|
{
|
|
`auto {
|
|
directory /tmp (.*) bliep
|
|
}`,
|
|
false, "/tmp", "bliep", `(.*)`, nil,
|
|
},
|
|
{
|
|
`auto {
|
|
directory /tmp (.*) bliep
|
|
transfer to 127.0.0.1
|
|
transfer to 127.0.0.2
|
|
upstream 8.8.8.8
|
|
}`,
|
|
false, "/tmp", "bliep", `(.*)`, []string{"127.0.0.1:53", "127.0.0.2:53"},
|
|
},
|
|
// errors
|
|
{
|
|
`auto example.org {
|
|
directory
|
|
}`,
|
|
true, "", "${1}", `db\.(.*)`, nil,
|
|
},
|
|
{
|
|
`auto example.org {
|
|
directory /tmp * {1}
|
|
}`,
|
|
true, "", "${1}", ``, nil,
|
|
},
|
|
{
|
|
`auto example.org {
|
|
directory /tmp .* {1}
|
|
}`,
|
|
true, "", "${1}", ``, nil,
|
|
},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
c := caddy.NewTestController("dns", test.inputFileRules)
|
|
a, err := autoParse(c)
|
|
|
|
if err == nil && test.shouldErr {
|
|
t.Fatalf("Test %d expected errors, but got no error", i)
|
|
} else if err != nil && !test.shouldErr {
|
|
t.Fatalf("Test %d expected no errors, but got '%v'", i, err)
|
|
} else if !test.shouldErr {
|
|
if a.loader.directory != test.expectedDirectory {
|
|
t.Fatalf("Test %d expected %v, got %v", i, test.expectedDirectory, a.loader.directory)
|
|
}
|
|
if a.loader.template != test.expectedTempl {
|
|
t.Fatalf("Test %d expected %v, got %v", i, test.expectedTempl, a.loader.template)
|
|
}
|
|
if a.loader.re.String() != test.expectedRe {
|
|
t.Fatalf("Test %d expected %v, got %v", i, test.expectedRe, a.loader.re)
|
|
}
|
|
if test.expectedTo != nil {
|
|
for j, got := range a.loader.transferTo {
|
|
if got != test.expectedTo[j] {
|
|
t.Fatalf("Test %d expected %v, got %v", i, test.expectedTo[j], got)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|