mirror of
https://github.com/coredns/coredns.git
synced 2025-11-01 10:43:17 -04:00
plugin/transfer: Zone transfer plugin (#3223)
* transfer plugin Signed-off-by: Chris O'Haver <cohaver@infoblox.com>
This commit is contained in:
85
plugin/transfer/setup_test.go
Normal file
85
plugin/transfer/setup_test.go
Normal file
@@ -0,0 +1,85 @@
|
||||
package transfer
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/caddyserver/caddy"
|
||||
)
|
||||
|
||||
func TestParse(t *testing.T) {
|
||||
tests := []struct {
|
||||
input string
|
||||
shouldErr bool
|
||||
exp *Transfer
|
||||
}{
|
||||
{`transfer example.net example.org {
|
||||
to 1.2.3.4 5.6.7.8:1053 [1::2]:34
|
||||
}
|
||||
transfer example.com example.edu {
|
||||
to * 1.2.3.4
|
||||
}`,
|
||||
false,
|
||||
&Transfer{
|
||||
xfrs: []*xfr{{
|
||||
Zones: []string{"example.net.", "example.org."},
|
||||
to: []string{"1.2.3.4:53", "5.6.7.8:1053", "[1::2]:34"},
|
||||
}, {
|
||||
Zones: []string{"example.com.", "example.edu."},
|
||||
to: []string{"*", "1.2.3.4:53"},
|
||||
}},
|
||||
},
|
||||
},
|
||||
// errors
|
||||
{`transfer example.net example.org {
|
||||
}`,
|
||||
true,
|
||||
nil,
|
||||
},
|
||||
{`transfer example.net example.org {
|
||||
invalid option
|
||||
}`,
|
||||
true,
|
||||
nil,
|
||||
},
|
||||
}
|
||||
for i, tc := range tests {
|
||||
c := caddy.NewTestController("dns", tc.input)
|
||||
transfer, err := parse(c)
|
||||
|
||||
if err == nil && tc.shouldErr {
|
||||
t.Fatalf("Test %d expected errors, but got no error", i)
|
||||
}
|
||||
if err != nil && !tc.shouldErr {
|
||||
t.Fatalf("Test %d expected no errors, but got '%v'", i, err)
|
||||
}
|
||||
if tc.shouldErr {
|
||||
continue
|
||||
}
|
||||
|
||||
if len(tc.exp.xfrs) != len(transfer.xfrs) {
|
||||
t.Fatalf("Test %d expected %d xfrs, got %d", i, len(tc.exp.xfrs), len(transfer.xfrs))
|
||||
}
|
||||
for j, x := range transfer.xfrs {
|
||||
// Check Zones
|
||||
if len(tc.exp.xfrs[j].Zones) != len(x.Zones) {
|
||||
t.Fatalf("Test %d expected %d zones, got %d", i, len(tc.exp.xfrs[i].Zones), len(x.Zones))
|
||||
}
|
||||
for k, zone := range x.Zones {
|
||||
if tc.exp.xfrs[j].Zones[k] != zone {
|
||||
t.Errorf("Test %d expected zone %v, got %v", i, tc.exp.xfrs[j].Zones[k], zone)
|
||||
|
||||
}
|
||||
}
|
||||
// Check to
|
||||
if len(tc.exp.xfrs[j].to) != len(x.to) {
|
||||
t.Fatalf("Test %d expected %d 'to' values, got %d", i, len(tc.exp.xfrs[i].to), len(x.to))
|
||||
}
|
||||
for k, to := range x.to {
|
||||
if tc.exp.xfrs[j].to[k] != to {
|
||||
t.Errorf("Test %d expected %v in 'to', got %v", i, tc.exp.xfrs[j].to[k], to)
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user