blob: 5fa491a01e221e185d80e712a3381bf8f75e9f36 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
package forward
import (
"testing"
"github.com/coredns/coredns/plugin/pkg/dnstest"
"github.com/miekg/dns"
)
func TestPersistent(t *testing.T) {
s := dnstest.NewServer(func(w dns.ResponseWriter, r *dns.Msg) {
ret := new(dns.Msg)
ret.SetReply(r)
w.WriteMsg(ret)
})
defer s.Close()
tr := newTransport(s.Addr, nil /* no TLS */)
defer tr.Stop()
c1, _ := tr.Dial("udp")
c2, _ := tr.Dial("udp")
c3, _ := tr.Dial("udp")
tr.Yield(c1)
tr.Yield(c2)
tr.Yield(c3)
if x := tr.Len(); x != 3 {
t.Errorf("Expected cache size to be 3, got %d", x)
}
tr.Dial("udp")
if x := tr.Len(); x != 2 {
t.Errorf("Expected cache size to be 2, got %d", x)
}
tr.Dial("udp")
if x := tr.Len(); x != 1 {
t.Errorf("Expected cache size to be 2, got %d", x)
}
}
|