@@ -22,7 +22,6 @@ import (
22
22
23
23
"github.com/stretchr/testify/assert"
24
24
corev1 "k8s.io/api/core/v1"
25
- v1 "k8s.io/api/core/v1"
26
25
"k8s.io/apimachinery/pkg/api/resource"
27
26
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
28
27
apiruntime "k8s.io/apimachinery/pkg/runtime"
@@ -108,7 +107,7 @@ func NodeResourcesPluginFactoryProxy(factoryFn frameworkruntime.PluginFactory, p
108
107
func TestPlugin_Score (t * testing.T ) {
109
108
110
109
var v1beta2args v1beta3.NodeResourcesFitPlusArgs
111
- v1beta2args .Resources = map [v1 .ResourceName ]v1beta3.ResourcesType {
110
+ v1beta2args .Resources = map [corev1 .ResourceName ]v1beta3.ResourcesType {
112
111
"nvidia.com/gpu" : {Type : k8sConfig .MostAllocated , Weight : 2 },
113
112
"cpu" : {Type : k8sConfig .LeastAllocated , Weight : 1 },
114
113
"memory" : {Type : k8sConfig .LeastAllocated , Weight : 1 },
@@ -131,14 +130,16 @@ func TestPlugin_Score(t *testing.T) {
131
130
},
132
131
Status : corev1.NodeStatus {
133
132
Allocatable : corev1.ResourceList {
134
- corev1 .ResourceCPU : resource .MustParse ("96" ),
135
- corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
136
- "nvidia.com/gpu" : resource .MustParse ("8" ),
133
+ corev1 .ResourceCPU : resource .MustParse ("96" ),
134
+ corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
135
+ "nvidia.com/gpu" : resource .MustParse ("8" ),
136
+ corev1 .ResourceEphemeralStorage : resource .MustParse ("100Gi" ),
137
137
},
138
138
Capacity : corev1.ResourceList {
139
- corev1 .ResourceCPU : resource .MustParse ("96" ),
140
- corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
141
- "nvidia.com/gpu" : resource .MustParse ("8" ),
139
+ corev1 .ResourceCPU : resource .MustParse ("96" ),
140
+ corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
141
+ "nvidia.com/gpu" : resource .MustParse ("8" ),
142
+ corev1 .ResourceEphemeralStorage : resource .MustParse ("100Gi" ),
142
143
},
143
144
},
144
145
},
@@ -148,16 +149,18 @@ func TestPlugin_Score(t *testing.T) {
148
149
},
149
150
Status : corev1.NodeStatus {
150
151
Allocatable : corev1.ResourceList {
151
- corev1 .ResourceCPU : resource .MustParse ("96" ),
152
- corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
153
- "nvidia.com/gpu" : resource .MustParse ("8" ),
154
- "xx.xx/xx" : resource .MustParse ("8" ),
152
+ corev1 .ResourceCPU : resource .MustParse ("96" ),
153
+ corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
154
+ "nvidia.com/gpu" : resource .MustParse ("8" ),
155
+ "xx.xx/xx" : resource .MustParse ("8" ),
156
+ corev1 .ResourceEphemeralStorage : resource .MustParse ("100Gi" ),
155
157
},
156
158
Capacity : corev1.ResourceList {
157
- corev1 .ResourceCPU : resource .MustParse ("96" ),
158
- corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
159
- "nvidia.com/gpu" : resource .MustParse ("8" ),
160
- "xx.xx/xx" : resource .MustParse ("8" ),
159
+ corev1 .ResourceCPU : resource .MustParse ("96" ),
160
+ corev1 .ResourceMemory : resource .MustParse ("512Gi" ),
161
+ "nvidia.com/gpu" : resource .MustParse ("8" ),
162
+ "xx.xx/xx" : resource .MustParse ("8" ),
163
+ corev1 .ResourceEphemeralStorage : resource .MustParse ("100Gi" ),
161
164
},
162
165
},
163
166
},
@@ -237,14 +240,16 @@ func TestPlugin_Score(t *testing.T) {
237
240
Name : "test-container" ,
238
241
Resources : corev1.ResourceRequirements {
239
242
Limits : corev1.ResourceList {
240
- corev1 .ResourceCPU : resource .MustParse ("16" ),
241
- corev1 .ResourceMemory : resource .MustParse ("32Gi" ),
242
- "nvidia.com/gpu" : resource .MustParse ("2" ),
243
+ corev1 .ResourceCPU : resource .MustParse ("16" ),
244
+ corev1 .ResourceMemory : resource .MustParse ("32Gi" ),
245
+ "nvidia.com/gpu" : resource .MustParse ("2" ),
246
+ corev1 .ResourceEphemeralStorage : resource .MustParse ("10Gi" ),
243
247
},
244
248
Requests : corev1.ResourceList {
245
- corev1 .ResourceCPU : resource .MustParse ("16" ),
246
- corev1 .ResourceMemory : resource .MustParse ("32Gi" ),
247
- "nvidia.com/gpu" : resource .MustParse ("2" ),
249
+ corev1 .ResourceCPU : resource .MustParse ("16" ),
250
+ corev1 .ResourceMemory : resource .MustParse ("32Gi" ),
251
+ "nvidia.com/gpu" : resource .MustParse ("2" ),
252
+ corev1 .ResourceEphemeralStorage : resource .MustParse ("10Gi" ),
248
253
},
249
254
},
250
255
},
0 commit comments