...
1
16
17
18
19 package fake
20
21 import (
22 "k8s.io/apimachinery/pkg/runtime"
23 "k8s.io/apimachinery/pkg/watch"
24 "k8s.io/client-go/discovery"
25 fakediscovery "k8s.io/client-go/discovery/fake"
26 "k8s.io/client-go/testing"
27 clientset "kubevirt.io/client-go/generated/containerized-data-importer/clientset/versioned"
28 cdiv1beta1 "kubevirt.io/client-go/generated/containerized-data-importer/clientset/versioned/typed/core/v1beta1"
29 fakecdiv1beta1 "kubevirt.io/client-go/generated/containerized-data-importer/clientset/versioned/typed/core/v1beta1/fake"
30 uploadv1beta1 "kubevirt.io/client-go/generated/containerized-data-importer/clientset/versioned/typed/upload/v1beta1"
31 fakeuploadv1beta1 "kubevirt.io/client-go/generated/containerized-data-importer/clientset/versioned/typed/upload/v1beta1/fake"
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) CdiV1beta1() cdiv1beta1.CdiV1beta1Interface {
83 return &fakecdiv1beta1.FakeCdiV1beta1{Fake: &c.Fake}
84 }
85
86
87 func (c *Clientset) UploadV1beta1() uploadv1beta1.UploadV1beta1Interface {
88 return &fakeuploadv1beta1.FakeUploadV1beta1{Fake: &c.Fake}
89 }
90
View as plain text