diff options
Diffstat (limited to 'middleware/autopath/setup_test.go')
| -rw-r--r-- | middleware/autopath/setup_test.go | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/middleware/autopath/setup_test.go b/middleware/autopath/setup_test.go index 0696a9179..b31110ddb 100644 --- a/middleware/autopath/setup_test.go +++ b/middleware/autopath/setup_test.go @@ -21,21 +21,18 @@ func TestSetupAutoPath(t *testing.T) { tests := []struct { input string shouldErr bool + expectedZone string expectedMw string // expected middleware. expectedSearch []string // expected search path expectedErrContent string // substring from the expected error. Empty for positive cases. }{ // positive - { - `autopath @kubernetes`, false, "kubernetes", nil, "", - }, - { - `autopath ` + resolv, false, "", []string{"bar.com.", "baz.com.", ""}, "", - }, + {`autopath @kubernetes`, false, "", "kubernetes", nil, ""}, + {`autopath example.org @kubernetes`, false, "example.org.", "kubernetes", nil, ""}, + {`autopath 10.0.0.0/8 @kubernetes`, false, "10.in-addr.arpa.", "kubernetes", nil, ""}, + {`autopath ` + resolv, false, "", "", []string{"bar.com.", "baz.com.", ""}, ""}, // negative - { - `autopath kubernetes`, true, "", nil, "open kubernetes: no such file or directory", - }, + {`autopath kubernetes`, true, "", "", nil, "open kubernetes: no such file or directory"}, } for i, test := range tests { @@ -64,6 +61,11 @@ func TestSetupAutoPath(t *testing.T) { t.Errorf("Test %d, wrong searchpath for input %s. Expected: '%v', actual: '%v'", i, test.input, test.expectedSearch, ap.search) } } + if !test.shouldErr && test.expectedZone != "" { + if test.expectedZone != ap.Zones[0] { + t.Errorf("Test %d, expected zone %q for input %s, got: %q", i, test.expectedZone, test.input, ap.Zones[0]) + } + } } } |