123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790 |
- /*
- Copyright 2018 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 wait
- import (
- "io/ioutil"
- "strings"
- "testing"
- "time"
- "github.com/davecgh/go-spew/spew"
- "k8s.io/apimachinery/pkg/api/meta"
- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
- "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
- "k8s.io/apimachinery/pkg/runtime"
- "k8s.io/apimachinery/pkg/runtime/schema"
- "k8s.io/apimachinery/pkg/types"
- "k8s.io/apimachinery/pkg/watch"
- "k8s.io/cli-runtime/pkg/genericclioptions"
- "k8s.io/cli-runtime/pkg/printers"
- "k8s.io/cli-runtime/pkg/resource"
- dynamicfakeclient "k8s.io/client-go/dynamic/fake"
- clienttesting "k8s.io/client-go/testing"
- )
- func newUnstructuredList(items ...*unstructured.Unstructured) *unstructured.UnstructuredList {
- list := &unstructured.UnstructuredList{}
- for i := range items {
- list.Items = append(list.Items, *items[i])
- }
- return list
- }
- func newUnstructured(apiVersion, kind, namespace, name string) *unstructured.Unstructured {
- return &unstructured.Unstructured{
- Object: map[string]interface{}{
- "apiVersion": apiVersion,
- "kind": kind,
- "metadata": map[string]interface{}{
- "namespace": namespace,
- "name": name,
- "uid": "some-UID-value",
- },
- },
- }
- }
- func newUnstructuredStatus(status *metav1.Status) runtime.Unstructured {
- obj, err := runtime.DefaultUnstructuredConverter.ToUnstructured(status)
- if err != nil {
- panic(err)
- }
- return &unstructured.Unstructured{
- Object: obj,
- }
- }
- func addCondition(in *unstructured.Unstructured, name, status string) *unstructured.Unstructured {
- conditions, _, _ := unstructured.NestedSlice(in.Object, "status", "conditions")
- conditions = append(conditions, map[string]interface{}{
- "type": name,
- "status": status,
- })
- unstructured.SetNestedSlice(in.Object, conditions, "status", "conditions")
- return in
- }
- func TestWaitForDeletion(t *testing.T) {
- scheme := runtime.NewScheme()
- tests := []struct {
- name string
- infos []*resource.Info
- fakeClient func() *dynamicfakeclient.FakeDynamicClient
- timeout time.Duration
- uidMap UIDMap
- expectedErr string
- validateActions func(t *testing.T, actions []clienttesting.Action)
- }{
- {
- name: "missing on get",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- return dynamicfakeclient.NewSimpleDynamicClient(scheme)
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 1 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles no infos",
- infos: []*resource.Info{},
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- return dynamicfakeclient.NewSimpleDynamicClient(scheme)
- },
- timeout: 10 * time.Second,
- expectedErr: errNoMatchingResources.Error(),
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 0 {
- t.Fatal(spew.Sdump(actions))
- }
- },
- },
- {
- name: "uid conflict on get",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructuredList(newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")), nil
- })
- count := 0
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- if count == 0 {
- count++
- fakeWatch := watch.NewRaceFreeFake()
- go func() {
- time.Sleep(100 * time.Millisecond)
- fakeWatch.Stop()
- }()
- return true, fakeWatch, nil
- }
- fakeWatch := watch.NewRaceFreeFake()
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- uidMap: UIDMap{
- ResourceLocation{Namespace: "ns-foo", Name: "name-foo"}: types.UID("some-UID-value"),
- ResourceLocation{GroupResource: schema.GroupResource{Group: "group", Resource: "theresource"}, Namespace: "ns-foo", Name: "name-foo"}: types.UID("some-nonmatching-UID-value"),
- },
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 1 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "times out",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructuredList(newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")), nil
- })
- return fakeClient
- },
- timeout: 1 * time.Second,
- expectedErr: "timed out waiting for the condition on theresource/name-foo",
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 2 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles watch close out",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- unstructuredObj := newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")
- unstructuredObj.SetResourceVersion("123")
- unstructuredList := newUnstructuredList(unstructuredObj)
- unstructuredList.SetResourceVersion("234")
- return true, unstructuredList, nil
- })
- count := 0
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- if count == 0 {
- count++
- fakeWatch := watch.NewRaceFreeFake()
- go func() {
- time.Sleep(100 * time.Millisecond)
- fakeWatch.Stop()
- }()
- return true, fakeWatch, nil
- }
- fakeWatch := watch.NewRaceFreeFake()
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 3 * time.Second,
- expectedErr: "timed out waiting for the condition on theresource/name-foo",
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 4 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") || actions[1].(clienttesting.WatchAction).GetWatchRestrictions().ResourceVersion != "234" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[2].Matches("list", "theresource") || actions[2].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[3].Matches("watch", "theresource") || actions[3].(clienttesting.WatchAction).GetWatchRestrictions().ResourceVersion != "234" {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles watch delete",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructuredList(newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")), nil
- })
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- fakeWatch := watch.NewRaceFreeFake()
- fakeWatch.Action(watch.Deleted, newUnstructured("group/version", "TheKind", "ns-foo", "name-foo"))
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 2 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles watch delete multiple",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource-1"},
- },
- Name: "name-foo-1",
- Namespace: "ns-foo",
- },
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource-2"},
- },
- Name: "name-foo-2",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("get", "theresource-1", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructured("group/version", "TheKind", "ns-foo", "name-foo-1"), nil
- })
- fakeClient.PrependReactor("get", "theresource-2", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructured("group/version", "TheKind", "ns-foo", "name-foo-2"), nil
- })
- fakeClient.PrependWatchReactor("theresource-1", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- fakeWatch := watch.NewRaceFreeFake()
- fakeWatch.Action(watch.Deleted, newUnstructured("group/version", "TheKind", "ns-foo", "name-foo-1"))
- return true, fakeWatch, nil
- })
- fakeClient.PrependWatchReactor("theresource-2", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- fakeWatch := watch.NewRaceFreeFake()
- fakeWatch.Action(watch.Deleted, newUnstructured("group/version", "TheKind", "ns-foo", "name-foo-2"))
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 2 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource-1") {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("list", "theresource-2") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "ignores watch error",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructuredList(newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")), nil
- })
- count := 0
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- fakeWatch := watch.NewRaceFreeFake()
- if count == 0 {
- fakeWatch.Error(newUnstructuredStatus(&metav1.Status{
- TypeMeta: metav1.TypeMeta{Kind: "Status", APIVersion: "v1"},
- Status: "Failure",
- Code: 500,
- Message: "Bad",
- }))
- fakeWatch.Stop()
- } else {
- fakeWatch.Action(watch.Deleted, newUnstructured("group/version", "TheKind", "ns-foo", "name-foo"))
- }
- count++
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 4 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- if !actions[2].Matches("list", "theresource") || actions[2].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[3].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- }
- for _, test := range tests {
- t.Run(test.name, func(t *testing.T) {
- fakeClient := test.fakeClient()
- o := &WaitOptions{
- ResourceFinder: genericclioptions.NewSimpleFakeResourceFinder(test.infos...),
- UIDMap: test.uidMap,
- DynamicClient: fakeClient,
- Timeout: test.timeout,
- Printer: printers.NewDiscardingPrinter(),
- ConditionFn: IsDeleted,
- IOStreams: genericclioptions.NewTestIOStreamsDiscard(),
- }
- err := o.RunWait()
- switch {
- case err == nil && len(test.expectedErr) == 0:
- case err != nil && len(test.expectedErr) == 0:
- t.Fatal(err)
- case err == nil && len(test.expectedErr) != 0:
- t.Fatalf("missing: %q", test.expectedErr)
- case err != nil && len(test.expectedErr) != 0:
- if !strings.Contains(err.Error(), test.expectedErr) {
- t.Fatalf("expected %q, got %q", test.expectedErr, err.Error())
- }
- }
- test.validateActions(t, fakeClient.Actions())
- })
- }
- }
- func TestWaitForCondition(t *testing.T) {
- scheme := runtime.NewScheme()
- tests := []struct {
- name string
- infos []*resource.Info
- fakeClient func() *dynamicfakeclient.FakeDynamicClient
- timeout time.Duration
- expectedErr string
- validateActions func(t *testing.T, actions []clienttesting.Action)
- }{
- {
- name: "present on get",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructuredList(addCondition(
- newUnstructured("group/version", "TheKind", "ns-foo", "name-foo"),
- "the-condition", "status-value",
- )), nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 1 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles no infos",
- infos: []*resource.Info{},
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- return dynamicfakeclient.NewSimpleDynamicClient(scheme)
- },
- timeout: 10 * time.Second,
- expectedErr: errNoMatchingResources.Error(),
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 0 {
- t.Fatal(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles empty object name",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- return dynamicfakeclient.NewSimpleDynamicClient(scheme)
- },
- timeout: 10 * time.Second,
- expectedErr: "resource name must be provided",
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 0 {
- t.Fatal(spew.Sdump(actions))
- }
- },
- },
- {
- name: "times out",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, addCondition(
- newUnstructured("group/version", "TheKind", "ns-foo", "name-foo"),
- "some-other-condition", "status-value",
- ), nil
- })
- return fakeClient
- },
- timeout: 1 * time.Second,
- expectedErr: "timed out waiting for the condition on theresource/name-foo",
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 2 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles watch close out",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- unstructuredObj := newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")
- unstructuredObj.SetResourceVersion("123")
- unstructuredList := newUnstructuredList(unstructuredObj)
- unstructuredList.SetResourceVersion("234")
- return true, unstructuredList, nil
- })
- count := 0
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- if count == 0 {
- count++
- fakeWatch := watch.NewRaceFreeFake()
- go func() {
- time.Sleep(100 * time.Millisecond)
- fakeWatch.Stop()
- }()
- return true, fakeWatch, nil
- }
- fakeWatch := watch.NewRaceFreeFake()
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 3 * time.Second,
- expectedErr: "timed out waiting for the condition on theresource/name-foo",
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 4 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") || actions[1].(clienttesting.WatchAction).GetWatchRestrictions().ResourceVersion != "234" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[2].Matches("list", "theresource") || actions[2].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[3].Matches("watch", "theresource") || actions[3].(clienttesting.WatchAction).GetWatchRestrictions().ResourceVersion != "234" {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles watch condition change",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructuredList(newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")), nil
- })
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- fakeWatch := watch.NewRaceFreeFake()
- fakeWatch.Action(watch.Modified, addCondition(
- newUnstructured("group/version", "TheKind", "ns-foo", "name-foo"),
- "the-condition", "status-value",
- ))
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 2 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "handles watch created",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- fakeWatch := watch.NewRaceFreeFake()
- fakeWatch.Action(watch.Added, addCondition(
- newUnstructured("group/version", "TheKind", "ns-foo", "name-foo"),
- "the-condition", "status-value",
- ))
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 2 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- {
- name: "ignores watch error",
- infos: []*resource.Info{
- {
- Mapping: &meta.RESTMapping{
- Resource: schema.GroupVersionResource{Group: "group", Version: "version", Resource: "theresource"},
- },
- Name: "name-foo",
- Namespace: "ns-foo",
- },
- },
- fakeClient: func() *dynamicfakeclient.FakeDynamicClient {
- fakeClient := dynamicfakeclient.NewSimpleDynamicClient(scheme)
- fakeClient.PrependReactor("list", "theresource", func(action clienttesting.Action) (handled bool, ret runtime.Object, err error) {
- return true, newUnstructuredList(newUnstructured("group/version", "TheKind", "ns-foo", "name-foo")), nil
- })
- count := 0
- fakeClient.PrependWatchReactor("theresource", func(action clienttesting.Action) (handled bool, ret watch.Interface, err error) {
- fakeWatch := watch.NewRaceFreeFake()
- if count == 0 {
- fakeWatch.Error(newUnstructuredStatus(&metav1.Status{
- TypeMeta: metav1.TypeMeta{Kind: "Status", APIVersion: "v1"},
- Status: "Failure",
- Code: 500,
- Message: "Bad",
- }))
- fakeWatch.Stop()
- } else {
- fakeWatch.Action(watch.Modified, addCondition(
- newUnstructured("group/version", "TheKind", "ns-foo", "name-foo"),
- "the-condition", "status-value",
- ))
- }
- count++
- return true, fakeWatch, nil
- })
- return fakeClient
- },
- timeout: 10 * time.Second,
- validateActions: func(t *testing.T, actions []clienttesting.Action) {
- if len(actions) != 4 {
- t.Fatal(spew.Sdump(actions))
- }
- if !actions[0].Matches("list", "theresource") || actions[0].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[1].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- if !actions[2].Matches("list", "theresource") || actions[2].(clienttesting.ListAction).GetListRestrictions().Fields.String() != "metadata.name=name-foo" {
- t.Error(spew.Sdump(actions))
- }
- if !actions[3].Matches("watch", "theresource") {
- t.Error(spew.Sdump(actions))
- }
- },
- },
- }
- for _, test := range tests {
- t.Run(test.name, func(t *testing.T) {
- fakeClient := test.fakeClient()
- o := &WaitOptions{
- ResourceFinder: genericclioptions.NewSimpleFakeResourceFinder(test.infos...),
- DynamicClient: fakeClient,
- Timeout: test.timeout,
- Printer: printers.NewDiscardingPrinter(),
- ConditionFn: ConditionalWait{conditionName: "the-condition", conditionStatus: "status-value", errOut: ioutil.Discard}.IsConditionMet,
- IOStreams: genericclioptions.NewTestIOStreamsDiscard(),
- }
- err := o.RunWait()
- switch {
- case err == nil && len(test.expectedErr) == 0:
- case err != nil && len(test.expectedErr) == 0:
- t.Fatal(err)
- case err == nil && len(test.expectedErr) != 0:
- t.Fatalf("missing: %q", test.expectedErr)
- case err != nil && len(test.expectedErr) != 0:
- if !strings.Contains(err.Error(), test.expectedErr) {
- t.Fatalf("expected %q, got %q", test.expectedErr, err.Error())
- }
- }
- test.validateActions(t, fakeClient.Actions())
- })
- }
- }
|