github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-06 02:34:03 +00:00
Code Issues Projects Releases Wiki Activity

daemonset registry: add unit test for UpdateStatus

Browse Source
This commit is contained in:
carlory 2023-11-15 15:19:31 +08:00
parent 7b1c378edd
commit 3475c3e955
1 changed files with 34 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
pkg/registry/apps/daemonset/storage/storage_test.go
View File

@ -19,12 +19,16 @@ package storage
import ( import (
"testing" "testing"
"github.com/google/go-cmp/cmp"
apiequality "k8s.io/apimachinery/pkg/api/equality"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/fields" "k8s.io/apimachinery/pkg/fields"
"k8s.io/apimachinery/pkg/labels" "k8s.io/apimachinery/pkg/labels"
"k8s.io/apimachinery/pkg/runtime" "k8s.io/apimachinery/pkg/runtime"
genericapirequest "k8s.io/apiserver/pkg/endpoints/request"
"k8s.io/apiserver/pkg/registry/generic" "k8s.io/apiserver/pkg/registry/generic"
genericregistrytest "k8s.io/apiserver/pkg/registry/generic/testing" genericregistrytest "k8s.io/apiserver/pkg/registry/generic/testing"
"k8s.io/apiserver/pkg/registry/rest"
etcd3testing "k8s.io/apiserver/pkg/storage/etcd3/testing" etcd3testing "k8s.io/apiserver/pkg/storage/etcd3/testing"
"k8s.io/kubernetes/pkg/apis/apps" "k8s.io/kubernetes/pkg/apis/apps"
api "k8s.io/kubernetes/pkg/apis/core" api "k8s.io/kubernetes/pkg/apis/core"
@ -189,6 +193,36 @@ func TestWatch(t *testing.T) {
) )
} }
func TestUpdateStatus(t *testing.T) {
storage, statusStorage, server := newStorage(t)
defer server.Terminate(t)
defer storage.Store.DestroyFunc()
dsStart := newValidDaemonSet()
ctx := genericapirequest.WithNamespace(genericapirequest.NewDefaultContext(), dsStart.Namespace)
key, _ := storage.KeyFunc(ctx, dsStart.Name)
err := storage.Storage.Create(ctx, key, dsStart, nil, 0, false)
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
ds := dsStart.DeepCopy()
ds.Status.ObservedGeneration = 1
_, _, err = statusStorage.Update(ctx, ds.Name, rest.DefaultUpdatedObjectInfo(ds), rest.ValidateAllObjectFunc, rest.ValidateAllObjectUpdateFunc, false, &metav1.UpdateOptions{})
if err != nil {
t.Fatalf("Unexpected error: %v", err)
}
obj, err := storage.Get(ctx, ds.Name, &metav1.GetOptions{})
if err != nil {
t.Errorf("unexpected error: %v", err)
}
dsOut := obj.(*apps.DaemonSet)
// only compare relevant changes b/c of difference in metadata
if !apiequality.Semantic.DeepEqual(ds.Status, dsOut.Status) {
t.Errorf("unexpected object: %s", cmp.Diff(ds.Status, dsOut.Status))
}
}
func TestShortNames(t *testing.T) { func TestShortNames(t *testing.T) {
storage, _, server := newStorage(t) storage, _, server := newStorage(t)
defer server.Terminate(t) defer server.Terminate(t)
@ -196,5 +230,3 @@ func TestShortNames(t *testing.T) {
expected := []string{"ds"} expected := []string{"ds"}
registrytest.AssertShortNames(t, storage, expected) registrytest.AssertShortNames(t, storage, expected)
} }
// TODO TestUpdateStatus