@@ -1179,8 +1179,6 @@ func TestSanitizeSamegpulist(t *testing.T) {
1179
1179
},
1180
1180
Spec : * getMockPodSpecMultiContSamegpu (),
1181
1181
}
1182
- clientset := fake .NewSimpleClientset (pod )
1183
- gas := NewGASExtender (clientset , false , false , "" )
1184
1182
1185
1183
wrongValueReason := map [string ]string {
1186
1184
"container1,container5" : "Listing absent containers makes gas-same-gpu ignored" ,
@@ -1191,7 +1189,7 @@ func TestSanitizeSamegpulist(t *testing.T) {
1191
1189
1192
1190
Convey ("Ensure no gas-same-gpu annotation returns blank list with no error" ,
1193
1191
t , func () {
1194
- containerNames , err := gas . containersRequestingSamegpu (pod )
1192
+ containerNames , err := containersRequestingSamegpu (pod )
1195
1193
So (len (containerNames ), ShouldEqual , 0 )
1196
1194
So (err , ShouldEqual , nil )
1197
1195
})
@@ -1201,7 +1199,7 @@ func TestSanitizeSamegpulist(t *testing.T) {
1201
1199
1202
1200
Convey (reason ,
1203
1201
t , func () {
1204
- containerNames , err := gas . containersRequestingSamegpu (pod )
1202
+ containerNames , err := containersRequestingSamegpu (pod )
1205
1203
So (len (containerNames ), ShouldEqual , 0 )
1206
1204
So ("malformed annotation" , ShouldEqual , err .Error ())
1207
1205
})
@@ -1211,33 +1209,29 @@ func TestSanitizeSamegpulist(t *testing.T) {
1211
1209
1212
1210
Convey ("Ensure correct annotation returns all listed container names with no error" ,
1213
1211
t , func () {
1214
- containerNames , err := gas . containersRequestingSamegpu (pod )
1212
+ containerNames , err := containersRequestingSamegpu (pod )
1215
1213
So (containerNames , ShouldResemble , map [string ]bool {"container2" : true , "container3" : true })
1216
1214
So (err , ShouldEqual , nil )
1217
1215
})
1218
1216
}
1219
1217
1220
1218
func TestSanitizeSamegpuResourcesRequest (t * testing.T ) {
1221
- pod := getFakePod ()
1222
- clientset := fake .NewSimpleClientset (pod )
1223
- gas := NewGASExtender (clientset , false , false , "" )
1224
-
1225
1219
Convey ("Tiles and monitoring resources are not allowed in same-gpu resourceRequests" ,
1226
1220
t , func () {
1227
1221
// fail because of tiles
1228
1222
samegpuIndexes := map [int ]bool {0 : true }
1229
1223
resourceRequests := []resourceMap {
1230
1224
{"gpu.intel.com/i915" : 1 , "gpu.intel.com/tiles" : 2 },
1231
1225
}
1232
- err := gas . sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1226
+ err := sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1233
1227
So (err .Error (), ShouldEqual , "resources conflict" )
1234
1228
1235
1229
// fail because of monitoring
1236
1230
samegpuIndexes = map [int ]bool {0 : true }
1237
1231
resourceRequests = []resourceMap {
1238
1232
{"gpu.intel.com/i915" : 1 , "gpu.intel.com/i915_monitoring" : 1 },
1239
1233
}
1240
- err = gas . sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1234
+ err = sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1241
1235
So (err .Error (), ShouldEqual , "resources conflict" )
1242
1236
1243
1237
// success
@@ -1248,7 +1242,7 @@ func TestSanitizeSamegpuResourcesRequest(t *testing.T) {
1248
1242
"gpu.intel.com/memory.max" : 8589934592 ,
1249
1243
},
1250
1244
}
1251
- err = gas . sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1245
+ err = sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1252
1246
So (err , ShouldEqual , nil )
1253
1247
})
1254
1248
@@ -1260,7 +1254,7 @@ func TestSanitizeSamegpuResourcesRequest(t *testing.T) {
1260
1254
{"gpu.intel.com/i915" : 1 , "gpu.intel.com/millicores" : 200 },
1261
1255
{"gpu.intel.com/i915" : 2 , "gpu.intel.com/millicores" : 200 },
1262
1256
}
1263
- err := gas . sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1257
+ err := sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1264
1258
So (err .Error (), ShouldEqual , "resources conflict" )
1265
1259
1266
1260
// Failure homogeneous
@@ -1269,7 +1263,7 @@ func TestSanitizeSamegpuResourcesRequest(t *testing.T) {
1269
1263
{"gpu.intel.com/i915" : 2 , "gpu.intel.com/millicores" : 200 },
1270
1264
{"gpu.intel.com/i915" : 2 , "gpu.intel.com/millicores" : 200 },
1271
1265
}
1272
- err = gas . sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1266
+ err = sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1273
1267
So (err .Error (), ShouldEqual , "resources conflict" )
1274
1268
1275
1269
// Success
@@ -1278,7 +1272,7 @@ func TestSanitizeSamegpuResourcesRequest(t *testing.T) {
1278
1272
{"gpu.intel.com/i915" : 1 , "gpu.intel.com/millicores" : 200 },
1279
1273
{"gpu.intel.com/i915" : 1 , "gpu.intel.com/millicores" : 300 },
1280
1274
}
1281
- err = gas . sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1275
+ err = sanitizeSamegpuResourcesRequest (samegpuIndexes , resourceRequests )
1282
1276
So (err , ShouldEqual , nil )
1283
1277
})
1284
1278
}
0 commit comments