diff --git a/test/e2e/common/BUILD b/test/e2e/common/BUILD index 68dba97fb2c..930903e7a65 100644 --- a/test/e2e/common/BUILD +++ b/test/e2e/common/BUILD @@ -24,6 +24,7 @@ go_library( "init_container.go", "kubelet.go", "kubelet_etc_hosts.go", + "lease.go", "lifecycle_hook.go", "networking.go", "node_lease.go", @@ -68,6 +69,7 @@ go_library( "//staging/src/k8s.io/apimachinery/pkg/util/diff:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/util/intstr:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/util/sets:go_default_library", + "//staging/src/k8s.io/apimachinery/pkg/util/strategicpatch:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/util/uuid:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/util/wait:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/watch:go_default_library", diff --git a/test/e2e/common/lease.go b/test/e2e/common/lease.go new file mode 100644 index 00000000000..b0b98a5cf01 --- /dev/null +++ b/test/e2e/common/lease.go @@ -0,0 +1,149 @@ +/* +Copyright 2019 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 common + +import ( + "encoding/json" + "fmt" + "time" + + coordinationv1 "k8s.io/api/coordination/v1" + apiequality "k8s.io/apimachinery/pkg/api/equality" + "k8s.io/apimachinery/pkg/api/errors" + metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" + "k8s.io/apimachinery/pkg/labels" + "k8s.io/apimachinery/pkg/types" + "k8s.io/apimachinery/pkg/util/strategicpatch" + "k8s.io/kubernetes/test/e2e/framework" + "k8s.io/utils/pointer" + + "github.com/onsi/ginkgo" +) + +func getPatchBytes(oldLease, newLease *coordinationv1.Lease) ([]byte, error) { + oldData, err := json.Marshal(oldLease) + if err != nil { + return nil, fmt.Errorf("failed to Marshal oldData: %v", err) + } + newData, err := json.Marshal(newLease) + if err != nil { + return nil, fmt.Errorf("failed to Marshal newData: %v", err) + } + patchBytes, err := strategicpatch.CreateTwoWayMergePatch(oldData, newData, coordinationv1.Lease{}) + if err != nil { + return nil, fmt.Errorf("failed to CreateTwoWayMergePatch: %v", err) + } + return patchBytes, nil +} + +var _ = framework.KubeDescribe("Lease", func() { + f := framework.NewDefaultFramework("lease-test") + + // TODO(wojtek-t): Promote to Conformance once proved to be stable. + ginkgo.It("API should be available", func() { + leaseClient := f.ClientSet.CoordinationV1().Leases(f.Namespace.Name) + + name := "lease" + lease := &coordinationv1.Lease{ + ObjectMeta: metav1.ObjectMeta{ + Name: name, + }, + Spec: coordinationv1.LeaseSpec{ + HolderIdentity: pointer.StringPtr("holder"), + LeaseDurationSeconds: pointer.Int32Ptr(30), + AcquireTime: &metav1.MicroTime{Time: time.Time{}.Add(2 * time.Second)}, + RenewTime: &metav1.MicroTime{Time: time.Time{}.Add(5 * time.Second)}, + LeaseTransitions: pointer.Int32Ptr(0), + }, + } + + createdLease, err := leaseClient.Create(lease) + framework.ExpectNoError(err, "creating Lease failed") + + readLease, err := leaseClient.Get(name, metav1.GetOptions{}) + framework.ExpectNoError(err, "couldn't read Lease") + framework.ExpectEqual(apiequality.Semantic.DeepEqual(lease.Spec, readLease.Spec), true) + + createdLease.Spec = coordinationv1.LeaseSpec{ + HolderIdentity: pointer.StringPtr("holder2"), + LeaseDurationSeconds: pointer.Int32Ptr(30), + AcquireTime: &metav1.MicroTime{Time: time.Time{}.Add(20 * time.Second)}, + RenewTime: &metav1.MicroTime{Time: time.Time{}.Add(50 * time.Second)}, + LeaseTransitions: pointer.Int32Ptr(1), + } + + _, err = leaseClient.Update(createdLease) + framework.ExpectNoError(err, "updating Lease failed") + + readLease, err = leaseClient.Get(name, metav1.GetOptions{}) + framework.ExpectNoError(err, "couldn't read Lease") + framework.ExpectEqual(apiequality.Semantic.DeepEqual(createdLease.Spec, readLease.Spec), true) + + patchedLease := readLease.DeepCopy() + patchedLease.Spec = coordinationv1.LeaseSpec{ + HolderIdentity: pointer.StringPtr("holder3"), + LeaseDurationSeconds: pointer.Int32Ptr(60), + AcquireTime: &metav1.MicroTime{Time: time.Time{}.Add(50 * time.Second)}, + RenewTime: &metav1.MicroTime{Time: time.Time{}.Add(70 * time.Second)}, + LeaseTransitions: pointer.Int32Ptr(2), + } + patchBytes, err := getPatchBytes(readLease, patchedLease) + framework.ExpectNoError(err, "creating patch failed") + + _, err = leaseClient.Patch(name, types.StrategicMergePatchType, patchBytes) + framework.ExpectNoError(err, "patching Lease failed") + + readLease, err = leaseClient.Get(name, metav1.GetOptions{}) + framework.ExpectNoError(err, "couldn't read Lease") + framework.ExpectEqual(apiequality.Semantic.DeepEqual(patchedLease.Spec, readLease.Spec), true) + + name2 := "lease2" + lease2 := &coordinationv1.Lease{ + ObjectMeta: metav1.ObjectMeta{ + Name: name2, + Labels: map[string]string{"deletecollection": "true"}, + }, + Spec: coordinationv1.LeaseSpec{ + HolderIdentity: pointer.StringPtr("holder"), + LeaseDurationSeconds: pointer.Int32Ptr(30), + AcquireTime: &metav1.MicroTime{Time: time.Time{}.Add(2 * time.Second)}, + RenewTime: &metav1.MicroTime{Time: time.Time{}.Add(5 * time.Second)}, + LeaseTransitions: pointer.Int32Ptr(0), + }, + } + _, err = leaseClient.Create(lease2) + framework.ExpectNoError(err, "creating Lease failed") + + leases, err := leaseClient.List(metav1.ListOptions{}) + framework.ExpectNoError(err, "couldn't list Leases") + framework.ExpectEqual(len(leases.Items), 2) + + selector := labels.Set(map[string]string{"deletecollection": "true"}).AsSelector() + err = leaseClient.DeleteCollection(&metav1.DeleteOptions{}, metav1.ListOptions{LabelSelector: selector.String()}) + framework.ExpectNoError(err, "couldn't delete collection") + + leases, err = leaseClient.List(metav1.ListOptions{}) + framework.ExpectNoError(err, "couldn't list Leases") + framework.ExpectEqual(len(leases.Items), 1) + + err = leaseClient.Delete(name, &metav1.DeleteOptions{}) + framework.ExpectNoError(err, "deleting Lease failed") + + _, err = leaseClient.Get(name, metav1.GetOptions{}) + framework.ExpectEqual(errors.IsNotFound(err), true) + }) +})