TrueCloudLab/coredns
16
0
Fork 2
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

plugin/forward: retry on cached tcp connection closed by peer (#1655)

Browse source 
* plugin/forward: retry on cached tcp connection closed by peer

* fix linter warnings

* fixed unit test

* modify error message
This commit is contained in:
Ruslan Drozhdzh 2018-04-06 15:41:48 +03:00 • committed by Miek Gieben
parent 848a5d7c79
commit e46ee9d9cc
5 changed files with 40 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

26
plugin/forward/persistent_test.go
View file

@ -19,9 +19,13 @@ func TestPersistent(t *testing.T) {
tr := newTransport(s.Addr, nil /* no TLS */)
defer tr.Stop()
c1, _ := tr.Dial("udp")
c2, _ := tr.Dial("udp")
c3, _ := tr.Dial("udp")
c1, cache1, _ := tr.Dial("udp")
c2, cache2, _ := tr.Dial("udp")
c3, cache3, _ := tr.Dial("udp")
if cache1 || cache2 || cache3 {
t.Errorf("Expected non-cached connection")
}
tr.Yield(c1)
tr.Yield(c2)
@ -31,13 +35,23 @@ func TestPersistent(t *testing.T) {
t.Errorf("Expected cache size to be 3, got %d", x)
}
tr.Dial("udp")
c4, cache4, _ := tr.Dial("udp")
if x := tr.Len(); x != 2 {
t.Errorf("Expected cache size to be 2, got %d", x)
}
tr.Dial("udp")
c5, cache5, _ := tr.Dial("udp")
if x := tr.Len(); x != 1 {
t.Errorf("Expected cache size to be 2, got %d", x)
t.Errorf("Expected cache size to be 1, got %d", x)
}
if cache4 == false || cache5 == false {
t.Errorf("Expected cached connection")
}
tr.Yield(c4)
tr.Yield(c5)
if x := tr.Len(); x != 3 {
t.Errorf("Expected cache size to be 3, got %d", x)
}
}