diff options
| author |  Ruslan Drozhdzh <30860269+rdrozhdzh@users.noreply.github.com>
| 2018-07-07 10:14:21 +0300 |
|---|---|---|
| committer |  Miek Gieben <miek@miek.nl>
| 2018-07-07 08:14:21 +0100 |
| commit | bc5090123487653ce502d5801007d03dbb6a2ba7 (patch) | |
| tree | 942ec6a99c31c2c6b477f7027da108aeafc5c857 /plugin/forward/setup_test.go | |
| parent | 7c41f2ce9f75a20d9bba3ab70b4cb8a133a0756f (diff) | |
| download | coredns-bc5090123487653ce502d5801007d03dbb6a2ba7.tar.gz coredns-bc5090123487653ce502d5801007d03dbb6a2ba7.tar.zst coredns-bc5090123487653ce502d5801007d03dbb6a2ba7.zip | |
plugin/forward: add prefer_udp option (#1944)
* plugin/forward: add prefer_udp option
* updated according to code review
- fixed linter warning
- removed metric parameter in Proxy.Connect()
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) } } } |