mirror of
https://github.com/kubernetes/client-go.git
synced 2025-07-04 02:37:39 +00:00
201 lines
9.9 KiB
Go
201 lines
9.9 KiB
Go
/*
|
|
Copyright 2024 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 watchlist
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
clientfeatures "k8s.io/client-go/features"
|
|
clientfeaturestesting "k8s.io/client-go/features/testing"
|
|
"k8s.io/utils/ptr"
|
|
)
|
|
|
|
// TestPrepareWatchListOptionsFromListOptions test the following cases:
|
|
//
|
|
// +--------------------------+-----------------+---------+-----------------+
|
|
// | ResourceVersionMatch | ResourceVersion | Limit | Continuation |
|
|
// +--------------------------+-----------------+---------+-----------------+
|
|
// | unset/NotOlderThan/Exact | unset/0/100 | unset/4 | unset/FakeToken |
|
|
// +--------------------------+-----------------+---------+-----------------+
|
|
func TestPrepareWatchListOptionsFromListOptions(t *testing.T) {
|
|
scenarios := []struct {
|
|
name string
|
|
listOptions metav1.ListOptions
|
|
enableWatchListFG bool
|
|
|
|
expectToPrepareWatchListOptions bool
|
|
expectedWatchListOptions metav1.ListOptions
|
|
}{
|
|
|
|
{
|
|
name: "can't enable watch list for: WatchListClient=off, RVM=unset, RV=unset, Limit=unset, Continuation=unset",
|
|
enableWatchListFG: false,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
// +----------------------+-----------------+-------+--------------+
|
|
// | ResourceVersionMatch | ResourceVersion | Limit | Continuation |
|
|
// +----------------------+-----------------+-------+--------------+
|
|
// | unset | unset | unset | unset |
|
|
// | unset | 0 | unset | unset |
|
|
// | unset | 100 | unset | unset |
|
|
// | unset | 0 | 4 | unset |
|
|
// | unset | 0 | unset | FakeToken |
|
|
// +----------------------+-----------------+-------+--------------+
|
|
{
|
|
name: "can enable watch list for: RVM=unset, RV=unset, Limit=unset, Continuation=unset",
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: true,
|
|
expectedWatchListOptions: expectedWatchListOptionsFor(""),
|
|
},
|
|
{
|
|
name: "can enable watch list for: RVM=unset, RV=0, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersion: "0"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: true,
|
|
expectedWatchListOptions: expectedWatchListOptionsFor("0"),
|
|
},
|
|
{
|
|
name: "can enable watch list for: RVM=unset, RV=100, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersion: "100"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: true,
|
|
expectedWatchListOptions: expectedWatchListOptionsFor("100"),
|
|
},
|
|
{
|
|
name: "legacy: can enable watch list for: RVM=unset, RV=0, Limit=4, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersion: "0", Limit: 4},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: true,
|
|
expectedWatchListOptions: expectedWatchListOptionsFor("0"),
|
|
},
|
|
{
|
|
name: "can't enable watch list for: RVM=unset, RV=0, Limit=unset, Continuation=FakeToken",
|
|
listOptions: metav1.ListOptions{ResourceVersion: "0", Continue: "FakeToken"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
// +----------------------+-----------------+-------+--------------+
|
|
// | ResourceVersionMatch | ResourceVersion | Limit | Continuation |
|
|
// +----------------------+-----------------+-------+--------------+
|
|
// | NotOlderThan | unset | unset | unset |
|
|
// | NotOlderThan | 0 | unset | unset |
|
|
// | NotOlderThan | 100 | unset | unset |
|
|
// | NotOlderThan | 0 | 4 | unset |
|
|
// | NotOlderThan | 0 | unset | FakeToken |
|
|
// +----------------------+-----------------+-------+--------------+
|
|
{
|
|
name: "can't enable watch list for: RVM=NotOlderThan, RV=unset, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchNotOlderThan},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
{
|
|
name: "can enable watch list for: RVM=NotOlderThan, RV=0, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchNotOlderThan, ResourceVersion: "0"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: true,
|
|
expectedWatchListOptions: expectedWatchListOptionsFor("0"),
|
|
},
|
|
{
|
|
name: "can enable watch list for: RVM=NotOlderThan, RV=100, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchNotOlderThan, ResourceVersion: "100"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: true,
|
|
expectedWatchListOptions: expectedWatchListOptionsFor("100"),
|
|
},
|
|
{
|
|
name: "legacy: can enable watch list for: RVM=NotOlderThan, RV=0, Limit=4, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchNotOlderThan, ResourceVersion: "0", Limit: 4},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: true,
|
|
expectedWatchListOptions: expectedWatchListOptionsFor("0"),
|
|
},
|
|
{
|
|
name: "can't enable watch list for: RVM=NotOlderThan, RV=0, Limit=unset, Continuation=FakeToken",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchNotOlderThan, ResourceVersion: "0", Continue: "FakeToken"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
|
|
// +----------------------+-----------------+-------+--------------+
|
|
// | ResourceVersionMatch | ResourceVersion | Limit | Continuation |
|
|
// +----------------------+-----------------+-------+--------------+
|
|
// | Exact | unset | unset | unset |
|
|
// | Exact | 0 | unset | unset |
|
|
// | Exact | 100 | unset | unset |
|
|
// | Exact | 0 | 4 | unset |
|
|
// | Exact | 0 | unset | FakeToken |
|
|
// +----------------------+-----------------+-------+--------------+
|
|
{
|
|
name: "can't enable watch list for: RVM=Exact, RV=unset, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchExact},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
{
|
|
name: "can enable watch list for: RVM=Exact, RV=0, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchExact, ResourceVersion: "0"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
{
|
|
name: "can enable watch list for: RVM=Exact, RV=100, Limit=unset, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchExact, ResourceVersion: "100"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
{
|
|
name: "can't enable watch list for: RVM=Exact, RV=0, Limit=4, Continuation=unset",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchExact, ResourceVersion: "0", Limit: 4},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
{
|
|
name: "can't enable watch list for: RVM=Exact, RV=0, Limit=unset, Continuation=FakeToken",
|
|
listOptions: metav1.ListOptions{ResourceVersionMatch: metav1.ResourceVersionMatchExact, ResourceVersion: "0", Continue: "FakeToken"},
|
|
enableWatchListFG: true,
|
|
expectToPrepareWatchListOptions: false,
|
|
},
|
|
}
|
|
for _, scenario := range scenarios {
|
|
t.Run(scenario.name, func(t *testing.T) {
|
|
clientfeaturestesting.SetFeatureDuringTest(t, clientfeatures.WatchListClient, scenario.enableWatchListFG)
|
|
|
|
watchListOptions, hasWatchListOptionsPrepared, err := PrepareWatchListOptionsFromListOptions(scenario.listOptions)
|
|
|
|
require.NoError(t, err)
|
|
require.Equal(t, scenario.expectToPrepareWatchListOptions, hasWatchListOptionsPrepared)
|
|
require.Equal(t, scenario.expectedWatchListOptions, watchListOptions)
|
|
})
|
|
}
|
|
}
|
|
|
|
func expectedWatchListOptionsFor(rv string) metav1.ListOptions {
|
|
var watchListOptions metav1.ListOptions
|
|
|
|
watchListOptions.ResourceVersion = rv
|
|
watchListOptions.ResourceVersionMatch = metav1.ResourceVersionMatchNotOlderThan
|
|
watchListOptions.Watch = true
|
|
watchListOptions.AllowWatchBookmarks = true
|
|
watchListOptions.SendInitialEvents = ptr.To(true)
|
|
|
|
return watchListOptions
|
|
}
|