add test case

This commit is contained in:
wackxu 2017-09-08 14:58:38 +08:00
parent eda3db550b
commit f05d4b7f1c

View File

@ -94,6 +94,24 @@ func TestPriorityMetadata(t *testing.T) {
Tolerations: tolerations,
},
}
podWithAffinityAndRequests := &v1.Pod{
Spec: v1.PodSpec{
Containers: []v1.Container{
{
Name: "container",
Image: "image",
ImagePullPolicy: "Always",
Resources: v1.ResourceRequirements{
Requests: v1.ResourceList{
v1.ResourceCPU: resource.MustParse("200m"),
v1.ResourceMemory: resource.MustParse("2000"),
},
},
},
},
Affinity: podAffinity,
},
}
tests := []struct {
pod *v1.Pod
test string
@ -122,6 +140,15 @@ func TestPriorityMetadata(t *testing.T) {
},
test: "Produce a priorityMetadata with specified requests",
},
{
pod: podWithAffinityAndRequests,
expected: &priorityMetadata{
nonZeroRequest: specifiedReqs,
podTolerations: nil,
affinity: podAffinity,
},
test: "Produce a priorityMetadata with specified requests",
},
}
for _, test := range tests {
ptData := PriorityMetadata(test.pod, nil)