26d1432ae6
* Update all plugins to use plugin/pkg/log I wish this could have been done with sed. Alas manually changed all callers to use the new plugin/pkg/log package. * Error -> Info * Add docs to debug plugin as well
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package proxy
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/coredns/coredns/plugin/test"
|
|
"github.com/coredns/coredns/request"
|
|
|
|
"github.com/mholt/caddy/caddyfile"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func TestUnhealthy(t *testing.T) {
|
|
// High HC interval, we want to test the HC after failed queries.
|
|
config := "proxy . %s {\n health_check /healthcheck:%s 10s \nfail_timeout 100ms\n}"
|
|
|
|
backend := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
r.Body.Close()
|
|
w.Write([]byte("OK"))
|
|
}))
|
|
defer backend.Close()
|
|
|
|
port := backend.URL[17:] // Remove all crap up to the port
|
|
back := backend.URL[7:] // Remove http://
|
|
|
|
c := caddyfile.NewDispenser("testfile", strings.NewReader(fmt.Sprintf(config, back, port)))
|
|
upstreams, err := NewStaticUpstreams(&c)
|
|
if err != nil {
|
|
t.Errorf("Expected no error. Got: %s", err)
|
|
}
|
|
p := &Proxy{Upstreams: &upstreams}
|
|
m := new(dns.Msg)
|
|
m.SetQuestion("example.org.", dns.TypeA)
|
|
state := request.Request{W: &test.ResponseWriter{}, Req: m}
|
|
|
|
// Should all fail.
|
|
for j := 0; j < failureCheck; j++ {
|
|
if _, err := p.Forward(state); err == nil {
|
|
t.Errorf("Expected error. Got: nil")
|
|
}
|
|
}
|
|
|
|
fails := atomic.LoadInt32(&upstreams[0].(*staticUpstream).Hosts[0].Fails)
|
|
if fails != 3 {
|
|
t.Errorf("Expected %d fails, got %d", 3, fails)
|
|
}
|
|
// HC should be kicked off, and reset the counter to 0
|
|
i := 0
|
|
for fails != 0 {
|
|
fails = atomic.LoadInt32(&upstreams[0].(*staticUpstream).Hosts[0].Fails)
|
|
time.Sleep(100 * time.Microsecond)
|
|
i++
|
|
}
|
|
}
|