diff options
Diffstat (limited to 'plugin/forward/setup_test.go')
| -rw-r--r-- | plugin/forward/setup_test.go | 42 |
1 files changed, 22 insertions, 20 deletions
diff --git a/plugin/forward/setup_test.go b/plugin/forward/setup_test.go index fba2359b9..a8140d410 100644 --- a/plugin/forward/setup_test.go +++ b/plugin/forward/setup_test.go @@ -10,28 +10,30 @@ import ( func TestSetup(t *testing.T) { tests := []struct { - input string - shouldErr bool - expectedFrom string - expectedIgnored []string - expectedFails uint32 - expectedForceTCP bool - expectedErr string + input string + shouldErr bool + expectedFrom string + expectedIgnored []string + expectedFails uint32 + expectedOpts options + expectedErr string }{ // positive - {"forward . 127.0.0.1", false, ".", nil, 2, false, ""}, - {"forward . 127.0.0.1 {\nexcept miek.nl\n}\n", false, ".", nil, 2, false, ""}, - {"forward . 127.0.0.1 {\nmax_fails 3\n}\n", false, ".", nil, 3, false, ""}, - {"forward . 127.0.0.1 {\nforce_tcp\n}\n", false, ".", nil, 2, true, ""}, - {"forward . 127.0.0.1:53", false, ".", nil, 2, false, ""}, - {"forward . 127.0.0.1:8080", false, ".", nil, 2, false, ""}, - {"forward . [::1]:53", false, ".", nil, 2, false, ""}, - {"forward . [2003::1]:53", false, ".", nil, 2, false, ""}, + {"forward . 127.0.0.1", false, ".", nil, 2, options{}, ""}, + {"forward . 127.0.0.1 {\nexcept miek.nl\n}\n", false, ".", nil, 2, options{}, ""}, + {"forward . 127.0.0.1 {\nmax_fails 3\n}\n", false, ".", nil, 3, options{}, ""}, + {"forward . 127.0.0.1 {\nforce_tcp\n}\n", false, ".", nil, 2, options{forceTCP: true}, ""}, + {"forward . 127.0.0.1 {\nprefer_udp\n}\n", false, ".", nil, 2, options{preferUDP: true}, ""}, + {"forward . 127.0.0.1 {\nforce_tcp\nprefer_udp\n}\n", false, ".", nil, 2, options{preferUDP: true, forceTCP: true}, ""}, + {"forward . 127.0.0.1:53", false, ".", nil, 2, options{}, ""}, + {"forward . 127.0.0.1:8080", false, ".", nil, 2, options{}, ""}, + {"forward . [::1]:53", false, ".", nil, 2, options{}, ""}, + {"forward . [2003::1]:53", false, ".", nil, 2, options{}, ""}, // negative - {"forward . a27.0.0.1", true, "", nil, 0, false, "not an IP"}, - {"forward . 127.0.0.1 {\nblaatl\n}\n", true, "", nil, 0, false, "unknown property"}, + {"forward . a27.0.0.1", true, "", nil, 0, options{}, "not an IP"}, + {"forward . 127.0.0.1 {\nblaatl\n}\n", true, "", nil, 0, options{}, "unknown property"}, {`forward . ::1 - forward com ::2`, true, "", nil, 0, false, "plugin"}, + forward com ::2`, true, "", nil, 0, options{}, "plugin"}, } for i, test := range tests { @@ -63,8 +65,8 @@ func TestSetup(t *testing.T) { if !test.shouldErr && f.maxfails != test.expectedFails { t.Errorf("Test %d: expected: %d, got: %d", i, test.expectedFails, f.maxfails) } - if !test.shouldErr && f.forceTCP != test.expectedForceTCP { - t.Errorf("Test %d: expected: %t, got: %t", i, test.expectedForceTCP, f.forceTCP) + if !test.shouldErr && f.opts != test.expectedOpts { + t.Errorf("Test %d: expected: %v, got: %v", i, test.expectedOpts, f.opts) } } } |