aboutsummaryrefslogtreecommitdiff
path: root/plugin/forward/forward_test.go
diff options
context:
space:
mode:
authorGravatar Chris O'Haver <cohaver@infoblox.com> 2020-05-29 12:30:26 -0400
committerGravatar GitHub <noreply@github.com> 2020-05-29 18:30:26 +0200
commit54fb2112ac7616db93d5b9e505079d821f15ec2d (patch)
tree25ad6561a7c42881e620db0e7bd2497fb562476d /plugin/forward/forward_test.go
parentbc2ba28865248291f0a4492979f960eb00430ae2 (diff)
downloadcoredns-54fb2112ac7616db93d5b9e505079d821f15ec2d.tar.gz
coredns-54fb2112ac7616db93d5b9e505079d821f15ec2d.tar.zst
coredns-54fb2112ac7616db93d5b9e505079d821f15ec2d.zip
plugin/forward/grpc: Revert forward/grpc policy dedup (#3919)
* revert de-dup Signed-off-by: Chris O'Haver <cohaver@infoblox.com> * unit test Signed-off-by: Chris O'Haver <cohaver@infoblox.com> * use roundrobin policy in test Signed-off-by: Chris O'Haver <cohaver@infoblox.com>
Diffstat (limited to 'plugin/forward/forward_test.go')
-rw-r--r--plugin/forward/forward_test.go24
1 files changed, 24 insertions, 0 deletions
diff --git a/plugin/forward/forward_test.go b/plugin/forward/forward_test.go
new file mode 100644
index 000000000..b0ef47ba9
--- /dev/null
+++ b/plugin/forward/forward_test.go
@@ -0,0 +1,24 @@
+package forward
+
+import (
+ "testing"
+)
+
+func TestList(t *testing.T) {
+ f := Forward{
+ proxies: []*Proxy{{addr: "1.1.1.1:53"}, {addr: "2.2.2.2:53"}, {addr: "3.3.3.3:53"}},
+ p: &roundRobin{},
+ }
+
+ expect := []*Proxy{{addr: "2.2.2.2:53"}, {addr: "1.1.1.1:53"}, {addr: "3.3.3.3:53"}}
+ got := f.List()
+
+ if len(got) != len(expect) {
+ t.Fatalf("Expected: %v results, got: %v", len(expect), len(got))
+ }
+ for i, p := range got {
+ if p.addr != expect[i].addr {
+ t.Fatalf("Expected proxy %v to be '%v', got: '%v'", i, expect[i].addr, p.addr)
+ }
+ }
+}