install watch reaction function to fake client

Signed-off-by: Ashish Naware <ashishnaware3@gmail.com>

Author: AshishNaware

testing/aliases.go

@@ -22,6 +22,8 @@ import (
 
 type Reactor = clientgotesting.Reactor
 type ReactionFunc = clientgotesting.ReactionFunc
+type WatchReactor = clientgotesting.WatchReactor
+type WatchReactionFunc = clientgotesting.WatchReactionFunc
 
 type Action = clientgotesting.Action
 type GetAction = clientgotesting.GetAction

testing/client.go

@@ -50,6 +50,7 @@ type clientWrapper struct {
 	StatusPatchActions      []PatchAction
 	genCount                int
 	reactionChain           []Reactor
+	watchReactionChain      []WatchReactor
 }
 
 var _ TestClient = (*clientWrapper)(nil)
@@ -67,6 +68,7 @@ func NewFakeClientWrapper(client client.Client, tracker clientgotesting.ObjectTr
 		StatusPatchActions:      []PatchAction{},
 		genCount:                0,
 		reactionChain:           []Reactor{},
+		watchReactionChain:      []WatchReactor{},
 	}
 	// generate names on create
 	c.AddReactor("create", "*", func(action Action) (bool, runtime.Object, error) {
@@ -115,6 +117,10 @@ func (w *clientWrapper) PrependReactor(verb, kind string, reaction ReactionFunc)
 	w.reactionChain = append([]Reactor{&clientgotesting.SimpleReactor{Verb: verb, Resource: kind, Reaction: reaction}}, w.reactionChain...)
 }
 
+func (w *clientWrapper) PrependWatchReactor(kind string, reaction WatchReactionFunc) {
+	w.watchReactionChain = append([]WatchReactor{&clientgotesting.SimpleWatchReactor{Resource: kind, Reaction: reaction}}, w.watchReactionChain...)
+}
+
 func (w *clientWrapper) objmeta(obj runtime.Object) (schema.GroupVersionResource, string, string, error) {
 	objref, err := ref.GetReference(w.Scheme(), obj)
 	if err != nil {

testing/config.go

@@ -69,6 +69,9 @@ type ExpectConfig struct {
 	// WithReactors installs each ReactionFunc into each fake clientset. ReactionFuncs intercept
 	// each call to the clientset providing the ability to mutate the resource or inject an error.
 	WithReactors []ReactionFunc
+	// WithWatchReactors installs WatchReactionFunc into the fake clientset. This provides ability
+	// to simulate events in the watcher.
+	WithWatchReactors []WatchReactionFunc
 	// GivenTracks provide a set of tracked resources to seed the tracker with
 	GivenTracks []TrackRequest
 
@@ -119,6 +122,11 @@ func (c *ExpectConfig) init() {
 			reactor := c.WithReactors[len(c.WithReactors)-1-i]
 			c.client.PrependReactor("*", "*", reactor)
 		}
+		for i := range c.WithWatchReactors {
+			// in reverse order since we prepend
+			watchReactor := c.WithWatchReactors[len(c.WithWatchReactors)-1-i]
+			c.client.PrependWatchReactor("*", watchReactor)
+		}
 		c.apiReader = c.createClient(apiGivenObjects, c.StatusSubResourceTypes)
 		c.recorder = &eventRecorder{
 			events: []Event{},