c39e5cd014
* plugin/health: add lameduck mode Add a way to configure lameduck more, i.e. set health to false, stop polling plugins. Then wait for a duration before shutting down. As the health middleware is configured early on in the plugin list, it will hold up all other shutdown, meaning we still answer queries. * Add New * More tests * golint * remove confusing text
45 lines
862 B
Go
45 lines
862 B
Go
package health
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mholt/caddy"
|
|
)
|
|
|
|
func TestSetupHealth(t *testing.T) {
|
|
tests := []struct {
|
|
input string
|
|
shouldErr bool
|
|
}{
|
|
{`health`, false},
|
|
{`health localhost:1234`, false},
|
|
{`health localhost:1234 {
|
|
lameduck 4s
|
|
}`, false},
|
|
{`health bla:a`, false},
|
|
|
|
{`health bla`, true},
|
|
{`health bla bla`, true},
|
|
{`health localhost:1234 {
|
|
lameduck a
|
|
}`, true},
|
|
{`health localhost:1234 {
|
|
lamedudk 4
|
|
} `, true},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
c := caddy.NewTestController("dns", test.input)
|
|
_, _, err := healthParse(c)
|
|
|
|
if test.shouldErr && err == nil {
|
|
t.Errorf("Test %d: Expected error but found none for input %s", i, test.input)
|
|
}
|
|
|
|
if err != nil {
|
|
if !test.shouldErr {
|
|
t.Errorf("Test %d: Expected no error but found one for input %s. Error was: %v", i, test.input, err)
|
|
}
|
|
}
|
|
}
|
|
}
|