d8714e64e4
* Rename middleware to plugin first pass; mostly used 'sed', few spots where I manually changed text. This still builds a coredns binary. * fmt error * Rename AddMiddleware to AddPlugin * Readd AddMiddleware to remain backwards compat
125 lines
2.5 KiB
Go
125 lines
2.5 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
|
|
no_reload
|
|
}`,
|
|
false, "/tmp", "${1}", `db\.(.*)`, nil,
|
|
},
|
|
{
|
|
`auto {
|
|
directory /tmp (.*) bliep
|
|
}`,
|
|
false, "/tmp", "bliep", `(.*)`, nil,
|
|
},
|
|
{
|
|
`auto {
|
|
directory /tmp (.*) bliep 10
|
|
}`,
|
|
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} aa
|
|
}`,
|
|
true, "", "${1}", ``, nil,
|
|
},
|
|
{
|
|
`auto example.org {
|
|
directory /tmp .* {1}
|
|
}`,
|
|
true, "", "${1}", ``, 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)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|