268555a30a
Automatic merge from submit-queue. If you want to cherry-pick this change to another branch, please follow the instructions <a href="https://github.com/kubernetes/community/blob/master/contributors/devel/cherry-picks.md">here</a>. feat(fakeclient): push event on watched channel on add/update/delete **What this PR does / why we need it**: This PR enables watch function for kubernetes [fakeclient](1bcf0b0a22/staging/src/k8s.io/client-go/kubernetes/fake/clientset_generated.go (L88)). This fake client add watchReactorFunction by wrapping [watch.NewFake](1bcf0b0a22/staging/src/k8s.io/client-go/kubernetes/fake/clientset_generated.go (L98)) which is a `chan Event` but actually nothing pushes objects into this channel. So all watch function called by fake client will never return or never receive any object. This PR intercepts ReactionFunc of `Create / Update / DeleteActionImpl` and will push the requested object to channel. Which issue(s) this PR fixes (optional, in fixes #<issue number>(, fixes #<issue_number>, ...) format, will close the issue(s) when PR gets merged): Fixes #54075 **Special notes for your reviewer**: **Release note**: ```dev-release-note enable watch function for fake client ```
180 lines
9.3 KiB
Go
180 lines
9.3 KiB
Go
/*
|
|
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 fake
|
|
|
|
import (
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/watch"
|
|
"k8s.io/client-go/discovery"
|
|
fakediscovery "k8s.io/client-go/discovery/fake"
|
|
"k8s.io/client-go/testing"
|
|
clientset "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset"
|
|
admissionregistrationinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/admissionregistration/internalversion"
|
|
fakeadmissionregistrationinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/admissionregistration/internalversion/fake"
|
|
appsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/apps/internalversion"
|
|
fakeappsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/apps/internalversion/fake"
|
|
authenticationinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/authentication/internalversion"
|
|
fakeauthenticationinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/authentication/internalversion/fake"
|
|
authorizationinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/authorization/internalversion"
|
|
fakeauthorizationinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/authorization/internalversion/fake"
|
|
autoscalinginternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/autoscaling/internalversion"
|
|
fakeautoscalinginternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/autoscaling/internalversion/fake"
|
|
batchinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/batch/internalversion"
|
|
fakebatchinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/batch/internalversion/fake"
|
|
certificatesinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/certificates/internalversion"
|
|
fakecertificatesinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/certificates/internalversion/fake"
|
|
coreinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/core/internalversion"
|
|
fakecoreinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/core/internalversion/fake"
|
|
eventsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/events/internalversion"
|
|
fakeeventsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/events/internalversion/fake"
|
|
extensionsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/extensions/internalversion"
|
|
fakeextensionsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/extensions/internalversion/fake"
|
|
networkinginternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/networking/internalversion"
|
|
fakenetworkinginternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/networking/internalversion/fake"
|
|
policyinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/policy/internalversion"
|
|
fakepolicyinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/policy/internalversion/fake"
|
|
rbacinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/rbac/internalversion"
|
|
fakerbacinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/rbac/internalversion/fake"
|
|
schedulinginternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/scheduling/internalversion"
|
|
fakeschedulinginternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/scheduling/internalversion/fake"
|
|
settingsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/settings/internalversion"
|
|
fakesettingsinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/settings/internalversion/fake"
|
|
storageinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/storage/internalversion"
|
|
fakestorageinternalversion "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/storage/internalversion/fake"
|
|
)
|
|
|
|
// NewSimpleClientset returns a clientset that will respond with the provided objects.
|
|
// It's backed by a very simple object tracker that processes creates, updates and deletions as-is,
|
|
// without applying any validations and/or defaults. It shouldn't be considered a replacement
|
|
// for a real clientset and is mostly useful in simple unit tests.
|
|
func NewSimpleClientset(objects ...runtime.Object) *Clientset {
|
|
o := testing.NewObjectTracker(scheme, codecs.UniversalDecoder())
|
|
for _, obj := range objects {
|
|
if err := o.Add(obj); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
fakePtr := testing.Fake{}
|
|
fakePtr.AddReactor("*", "*", testing.ObjectReaction(o))
|
|
fakePtr.AddWatchReactor("*", func(action testing.Action) (handled bool, ret watch.Interface, err error) {
|
|
gvr := action.GetResource()
|
|
ns := action.GetNamespace()
|
|
watch, err := o.Watch(gvr, ns)
|
|
if err != nil {
|
|
return false, nil, err
|
|
}
|
|
return true, watch, nil
|
|
})
|
|
|
|
return &Clientset{fakePtr, &fakediscovery.FakeDiscovery{Fake: &fakePtr}}
|
|
}
|
|
|
|
// Clientset implements clientset.Interface. Meant to be embedded into a
|
|
// struct to get a default implementation. This makes faking out just the method
|
|
// you want to test easier.
|
|
type Clientset struct {
|
|
testing.Fake
|
|
discovery *fakediscovery.FakeDiscovery
|
|
}
|
|
|
|
func (c *Clientset) Discovery() discovery.DiscoveryInterface {
|
|
return c.discovery
|
|
}
|
|
|
|
var _ clientset.Interface = &Clientset{}
|
|
|
|
// Admissionregistration retrieves the AdmissionregistrationClient
|
|
func (c *Clientset) Admissionregistration() admissionregistrationinternalversion.AdmissionregistrationInterface {
|
|
return &fakeadmissionregistrationinternalversion.FakeAdmissionregistration{Fake: &c.Fake}
|
|
}
|
|
|
|
// Core retrieves the CoreClient
|
|
func (c *Clientset) Core() coreinternalversion.CoreInterface {
|
|
return &fakecoreinternalversion.FakeCore{Fake: &c.Fake}
|
|
}
|
|
|
|
// Apps retrieves the AppsClient
|
|
func (c *Clientset) Apps() appsinternalversion.AppsInterface {
|
|
return &fakeappsinternalversion.FakeApps{Fake: &c.Fake}
|
|
}
|
|
|
|
// Authentication retrieves the AuthenticationClient
|
|
func (c *Clientset) Authentication() authenticationinternalversion.AuthenticationInterface {
|
|
return &fakeauthenticationinternalversion.FakeAuthentication{Fake: &c.Fake}
|
|
}
|
|
|
|
// Authorization retrieves the AuthorizationClient
|
|
func (c *Clientset) Authorization() authorizationinternalversion.AuthorizationInterface {
|
|
return &fakeauthorizationinternalversion.FakeAuthorization{Fake: &c.Fake}
|
|
}
|
|
|
|
// Autoscaling retrieves the AutoscalingClient
|
|
func (c *Clientset) Autoscaling() autoscalinginternalversion.AutoscalingInterface {
|
|
return &fakeautoscalinginternalversion.FakeAutoscaling{Fake: &c.Fake}
|
|
}
|
|
|
|
// Batch retrieves the BatchClient
|
|
func (c *Clientset) Batch() batchinternalversion.BatchInterface {
|
|
return &fakebatchinternalversion.FakeBatch{Fake: &c.Fake}
|
|
}
|
|
|
|
// Certificates retrieves the CertificatesClient
|
|
func (c *Clientset) Certificates() certificatesinternalversion.CertificatesInterface {
|
|
return &fakecertificatesinternalversion.FakeCertificates{Fake: &c.Fake}
|
|
}
|
|
|
|
// Events retrieves the EventsClient
|
|
func (c *Clientset) Events() eventsinternalversion.EventsInterface {
|
|
return &fakeeventsinternalversion.FakeEvents{Fake: &c.Fake}
|
|
}
|
|
|
|
// Extensions retrieves the ExtensionsClient
|
|
func (c *Clientset) Extensions() extensionsinternalversion.ExtensionsInterface {
|
|
return &fakeextensionsinternalversion.FakeExtensions{Fake: &c.Fake}
|
|
}
|
|
|
|
// Networking retrieves the NetworkingClient
|
|
func (c *Clientset) Networking() networkinginternalversion.NetworkingInterface {
|
|
return &fakenetworkinginternalversion.FakeNetworking{Fake: &c.Fake}
|
|
}
|
|
|
|
// Policy retrieves the PolicyClient
|
|
func (c *Clientset) Policy() policyinternalversion.PolicyInterface {
|
|
return &fakepolicyinternalversion.FakePolicy{Fake: &c.Fake}
|
|
}
|
|
|
|
// Rbac retrieves the RbacClient
|
|
func (c *Clientset) Rbac() rbacinternalversion.RbacInterface {
|
|
return &fakerbacinternalversion.FakeRbac{Fake: &c.Fake}
|
|
}
|
|
|
|
// Scheduling retrieves the SchedulingClient
|
|
func (c *Clientset) Scheduling() schedulinginternalversion.SchedulingInterface {
|
|
return &fakeschedulinginternalversion.FakeScheduling{Fake: &c.Fake}
|
|
}
|
|
|
|
// Settings retrieves the SettingsClient
|
|
func (c *Clientset) Settings() settingsinternalversion.SettingsInterface {
|
|
return &fakesettingsinternalversion.FakeSettings{Fake: &c.Fake}
|
|
}
|
|
|
|
// Storage retrieves the StorageClient
|
|
func (c *Clientset) Storage() storageinternalversion.StorageInterface {
|
|
return &fakestorageinternalversion.FakeStorage{Fake: &c.Fake}
|
|
}
|