Rename and add unit test for ImageSizes

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

pkg/scheduler/schedulercache/node_info.go

@@ -293,8 +293,8 @@ func (n *NodeInfo) UsedPorts() util.HostPortInfo {
 	return n.usedPorts
 }
 
-// Images returns the image size information on this node.
-func (n *NodeInfo) Images() map[string]int64 {
+// ImageSizes returns the image size information on this node.
+func (n *NodeInfo) ImageSizes() map[string]int64 {
 	if n == nil {
 		return nil
 	}

pkg/scheduler/schedulercache/node_info_test.go

@@ -235,6 +235,46 @@ func TestSetMaxResource(t *testing.T) {
 	}
 }
 
+func TestImageSizes(t *testing.T) {
+	ni := fakeNodeInfo()
+	ni.node = &v1.Node{
+		ObjectMeta: metav1.ObjectMeta{
+			Name: "test-node",
+		},
+		Status: v1.NodeStatus{
+			Images: []v1.ContainerImage{
+				{
+					Names: []string{
+						"gcr.io/10",
+						"gcr.io/10:v1",
+					},
+					SizeBytes: int64(10 * 1024 * 1024),
+				},
+				{
+					Names: []string{
+						"gcr.io/50",
+						"gcr.io/50:v1",
+					},
+					SizeBytes: int64(50 * 1024 * 1024),
+				},
+			},
+		},
+	}
+
+	ni.updateImageSizes()
+	expected := map[string]int64{
+		"gcr.io/10":    10 * 1024 * 1024,
+		"gcr.io/10:v1": 10 * 1024 * 1024,
+		"gcr.io/50":    50 * 1024 * 1024,
+		"gcr.io/50:v1": 50 * 1024 * 1024,
+	}
+
+	imageSizes := ni.ImageSizes()
+	if !reflect.DeepEqual(expected, imageSizes) {
+		t.Errorf("expected: %#v, got: %#v", expected, imageSizes)
+	}
+}
+
 func TestNewNodeInfo(t *testing.T) {
 	nodeName := "test-node"
 	pods := []*v1.Pod{