mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-11-14 02:10:21 +00:00
86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
|
/*
|
||
|
Copyright 2023 ceph-csi authors.
|
||
|
|
||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
you may not use this file except in compliance with the License.
|
||
|
You may obtain a copy of the License at
|
||
|
|
||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
||
|
Unless required by applicable law or agreed to in writing, software
|
||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
See the License for the specific language governing permissions and
|
||
|
limitations under the License.
|
||
|
*/
|
||
|
|
||
|
package healthchecker
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestManager(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
|
||
|
volumeID := "fake-volume-id"
|
||
|
volumePath := t.TempDir()
|
||
|
mgr := NewHealthCheckManager()
|
||
|
|
||
|
// expected to have an error in msg
|
||
|
healthy, msg := mgr.IsHealthy(volumeID, volumePath)
|
||
|
if !(healthy && msg != nil) {
|
||
|
t.Error("ConditionChecker was not started yet, did not get an error")
|
||
|
}
|
||
|
|
||
|
t.Log("start the checker")
|
||
|
err := mgr.StartChecker(volumeID, volumePath, StatCheckerType)
|
||
|
if err != nil {
|
||
|
t.Fatalf("ConditionChecker could not get started: %v", err)
|
||
|
}
|
||
|
|
||
|
t.Log("check health, should be healthy")
|
||
|
healthy, msg = mgr.IsHealthy(volumeID, volumePath)
|
||
|
if !healthy || err != nil {
|
||
|
t.Errorf("volume is unhealthy: %s", msg)
|
||
|
}
|
||
|
|
||
|
t.Log("stop the checker")
|
||
|
mgr.StopChecker(volumeID, volumePath)
|
||
|
}
|
||
|
|
||
|
func TestSharedChecker(t *testing.T) {
|
||
|
t.Parallel()
|
||
|
|
||
|
volumeID := "fake-volume-id"
|
||
|
volumePath := t.TempDir()
|
||
|
mgr := NewHealthCheckManager()
|
||
|
|
||
|
// expected to have an error in msg
|
||
|
healthy, msg := mgr.IsHealthy(volumeID, volumePath)
|
||
|
if !(healthy && msg != nil) {
|
||
|
t.Error("ConditionChecker was not started yet, did not get an error")
|
||
|
}
|
||
|
|
||
|
t.Log("start the checker")
|
||
|
err := mgr.StartSharedChecker(volumeID, volumePath, StatCheckerType)
|
||
|
if err != nil {
|
||
|
t.Fatalf("ConditionChecker could not get started: %v", err)
|
||
|
}
|
||
|
|
||
|
t.Log("check health, should be healthy")
|
||
|
healthy, msg = mgr.IsHealthy(volumeID, volumePath)
|
||
|
if !healthy || err != nil {
|
||
|
t.Errorf("volume is unhealthy: %s", msg)
|
||
|
}
|
||
|
|
||
|
t.Log("check health, should be healthy, path is ignored")
|
||
|
healthy, msg = mgr.IsHealthy(volumeID, "different-path")
|
||
|
if !healthy || err != nil {
|
||
|
t.Errorf("volume is unhealthy: %s", msg)
|
||
|
}
|
||
|
|
||
|
t.Log("stop the checker")
|
||
|
mgr.StopSharedChecker(volumeID)
|
||
|
}
|