apiserver/storage: decrease running time of RunWatchSemantics

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

@@ -28,6 +28,7 @@ import (
 
 	apiequality "k8s.io/apimachinery/pkg/api/equality"
 	"k8s.io/apimachinery/pkg/api/errors"
+	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/fields"
 	"k8s.io/apimachinery/pkg/labels"
@@ -1236,6 +1237,7 @@ func RunSendInitialEventsBackwardCompatibility(ctx context.Context, t *testing.T
 // - false indicates the value of the param was set to "false" by a test case
 // - true  indicates the value of the param was set to "true" by a test case
 func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interface) {
+	featuregatetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.WatchList, true)
 	trueVal, falseVal := true, false
 	addEventsFromCreatedPods := func(createdInitialPods []*example.Pod) []watch.Event {
 		var ret []watch.Event
@@ -1244,8 +1246,8 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 		}
 		return ret
 	}
-	initialEventsEndFromLastCreatedPod := func(createdInitialPods []*example.Pod) []watch.Event {
+	initialEventsEndFromLastCreatedPod := func(createdInitialPods []*example.Pod) watch.Event {
-		return []watch.Event{{
+		return watch.Event{
 			Type: watch.Bookmark,
 			Object: &example.Pod{
 				ObjectMeta: metav1.ObjectMeta{
@@ -1253,7 +1255,7 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 					Annotations:     map[string]string{metav1.InitialEventsAnnotationKey: "true"},
 				},
 			},
-		}}
+		}
 	}
 	scenarios := []struct {
 		name                string
@@ -1268,7 +1270,7 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 		podsAfterEstablishingWatch []*example.Pod
 
 		expectedInitialEvents                      func(createdInitialPods []*example.Pod) []watch.Event
-		expectedInitialEventsBookmark        func(createdInitialPods []*example.Pod) []watch.Event
+		expectedInitialEventsBookmarkWithMinimalRV func(createdInitialPods []*example.Pod) watch.Event
 		expectedEventsAfterEstablishingWatch       func(createdPodsAfterWatch []*example.Pod) []watch.Event
 	}{
 		{
@@ -1277,7 +1279,7 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 			sendInitialEvents:     &trueVal,
 			initialPods:           []*example.Pod{makePod("1"), makePod("2"), makePod("3")},
 			expectedInitialEvents: addEventsFromCreatedPods,
-			expectedInitialEventsBookmark:        initialEventsEndFromLastCreatedPod,
+			expectedInitialEventsBookmarkWithMinimalRV: initialEventsEndFromLastCreatedPod,
 			podsAfterEstablishingWatch:                 []*example.Pod{makePod("4"), makePod("5")},
 			expectedEventsAfterEstablishingWatch:       addEventsFromCreatedPods,
 		},
@@ -1312,7 +1314,7 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 			resourceVersion:       "0",
 			initialPods:           []*example.Pod{makePod("1"), makePod("2"), makePod("3")},
 			expectedInitialEvents: addEventsFromCreatedPods,
-			expectedInitialEventsBookmark:        initialEventsEndFromLastCreatedPod,
+			expectedInitialEventsBookmarkWithMinimalRV: initialEventsEndFromLastCreatedPod,
 			podsAfterEstablishingWatch:                 []*example.Pod{makePod("4"), makePod("5")},
 			expectedEventsAfterEstablishingWatch:       addEventsFromCreatedPods,
 		},
@@ -1350,7 +1352,7 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 			resourceVersion:       "1",
 			initialPods:           []*example.Pod{makePod("1"), makePod("2"), makePod("3")},
 			expectedInitialEvents: addEventsFromCreatedPods,
-			expectedInitialEventsBookmark:        initialEventsEndFromLastCreatedPod,
+			expectedInitialEventsBookmarkWithMinimalRV: initialEventsEndFromLastCreatedPod,
 			podsAfterEstablishingWatch:                 []*example.Pod{makePod("4"), makePod("5")},
 			expectedEventsAfterEstablishingWatch:       addEventsFromCreatedPods,
 		},
@@ -1390,7 +1392,7 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 			useCurrentRV:          true,
 			initialPods:           []*example.Pod{makePod("1"), makePod("2"), makePod("3")},
 			expectedInitialEvents: addEventsFromCreatedPods,
-			expectedInitialEventsBookmark:        initialEventsEndFromLastCreatedPod,
+			expectedInitialEventsBookmarkWithMinimalRV: initialEventsEndFromLastCreatedPod,
 			podsAfterEstablishingWatch:                 []*example.Pod{makePod("4"), makePod("5")},
 			expectedEventsAfterEstablishingWatch:       addEventsFromCreatedPods,
 		},
@@ -1439,14 +1441,11 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 	}
 	for idx, scenario := range scenarios {
 		t.Run(scenario.name, func(t *testing.T) {
+			t.Parallel()
 			// set up env
-			featuregatetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.WatchList, true)
 			if scenario.expectedInitialEvents == nil {
 				scenario.expectedInitialEvents = func(_ []*example.Pod) []watch.Event { return nil }
 			}
-			if scenario.expectedInitialEventsBookmark == nil {
-				scenario.expectedInitialEventsBookmark = func(_ []*example.Pod) []watch.Event { return nil }
-			}
 			if scenario.expectedEventsAfterEstablishingWatch == nil {
 				scenario.expectedEventsAfterEstablishingWatch = func(_ []*example.Pod) []watch.Event { return nil }
 			}
@@ -1480,7 +1479,25 @@ func RunWatchSemantics(ctx context.Context, t *testing.T, store storage.Interfac
 
 			// make sure we only get initial events
 			testCheckResultsInStrictOrder(t, w, scenario.expectedInitialEvents(createdPods))
-			testCheckResultsInStrictOrder(t, w, scenario.expectedInitialEventsBookmark(createdPods))
+
+			// make sure that the actual bookmark has at least RV >= to the expected one
+			if scenario.expectedInitialEventsBookmarkWithMinimalRV != nil {
+				testCheckResultFunc(t, w, func(actualEvent watch.Event) {
+					expectedBookmarkEventWithMinRV := scenario.expectedInitialEventsBookmarkWithMinimalRV(createdPods)
+					expectedObj, err := meta.Accessor(expectedBookmarkEventWithMinRV.Object)
+					require.NoError(t, err)
+
+					actualObj, err := meta.Accessor(actualEvent.Object)
+					require.NoError(t, err)
+
+					require.GreaterOrEqual(t, actualObj.GetResourceVersion(), expectedObj.GetResourceVersion())
+
+					// once we know that the RV is at least >= the expected one
+					// rewrite it so that we can compare the objs
+					expectedObj.SetResourceVersion(actualObj.GetResourceVersion())
+					expectNoDiff(t, "incorrect event", expectedBookmarkEventWithMinRV, actualEvent)
+				})
+			}
 
 			createdPods = []*example.Pod{}
 			// add a pod that is greater than the storage's RV when the watch was started