Rename and add unit test for ImageSizes

2018-05-24 23:29:23 -07:00
parent d7c40cf69e
commit 096dda3768
3 changed files with 43 additions and 3 deletions
--- a/pkg/scheduler/algorithm/priorities/image_locality.go
+++ b/pkg/scheduler/algorithm/priorities/image_locality.go
@@ -72,7 +72,7 @@ func calculateScoreFromSize(sumSize int64) int {
 func totalImageSize(nodeInfo *schedulercache.NodeInfo, containers []v1.Container) int64 {
 	var total int64

-	imageSizes := nodeInfo.Images()
+	imageSizes := nodeInfo.ImageSizes()
 	for _, container := range containers {
 		if size, ok := imageSizes[container.Image]; ok {
 			total += size