Chore: add labels tweak function

This commit is contained in:
Qirui 2023-03-20 13:06:56 +08:00
parent ec34891782
commit 5e0161b3de

View File

@ -74,6 +74,15 @@ func tweakNamespace(ns string) statefulSetTweak {
} }
} }
func tweakLabels(key string, value string) statefulSetTweak {
return func(ss *apps.StatefulSet) {
if ss.ObjectMeta.Labels == nil {
ss.ObjectMeta.Labels = map[string]string{}
}
ss.ObjectMeta.Labels[key] = value
}
}
func tweakAnnotations(key string, value string) statefulSetTweak { func tweakAnnotations(key string, value string) statefulSetTweak {
return func(ss *apps.StatefulSet) { return func(ss *apps.StatefulSet) {
if ss.ObjectMeta.Annotations == nil { if ss.ObjectMeta.Annotations == nil {
@ -315,21 +324,9 @@ func TestValidateStatefulSet(t *testing.T) {
}, },
{ {
name: "invalid_label", name: "invalid_label",
set: apps.StatefulSet{ set: mkStatefulSet(&validPodTemplate,
ObjectMeta: metav1.ObjectMeta{ tweakLabels("NoUppercaseOrSpecialCharsLike=Equals", "bar"),
Name: "abc-123", ),
Namespace: metav1.NamespaceDefault,
Labels: map[string]string{
"NoUppercaseOrSpecialCharsLike=Equals": "bar",
},
},
Spec: apps.StatefulSetSpec{
PodManagementPolicy: apps.OrderedReadyPodManagement,
Selector: &metav1.LabelSelector{MatchLabels: validLabels},
Template: validPodTemplate.Template,
UpdateStrategy: apps.StatefulSetUpdateStrategy{Type: apps.RollingUpdateStatefulSetStrategyType},
},
},
errs: field.ErrorList{ errs: field.ErrorList{
field.Invalid(field.NewPath("metadata", "labels"), nil, ""), field.Invalid(field.NewPath("metadata", "labels"), nil, ""),
}, },