Switch exec to look at exit code not output status.
This commit is contained in:
@@ -41,16 +41,40 @@ type healthCheckTest struct {
|
||||
err error
|
||||
}
|
||||
|
||||
type fakeExitError struct {
|
||||
exited bool
|
||||
statusCode int
|
||||
}
|
||||
|
||||
func (f *fakeExitError) String() string {
|
||||
return f.Error()
|
||||
}
|
||||
|
||||
func (f *fakeExitError) Error() string {
|
||||
return "fake exit"
|
||||
}
|
||||
|
||||
func (f *fakeExitError) Exited() bool {
|
||||
return f.exited
|
||||
}
|
||||
|
||||
func (f *fakeExitError) ExitStatus() int {
|
||||
return f.statusCode
|
||||
}
|
||||
|
||||
func TestExec(t *testing.T) {
|
||||
prober := New()
|
||||
fake := FakeCmd{}
|
||||
|
||||
tests := []healthCheckTest{
|
||||
// Ok
|
||||
{probe.Success, false, []byte("OK"), nil},
|
||||
// Ok
|
||||
{probe.Success, false, []byte("OK"), &fakeExitError{true, 0}},
|
||||
// Run returns error
|
||||
{probe.Unknown, true, []byte("OK, NOT"), fmt.Errorf("test error")},
|
||||
// Unhealthy
|
||||
{probe.Failure, false, []byte("Fail"), nil},
|
||||
{probe.Failure, false, []byte("Fail"), &fakeExitError{true, 1}},
|
||||
}
|
||||
for _, test := range tests {
|
||||
fake.out = test.output
|
||||
|
Reference in New Issue
Block a user