mirror of
				https://github.com/coredns/coredns.git
				synced 2025-11-03 18:53:13 -05:00 
			
		
		
		
	* 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
		
			
				
	
	
		
			46 lines
		
	
	
		
			862 B
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			46 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)
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 |