mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-08-23 18:40:53 +00:00
storage: etcd: TestGetToList: use sub-tests
Signed-off-by: Steve Kuznetsov <skuznets@redhat.com>
This commit is contained in:
parent
aa0e6320d5
commit
f1ded3b0c3
@ -35,7 +35,7 @@ import (
|
|||||||
clientv3 "go.etcd.io/etcd/client/v3"
|
clientv3 "go.etcd.io/etcd/client/v3"
|
||||||
"google.golang.org/grpc/grpclog"
|
"google.golang.org/grpc/grpclog"
|
||||||
|
|
||||||
apitesting "k8s.io/apimachinery/pkg/api/apitesting"
|
"k8s.io/apimachinery/pkg/api/apitesting"
|
||||||
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
||||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||||
"k8s.io/apimachinery/pkg/conversion"
|
"k8s.io/apimachinery/pkg/conversion"
|
||||||
@ -542,68 +542,80 @@ func TestGetToList(t *testing.T) {
|
|||||||
currentRV, _ := strconv.Atoi(storedObj.ResourceVersion)
|
currentRV, _ := strconv.Atoi(storedObj.ResourceVersion)
|
||||||
|
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
|
name string
|
||||||
key string
|
key string
|
||||||
pred storage.SelectionPredicate
|
pred storage.SelectionPredicate
|
||||||
expectedOut []*example.Pod
|
expectedOut []*example.Pod
|
||||||
rv string
|
rv string
|
||||||
rvMatch metav1.ResourceVersionMatch
|
rvMatch metav1.ResourceVersionMatch
|
||||||
expectRVTooLarge bool
|
expectRVTooLarge bool
|
||||||
}{{ // test GetToList on existing key
|
}{{
|
||||||
|
name: "test GetToList on existing key",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
}, { // test GetToList on existing key with minimum resource version set to 0
|
}, {
|
||||||
|
name: "test GetToList on existing key with minimum resource version set to 0",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
rv: "0",
|
rv: "0",
|
||||||
}, { // test GetToList on existing key with minimum resource version set to 0, match=minimum
|
}, {
|
||||||
|
name: "test GetToList on existing key with minimum resource version set to 0, match=minimum",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
rv: "0",
|
rv: "0",
|
||||||
rvMatch: metav1.ResourceVersionMatchNotOlderThan,
|
rvMatch: metav1.ResourceVersionMatchNotOlderThan,
|
||||||
}, { // test GetToList on existing key with minimum resource version set to current resource version
|
}, {
|
||||||
|
name: "test GetToList on existing key with minimum resource version set to current resource version",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
rv: fmt.Sprintf("%d", currentRV),
|
rv: fmt.Sprintf("%d", currentRV),
|
||||||
}, { // test GetToList on existing key with minimum resource version set to current resource version, match=minimum
|
}, {
|
||||||
|
name: "test GetToList on existing key with minimum resource version set to current resource version, match=minimum",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
rv: fmt.Sprintf("%d", currentRV),
|
rv: fmt.Sprintf("%d", currentRV),
|
||||||
rvMatch: metav1.ResourceVersionMatchNotOlderThan,
|
rvMatch: metav1.ResourceVersionMatchNotOlderThan,
|
||||||
}, { // test GetToList on existing key with minimum resource version set to previous resource version, match=minimum
|
}, {
|
||||||
|
name: "test GetToList on existing key with minimum resource version set to previous resource version, match=minimum",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
rv: fmt.Sprintf("%d", prevRV),
|
rv: fmt.Sprintf("%d", prevRV),
|
||||||
rvMatch: metav1.ResourceVersionMatchNotOlderThan,
|
rvMatch: metav1.ResourceVersionMatchNotOlderThan,
|
||||||
}, { // test GetToList on existing key with resource version set to current resource version, match=exact
|
}, {
|
||||||
|
name: "test GetToList on existing key with resource version set to current resource version, match=exact",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
rv: fmt.Sprintf("%d", currentRV),
|
rv: fmt.Sprintf("%d", currentRV),
|
||||||
rvMatch: metav1.ResourceVersionMatchExact,
|
rvMatch: metav1.ResourceVersionMatchExact,
|
||||||
}, { // test GetToList on existing key with resource version set to previous resource version, match=exact
|
}, {
|
||||||
|
name: "test GetToList on existing key with resource version set to previous resource version, match=exact",
|
||||||
key: prevKey,
|
key: prevKey,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{prevStoredObj},
|
expectedOut: []*example.Pod{prevStoredObj},
|
||||||
rv: fmt.Sprintf("%d", prevRV),
|
rv: fmt.Sprintf("%d", prevRV),
|
||||||
rvMatch: metav1.ResourceVersionMatchExact,
|
rvMatch: metav1.ResourceVersionMatchExact,
|
||||||
}, { // test GetToList on existing key with minimum resource version set too high
|
}, {
|
||||||
|
name: "test GetToList on existing key with minimum resource version set too high",
|
||||||
key: key,
|
key: key,
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: []*example.Pod{storedObj},
|
expectedOut: []*example.Pod{storedObj},
|
||||||
rv: fmt.Sprintf("%d", currentRV+1),
|
rv: fmt.Sprintf("%d", currentRV+1),
|
||||||
expectRVTooLarge: true,
|
expectRVTooLarge: true,
|
||||||
}, { // test GetToList on non-existing key
|
}, {
|
||||||
|
name: "test GetToList on non-existing key",
|
||||||
key: "/non-existing",
|
key: "/non-existing",
|
||||||
pred: storage.Everything,
|
pred: storage.Everything,
|
||||||
expectedOut: nil,
|
expectedOut: nil,
|
||||||
}, { // test GetToList with matching pod name
|
}, {
|
||||||
key: "/non-existing",
|
name: "test GetToList with matching pod name",
|
||||||
|
key: "/non-existing",
|
||||||
pred: storage.SelectionPredicate{
|
pred: storage.SelectionPredicate{
|
||||||
Label: labels.Everything(),
|
Label: labels.Everything(),
|
||||||
Field: fields.ParseSelectorOrDie("metadata.name!=" + storedObj.Name),
|
Field: fields.ParseSelectorOrDie("metadata.name!=" + storedObj.Name),
|
||||||
@ -615,33 +627,35 @@ func TestGetToList(t *testing.T) {
|
|||||||
expectedOut: nil,
|
expectedOut: nil,
|
||||||
}}
|
}}
|
||||||
|
|
||||||
for i, tt := range tests {
|
for _, tt := range tests {
|
||||||
out := &example.PodList{}
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
err := store.GetToList(ctx, tt.key, storage.ListOptions{ResourceVersion: tt.rv, ResourceVersionMatch: tt.rvMatch, Predicate: tt.pred}, out)
|
out := &example.PodList{}
|
||||||
|
err := store.GetToList(ctx, tt.key, storage.ListOptions{ResourceVersion: tt.rv, ResourceVersionMatch: tt.rvMatch, Predicate: tt.pred}, out)
|
||||||
|
|
||||||
if tt.expectRVTooLarge {
|
if tt.expectRVTooLarge {
|
||||||
if err == nil || !storage.IsTooLargeResourceVersion(err) {
|
if err == nil || !storage.IsTooLargeResourceVersion(err) {
|
||||||
t.Errorf("#%d: expecting resource version too high error, but get: %s", i, err)
|
t.Errorf("%s: expecting resource version too high error, but get: %s", tt.name, err)
|
||||||
|
}
|
||||||
|
return
|
||||||
}
|
}
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("GetToList failed: %v", err)
|
t.Fatalf("GetToList failed: %v", err)
|
||||||
}
|
|
||||||
if len(out.ResourceVersion) == 0 {
|
|
||||||
t.Errorf("#%d: unset resourceVersion", i)
|
|
||||||
}
|
|
||||||
if len(out.Items) != len(tt.expectedOut) {
|
|
||||||
t.Errorf("#%d: length of list want=%d, get=%d", i, len(tt.expectedOut), len(out.Items))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
for j, wantPod := range tt.expectedOut {
|
|
||||||
getPod := &out.Items[j]
|
|
||||||
if !reflect.DeepEqual(wantPod, getPod) {
|
|
||||||
t.Errorf("#%d: pod want=%#v, get=%#v", i, wantPod, getPod)
|
|
||||||
}
|
}
|
||||||
}
|
if len(out.ResourceVersion) == 0 {
|
||||||
|
t.Errorf("%s: unset resourceVersion", tt.name)
|
||||||
|
}
|
||||||
|
if len(out.Items) != len(tt.expectedOut) {
|
||||||
|
t.Errorf("%s: length of list want=%d, get=%d", tt.name, len(tt.expectedOut), len(out.Items))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for j, wantPod := range tt.expectedOut {
|
||||||
|
getPod := &out.Items[j]
|
||||||
|
if !reflect.DeepEqual(wantPod, getPod) {
|
||||||
|
t.Errorf("%s: pod want=%#v, get=%#v", tt.name, wantPod, getPod)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user