@@ -22,7 +22,6 @@ import (
2222
2323 "github.com/stretchr/testify/assert"
2424 corev1 "k8s.io/api/core/v1"
25- v1 "k8s.io/api/core/v1"
2625 "k8s.io/apimachinery/pkg/api/resource"
2726 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
2827 apiruntime "k8s.io/apimachinery/pkg/runtime"
@@ -108,7 +107,7 @@ func NodeResourcesPluginFactoryProxy(factoryFn frameworkruntime.PluginFactory, p
108107func TestPlugin_Score (t * testing.T ) {
109108
110109 var v1beta2args v1beta3.NodeResourcesFitPlusArgs
111- v1beta2args .Resources = map [v1 .ResourceName ]v1beta3.ResourcesType {
110+ v1beta2args .Resources = map [corev1 .ResourceName ]v1beta3.ResourcesType {
112111 "nvidia.com/gpu" : {Type : k8sConfig .MostAllocated , Weight : 2 },
113112 "cpu" : {Type : k8sConfig .LeastAllocated , Weight : 1 },
114113 "memory" : {Type : k8sConfig .LeastAllocated , Weight : 1 },
@@ -131,14 +130,16 @@ func TestPlugin_Score(t *testing.T) {
131130 },
132131 Status : corev1.NodeStatus {
133132 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" ),
137137 },
138138 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" ),
142143 },
143144 },
144145 },
@@ -148,16 +149,18 @@ func TestPlugin_Score(t *testing.T) {
148149 },
149150 Status : corev1.NodeStatus {
150151 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" ),
155157 },
156158 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" ),
161164 },
162165 },
163166 },
@@ -237,14 +240,16 @@ func TestPlugin_Score(t *testing.T) {
237240 Name : "test-container" ,
238241 Resources : corev1.ResourceRequirements {
239242 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" ),
243247 },
244248 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" ),
248253 },
249254 },
250255 },
0 commit comments