16504234e5
* plugin/forward: on demand healtchecking Only start doing health checks when we encouner an error (any error). This uses the new pluing/pkg/up package to abstract away the actual checking. This reduces the LOC quite a bit; does need more testing, unit testing and tcpdumping a bit. * fix tests * Fix readme * Use pkg/up for healthchecks * remove unused channel * more cleanups * update readme * * Again do go generate and go build; still referencing the wrong forward repo? Anyway fixed. * Use pkg/up for doing the healtchecks to cut back on unwanted queries * Change up.Func to return an error instead of a boolean. * Drop the string target argument as it doesn't make sense. * Add healthcheck test on failing to get an upstream answer. TODO(miek): double check Forward and Lookup and how they interact with HC, and if we correctly call close() on those * actual test * Tests here * more tests * try getting rid of host * Get rid of the host indirection * Finish removing hosts * moar testing * import fmt * field is not used * docs * move some stuff * bring back health_check * maxfails=0 test * git and merging, bah * review
43 lines
767 B
Go
43 lines
767 B
Go
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)
|
|
}
|
|
}
|