...
1
16
17
18
19 package fake
20
21 import (
22 clientset "github.com/kubernetes-csi/external-snapshotter/client/v4/clientset/versioned"
23 snapshotv1 "github.com/kubernetes-csi/external-snapshotter/client/v4/clientset/versioned/typed/volumesnapshot/v1"
24 fakesnapshotv1 "github.com/kubernetes-csi/external-snapshotter/client/v4/clientset/versioned/typed/volumesnapshot/v1/fake"
25 snapshotv1beta1 "github.com/kubernetes-csi/external-snapshotter/client/v4/clientset/versioned/typed/volumesnapshot/v1beta1"
26 fakesnapshotv1beta1 "github.com/kubernetes-csi/external-snapshotter/client/v4/clientset/versioned/typed/volumesnapshot/v1beta1/fake"
27 "k8s.io/apimachinery/pkg/runtime"
28 "k8s.io/apimachinery/pkg/watch"
29 "k8s.io/client-go/discovery"
30 fakediscovery "k8s.io/client-go/discovery/fake"
31 "k8s.io/client-go/testing"
32 )
33
34
35
36
37
38 func NewSimpleClientset(objects ...runtime.Object) *Clientset {
39 o := testing.NewObjectTracker(scheme, codecs.UniversalDecoder())
40 for _, obj := range objects {
41 if err := o.Add(obj); err != nil {
42 panic(err)
43 }
44 }
45
46 cs := &Clientset{tracker: o}
47 cs.discovery = &fakediscovery.FakeDiscovery{Fake: &cs.Fake}
48 cs.AddReactor("*", "*", testing.ObjectReaction(o))
49 cs.AddWatchReactor("*", func(action testing.Action) (handled bool, ret watch.Interface, err error) {
50 gvr := action.GetResource()
51 ns := action.GetNamespace()
52 watch, err := o.Watch(gvr, ns)
53 if err != nil {
54 return false, nil, err
55 }
56 return true, watch, nil
57 })
58
59 return cs
60 }
61
62
63
64
65 type Clientset struct {
66 testing.Fake
67 discovery *fakediscovery.FakeDiscovery
68 tracker testing.ObjectTracker
69 }
70
71 func (c *Clientset) Discovery() discovery.DiscoveryInterface {
72 return c.discovery
73 }
74
75 func (c *Clientset) Tracker() testing.ObjectTracker {
76 return c.tracker
77 }
78
79 var _ clientset.Interface = &Clientset{}
80
81
82 func (c *Clientset) SnapshotV1beta1() snapshotv1beta1.SnapshotV1beta1Interface {
83 return &fakesnapshotv1beta1.FakeSnapshotV1beta1{Fake: &c.Fake}
84 }
85
86
87 func (c *Clientset) SnapshotV1() snapshotv1.SnapshotV1Interface {
88 return &fakesnapshotv1.FakeSnapshotV1{Fake: &c.Fake}
89 }
90
View as plain text