@@ -27,42 +27,10 @@ import (
27
27
"k8s.io/apimachinery/pkg/api/resource"
28
28
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
29
29
"k8s.io/client-go/kubernetes/fake"
30
- corev1 "k8s.io/client-go/kubernetes/typed/core/v1"
31
- fakecorev1 "k8s.io/client-go/kubernetes/typed/core/v1/fake"
32
30
"k8s.io/kubelet/pkg/apis/deviceplugin/v1beta1"
33
31
podresourcesv1 "k8s.io/kubelet/pkg/apis/podresources/v1"
34
32
)
35
33
36
- // mockClient implements enough of k8s API for the resource manager tests.
37
- type mockClient struct {
38
- fake.Clientset
39
- mockCoreV1
40
- }
41
-
42
- func (m * mockClient ) CoreV1 () corev1.CoreV1Interface {
43
- return m
44
- }
45
-
46
- type mockCoreV1 struct {
47
- fakecorev1.FakeCoreV1
48
- mockPods
49
- }
50
-
51
- func (m * mockCoreV1 ) Pods (namespace string ) corev1.PodInterface {
52
- return m
53
- }
54
-
55
- type mockPods struct {
56
- fakecorev1.FakePods
57
- pods []v1.Pod
58
- }
59
-
60
- func (m * mockPods ) List (ctx context.Context , opts metav1.ListOptions ) (* v1.PodList , error ) {
61
- return & v1.PodList {
62
- Items : m .pods ,
63
- }, nil
64
- }
65
-
66
34
type mockPodResources struct {
67
35
pods []v1.Pod
68
36
}
@@ -99,8 +67,15 @@ func newMockResourceManager(pods []v1.Pod) ResourceManager {
99
67
os .Exit (1 )
100
68
}
101
69
102
- mc := & mockClient {}
103
- mc .mockCoreV1 .mockPods .pods = pods
70
+ mc := fake .NewClientset ()
71
+
72
+ for _ , p := range pods {
73
+ _ , err = mc .CoreV1 ().Pods (p .Namespace ).Create (context .Background (), & p , metav1.CreateOptions {})
74
+ if err != nil {
75
+ fmt .Printf ("failed to Create Pod: %v\n " , err )
76
+ }
77
+ }
78
+
104
79
rm := resourceManager {
105
80
clientset : mc ,
106
81
nodeName : "TestNode" ,
0 commit comments