forked from TrueCloudLab/distribution
a1b49d3d17
Allow health checkers to abort if the request context is canceled. Modify the checkers to respect context cancelation and return wrapped errors so the caller of CheckStatus() would be able to discriminate true failed checks from checks which were aborted because the context became done. Signed-off-by: Cory Snider <csnider@mirantis.com>
26 lines
765 B
Go
26 lines
765 B
Go
package checks
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestFileChecker(t *testing.T) {
|
|
if err := FileChecker("/tmp").Check(context.Background()); err == nil {
|
|
t.Errorf("/tmp was expected as exists")
|
|
}
|
|
|
|
if err := FileChecker("NoSuchFileFromMoon").Check(context.Background()); err != nil {
|
|
t.Errorf("NoSuchFileFromMoon was expected as not exists, error:%v", err)
|
|
}
|
|
}
|
|
|
|
func TestHTTPChecker(t *testing.T) {
|
|
if err := HTTPChecker("https://www.google.cybertron", 200, 0, nil).Check(context.Background()); err == nil {
|
|
t.Errorf("Google on Cybertron was expected as not exists")
|
|
}
|
|
|
|
if err := HTTPChecker("https://www.google.pt", 200, 0, nil).Check(context.Background()); err != nil {
|
|
t.Errorf("Google at Portugal was expected as exists, error:%v", err)
|
|
}
|
|
}
|