123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285 |
- /*
- Copyright 2016 The Kubernetes 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 replicaset
- import (
- "reflect"
- "testing"
- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
- "k8s.io/apimachinery/pkg/util/validation/field"
- genericapirequest "k8s.io/apiserver/pkg/endpoints/request"
- "k8s.io/apiserver/pkg/registry/rest"
- "k8s.io/kubernetes/pkg/apis/apps"
- api "k8s.io/kubernetes/pkg/apis/core"
- )
- const (
- fakeImageName = "fake-name"
- fakeImage = "fakeimage"
- replicasetName = "test-replicaset"
- namespace = "test-namespace"
- )
- func TestReplicaSetStrategy(t *testing.T) {
- ctx := genericapirequest.NewDefaultContext()
- if !Strategy.NamespaceScoped() {
- t.Errorf("ReplicaSet must be namespace scoped")
- }
- if Strategy.AllowCreateOnUpdate() {
- t.Errorf("ReplicaSet should not allow create on update")
- }
- validSelector := map[string]string{"a": "b"}
- validPodTemplate := api.PodTemplate{
- Template: api.PodTemplateSpec{
- ObjectMeta: metav1.ObjectMeta{
- Labels: validSelector,
- },
- Spec: api.PodSpec{
- RestartPolicy: api.RestartPolicyAlways,
- DNSPolicy: api.DNSClusterFirst,
- Containers: []api.Container{{Name: "abc", Image: "image", ImagePullPolicy: "IfNotPresent", TerminationMessagePolicy: api.TerminationMessageReadFile}},
- },
- },
- }
- rs := &apps.ReplicaSet{
- ObjectMeta: metav1.ObjectMeta{Name: "abc", Namespace: metav1.NamespaceDefault},
- Spec: apps.ReplicaSetSpec{
- Selector: &metav1.LabelSelector{MatchLabels: validSelector},
- Template: validPodTemplate.Template,
- },
- Status: apps.ReplicaSetStatus{
- Replicas: 1,
- ObservedGeneration: int64(10),
- },
- }
- Strategy.PrepareForCreate(ctx, rs)
- if rs.Status.Replicas != 0 {
- t.Error("ReplicaSet should not allow setting status.replicas on create")
- }
- if rs.Status.ObservedGeneration != int64(0) {
- t.Error("ReplicaSet should not allow setting status.observedGeneration on create")
- }
- errs := Strategy.Validate(ctx, rs)
- if len(errs) != 0 {
- t.Errorf("Unexpected error validating %v", errs)
- }
- invalidRc := &apps.ReplicaSet{
- ObjectMeta: metav1.ObjectMeta{Name: "bar", ResourceVersion: "4"},
- }
- Strategy.PrepareForUpdate(ctx, invalidRc, rs)
- errs = Strategy.ValidateUpdate(ctx, invalidRc, rs)
- if len(errs) == 0 {
- t.Errorf("Expected a validation error")
- }
- if invalidRc.ResourceVersion != "4" {
- t.Errorf("Incoming resource version on update should not be mutated")
- }
- }
- func TestReplicaSetStatusStrategy(t *testing.T) {
- ctx := genericapirequest.NewDefaultContext()
- if !StatusStrategy.NamespaceScoped() {
- t.Errorf("ReplicaSet must be namespace scoped")
- }
- if StatusStrategy.AllowCreateOnUpdate() {
- t.Errorf("ReplicaSet should not allow create on update")
- }
- validSelector := map[string]string{"a": "b"}
- validPodTemplate := api.PodTemplate{
- Template: api.PodTemplateSpec{
- ObjectMeta: metav1.ObjectMeta{
- Labels: validSelector,
- },
- Spec: api.PodSpec{
- RestartPolicy: api.RestartPolicyAlways,
- DNSPolicy: api.DNSClusterFirst,
- Containers: []api.Container{{Name: "abc", Image: "image", ImagePullPolicy: "IfNotPresent"}},
- },
- },
- }
- oldRS := &apps.ReplicaSet{
- ObjectMeta: metav1.ObjectMeta{Name: "abc", Namespace: metav1.NamespaceDefault, ResourceVersion: "10"},
- Spec: apps.ReplicaSetSpec{
- Replicas: 3,
- Selector: &metav1.LabelSelector{MatchLabels: validSelector},
- Template: validPodTemplate.Template,
- },
- Status: apps.ReplicaSetStatus{
- Replicas: 1,
- ObservedGeneration: int64(10),
- },
- }
- newRS := &apps.ReplicaSet{
- ObjectMeta: metav1.ObjectMeta{Name: "abc", Namespace: metav1.NamespaceDefault, ResourceVersion: "9"},
- Spec: apps.ReplicaSetSpec{
- Replicas: 1,
- Selector: &metav1.LabelSelector{MatchLabels: validSelector},
- Template: validPodTemplate.Template,
- },
- Status: apps.ReplicaSetStatus{
- Replicas: 3,
- ObservedGeneration: int64(11),
- },
- }
- StatusStrategy.PrepareForUpdate(ctx, newRS, oldRS)
- if newRS.Status.Replicas != 3 {
- t.Errorf("ReplicaSet status updates should allow change of replicas: %v", newRS.Status.Replicas)
- }
- if newRS.Spec.Replicas != 3 {
- t.Errorf("PrepareForUpdate should have preferred spec")
- }
- errs := StatusStrategy.ValidateUpdate(ctx, newRS, oldRS)
- if len(errs) != 0 {
- t.Errorf("Unexpected error %v", errs)
- }
- }
- func TestSelectorImmutability(t *testing.T) {
- tests := []struct {
- requestInfo genericapirequest.RequestInfo
- oldSelectorLabels map[string]string
- newSelectorLabels map[string]string
- expectedErrorList field.ErrorList
- }{
- {
- genericapirequest.RequestInfo{
- APIGroup: "apps",
- APIVersion: "v1beta2",
- Resource: "replicasets",
- },
- map[string]string{"a": "b"},
- map[string]string{"c": "d"},
- field.ErrorList{
- &field.Error{
- Type: field.ErrorTypeInvalid,
- Field: field.NewPath("spec").Child("selector").String(),
- BadValue: &metav1.LabelSelector{
- MatchLabels: map[string]string{"c": "d"},
- MatchExpressions: []metav1.LabelSelectorRequirement{},
- },
- Detail: "field is immutable",
- },
- },
- },
- {
- genericapirequest.RequestInfo{
- APIGroup: "extensions",
- APIVersion: "v1beta1",
- Resource: "replicasets",
- },
- map[string]string{"a": "b"},
- map[string]string{"c": "d"},
- field.ErrorList{},
- },
- }
- for _, test := range tests {
- oldReplicaSet := newReplicaSetWithSelectorLabels(test.oldSelectorLabels)
- newReplicaSet := newReplicaSetWithSelectorLabels(test.newSelectorLabels)
- context := genericapirequest.NewContext()
- context = genericapirequest.WithRequestInfo(context, &test.requestInfo)
- errorList := rsStrategy{}.ValidateUpdate(context, newReplicaSet, oldReplicaSet)
- if !reflect.DeepEqual(test.expectedErrorList, errorList) {
- t.Errorf("Unexpected error list, expected: %v, actual: %v", test.expectedErrorList, errorList)
- }
- }
- }
- func newReplicaSetWithSelectorLabels(selectorLabels map[string]string) *apps.ReplicaSet {
- return &apps.ReplicaSet{
- ObjectMeta: metav1.ObjectMeta{
- Name: replicasetName,
- Namespace: namespace,
- ResourceVersion: "1",
- },
- Spec: apps.ReplicaSetSpec{
- Selector: &metav1.LabelSelector{
- MatchLabels: selectorLabels,
- MatchExpressions: []metav1.LabelSelectorRequirement{},
- },
- Template: api.PodTemplateSpec{
- ObjectMeta: metav1.ObjectMeta{
- Labels: selectorLabels,
- },
- Spec: api.PodSpec{
- RestartPolicy: api.RestartPolicyAlways,
- DNSPolicy: api.DNSClusterFirst,
- Containers: []api.Container{{Name: fakeImageName, Image: fakeImage, ImagePullPolicy: "IfNotPresent", TerminationMessagePolicy: api.TerminationMessageReadFile}},
- },
- },
- },
- }
- }
- func TestReplicasetDefaultGarbageCollectionPolicy(t *testing.T) {
- // Make sure we correctly implement the interface.
- // Otherwise a typo could silently change the default.
- var gcds rest.GarbageCollectionDeleteStrategy = Strategy
- tests := []struct {
- requestInfo genericapirequest.RequestInfo
- expectedGCPolicy rest.GarbageCollectionPolicy
- isNilRequestInfo bool
- }{
- {
- genericapirequest.RequestInfo{
- APIGroup: "extensions",
- APIVersion: "v1beta1",
- Resource: "replicasets",
- },
- rest.OrphanDependents,
- false,
- },
- {
- genericapirequest.RequestInfo{
- APIGroup: "apps",
- APIVersion: "v1beta2",
- Resource: "replicasets",
- },
- rest.OrphanDependents,
- false,
- },
- {
- genericapirequest.RequestInfo{
- APIGroup: "apps",
- APIVersion: "v1",
- Resource: "replicasets",
- },
- rest.DeleteDependents,
- false,
- },
- {
- expectedGCPolicy: rest.DeleteDependents,
- isNilRequestInfo: true,
- },
- }
- for _, test := range tests {
- context := genericapirequest.NewContext()
- if !test.isNilRequestInfo {
- context = genericapirequest.WithRequestInfo(context, &test.requestInfo)
- }
- if got, want := gcds.DefaultGarbageCollectionPolicy(context), test.expectedGCPolicy; got != want {
- t.Errorf("%s/%s: DefaultGarbageCollectionPolicy() = %#v, want %#v", test.requestInfo.APIGroup,
- test.requestInfo.APIVersion, got, want)
- }
- }
- }
|