forked from rook/rook
Blaine Gardner
03ea5f0a7c
Configure the Ceph mgr daemon completely from the operator a la the recent changes to the Ceph mon operator. The pod spec for the mgr changed quite a bit, and instead of updating the unit tests of questionable use, some additional unit test tools applicable to any Ceph daemon have been added and used with the mgr. The mgrs unit tests should now be more useful and get in the way of devs less, and they can be used by other daemons later. Signed-off-by: Blaine Gardner <blaine.gardner@suse.com>
94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
/*
|
|
Copyright 2018 The Rook Authors. All rights reserved.
|
|
|
|
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 test
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/rook/rook/pkg/operator/ceph/config"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const (
|
|
labelMismatchFormat = "label mismatch: expected={%s: %s} present={%s: %s}"
|
|
)
|
|
|
|
func checkLabel(key, value string, labels map[string]string) error {
|
|
v, ok := labels[key]
|
|
if !ok {
|
|
return fmt.Errorf("label not present: expected={%s: %s}", key, value)
|
|
}
|
|
if v != value {
|
|
return fmt.Errorf("label mismatch: expected={%s: %s} present={%s: %s}", key, value, key, v)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func combineErrors(errors ...error) error {
|
|
errText := ""
|
|
failure := false
|
|
for _, e := range errors {
|
|
if e != nil {
|
|
failure = true
|
|
errText = fmt.Sprintf("%v: %s", e, errText) // Will result in string ending in ": "
|
|
}
|
|
}
|
|
if failure {
|
|
errText = strings.TrimRight(errText, ": ") // Remove ": " from end
|
|
return fmt.Errorf("%s", errText)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// VerifyAppLabels returns a descriptive error if app labels are not present or not as expected.
|
|
func VerifyAppLabels(appName, namespace string, labels map[string]string) error {
|
|
errA := checkLabel("app", appName, labels)
|
|
errB := checkLabel("rook_cluster", namespace, labels)
|
|
return combineErrors(errA, errB)
|
|
}
|
|
|
|
// VerifyPodLabels returns a descriptive error if pod labels are not present or not as expected.
|
|
func VerifyPodLabels(appName, namespace, daemonType, daemonID string, labels map[string]string) error {
|
|
errA := VerifyAppLabels(appName, namespace, labels)
|
|
errB := checkLabel("ceph_daemon_id", daemonID, labels)
|
|
errC := checkLabel(daemonType, daemonID, labels)
|
|
return combineErrors(errA, errB, errC)
|
|
}
|
|
|
|
// AssertLabelsContainCephRequirements asserts that the the labels under test contain the labels
|
|
// which all Ceph pods should have. This can be used with labels for Kubernetes Deployments,
|
|
// DaemonSets, etc.
|
|
func AssertLabelsContainCephRequirements(
|
|
t *testing.T, labels map[string]string,
|
|
daemonType config.DaemonType, daemonID, appName, namespace string,
|
|
) {
|
|
resourceLabels := []string{}
|
|
for k, v := range labels {
|
|
resourceLabels = append(resourceLabels, fmt.Sprintf("%s=%s", k, v))
|
|
}
|
|
expectedLabels := []string{
|
|
"app=" + appName,
|
|
"ceph_daemon_id=" + daemonID,
|
|
string(daemonType) + "=" + daemonID,
|
|
"rook_cluster" + "=" + namespace,
|
|
}
|
|
assert.Subset(t, resourceLabels, expectedLabels,
|
|
"labels on resource do not match Ceph requirements", labels)
|
|
}
|