github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-07 19:23:40 +00:00
Code Issues Projects Releases Wiki Activity

apiserver/storage: improve RunWatchSemanticInitialEventsExtended test

Browse Source 
changes the test to populate the underlying data store with
more data to trigger potential ordering issues.
This commit is contained in:
Lukasz Szaszkiewicz 2024-01-10 11:08:35 +01:00
parent d2b4928669
commit 20ded27570
1 changed files with 23 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
staging/src/k8s.io/apiserver/pkg/storage/testing/watcher_tests.go
View File

@ -1513,31 +1513,36 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
// by adding the pod to a different ns to advance the global RV
func RunWatchSemanticInitialEventsExtended(ctx context.Context, t *testing.T, store storage.Interface) {
trueVal := true
expectedInitialEventsInStrictOrder := func(firstPod, secondPod *example.Pod) []watch.Event {
return []watch.Event{
{Type: watch.Added, Object: firstPod},
{Type: watch.Bookmark, Object: &example.Pod{
ObjectMeta: metav1.ObjectMeta{
ResourceVersion: secondPod.ResourceVersion,
Annotations: map[string]string{"k8s.io/initial-events-end": "true"},
},
}},
expectedInitialEventsInStrictOrder := func(initialPods []*example.Pod, globalResourceVersion string) []watch.Event {
watchEvents := []watch.Event{}
for _, initialPod := range initialPods {
watchEvents = append(watchEvents, watch.Event{Type: watch.Added, Object: initialPod})
}
watchEvents = append(watchEvents, watch.Event{Type: watch.Bookmark, Object: &example.Pod{
ObjectMeta: metav1.ObjectMeta{
ResourceVersion: globalResourceVersion,
Annotations: map[string]string{"k8s.io/initial-events-end": "true"},
},
}})
return watchEvents
}
defer featuregatetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.WatchList, true)()
initialPods := []*example.Pod{}
ns := "ns-foo"
pod := makePod("1")
pod.Namespace = ns
firstPod := &example.Pod{}
err := store.Create(ctx, computePodKey(pod), pod, firstPod, 0)
require.NoError(t, err, "failed to add a pod: %v")
for _, initialPod := range []*example.Pod{makePod("1"), makePod("2"), makePod("3"), makePod("4"), makePod("5")} {
initialPod.Namespace = ns
out := &example.Pod{}
err := store.Create(ctx, computePodKey(initialPod), initialPod, out, 0)
require.NoError(t, err, "failed to add a pod: %v")
initialPods = append(initialPods, out)
}
// add the pod to a different ns to advance the global RV
pod = makePod("2")
pod := makePod("1")
pod.Namespace = "other-ns-foo"
secondPod := &example.Pod{}
err = store.Create(ctx, computePodKey(pod), pod, secondPod, 0)
otherNsPod := &example.Pod{}
err := store.Create(ctx, computePodKey(pod), pod, otherNsPod, 0)
require.NoError(t, err, "failed to add a pod: %v")
opts := storage.ListOptions{Predicate: storage.Everything, Recursive: true}
@ -1550,7 +1555,7 @@ func RunWatchSemanticInitialEventsExtended(ctx context.Context, t *testing.T, st
// make sure we only get initial events from the first ns
// followed by the bookmark with the global RV
testCheckResultsInStrictOrder(t, w, expectedInitialEventsInStrictOrder(firstPod, secondPod))
testCheckResultsInStrictOrder(t, w, expectedInitialEventsInStrictOrder(initialPods, otherNsPod.ResourceVersion))
testCheckNoMoreResults(t, w)
}