forked from rook/rook
subhamkrai
f9fafe62d4
this commit will enable one more linter in golangci-lint. Signed-off-by: subhamkrai <subhamkumarrai03@gmail.com>
92 lines
2.9 KiB
Go
92 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"
|
|
|
|
e "github.com/pkg/errors"
|
|
optest "github.com/rook/rook/pkg/operator/test"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func checkLabel(key, value string, labels map[string]string) error {
|
|
v, ok := labels[key]
|
|
if !ok {
|
|
return e.Errorf("label not present: expected={%s: %s}", key, value)
|
|
}
|
|
if v != value {
|
|
return e.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 e.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, daemonID, appName, namespace string,
|
|
) {
|
|
optest.AssertLabelsContainRookRequirements(t, labels, appName)
|
|
|
|
resourceLabels := []string{}
|
|
for k, v := range labels {
|
|
resourceLabels = append(resourceLabels, fmt.Sprintf("%s=%s", k, v))
|
|
}
|
|
expectedLabels := []string{
|
|
"ceph_daemon_id=" + daemonID,
|
|
string(daemonType) + "=" + daemonID,
|
|
"rook_cluster" + "=" + namespace,
|
|
}
|
|
assert.Subset(t, resourceLabels, expectedLabels,
|
|
"labels on resource do not match Ceph requirements", labels)
|
|
}
|