Fix some unit test error

This commit is contained in:
vivzbansal 2024-11-08 17:13:00 +00:00
parent 1cf4587277
commit 6cf5b80c64
2 changed files with 43 additions and 74 deletions

View File

@ -25601,24 +25601,6 @@ func TestValidatePodResize(t *testing.T) {
),
)...)
}
mkPodWithInitCtrs := func(req, lim core.ResourceList, tweaks ...podtest.TweakContainer) *core.Pod {
tweaks = append(tweaks, podtest.SetContainerRestartPolicy(containerRestartPolicyAlways))
return podtest.MakePod("pod",
podtest.SetInitContainers(
podtest.MakeContainer(
"restartable-init",
append(tweaks,
podtest.SetContainerResources(
core.ResourceRequirements{
Requests: req,
Limits: lim,
},
),
)...,
),
),
)
}
mkPodWithInitContainers := func(req, lim core.ResourceList, restartPolicy core.ContainerRestartPolicy, tweaks ...podtest.Tweak) *core.Pod {
return podtest.MakePod("pod", append(tweaks,
@ -25638,11 +25620,10 @@ func TestValidatePodResize(t *testing.T) {
}
tests := []struct {
test string
old *core.Pod
new *core.Pod
isSideCarCtr bool
err string
test string
old *core.Pod
new *core.Pod
err string
}{
{
test: "pod-level resources with container cpu limit change",
@ -25930,9 +25911,8 @@ func TestValidatePodResize(t *testing.T) {
test: "Pod QoS unchanged, burstable -> burstable, remove cpu and memory requests",
old: mkPodWithInitContainers(getResources("100m", "100Mi", "", ""), getResources("100m", "", "", ""), ""),
new: mkPodWithInitContainers(core.ResourceList{}, getResources("100m", "", "", ""), ""),
err: "spec: Forbidden: only cpu and memory resources are mutable",
},
{
err: "spec: Forbidden: cpu and memory resources for only sidecar containers are mutable",
}, {
test: "Pod with nil Resource field in Status",
old: mkPod(core.ResourceList{}, getResources("100m", "0", "1Gi", ""), podtest.SetStatus(core.PodStatus{
ContainerStatuses: []core.ContainerStatus{{
@ -25999,48 +25979,40 @@ func TestValidatePodResize(t *testing.T) {
})),
new: mkPod(core.ResourceList{}, getResources("200m", "0", "1Gi", "")),
err: "",
},
{
test: "cpu limit change for sidecar containers",
old: mkPodWithInitCtrs(core.ResourceList{}, getResources("100m", "0", "1Gi", "")),
new: mkPodWithInitCtrs(core.ResourceList{}, getResources("200m", "0", "1Gi", "")),
isSideCarCtr: true,
err: "",
}, {
test: "memory limit change for sidecar containers",
old: mkPodWithInitCtrs(core.ResourceList{}, getResources("100m", "200Mi", "", "")),
new: mkPodWithInitCtrs(core.ResourceList{}, getResources("100m", "100Mi", "", "")),
isSideCarCtr: true,
err: "",
test: "cpu limit change for sidecar containers",
old: mkPodWithInitContainers(core.ResourceList{}, getResources("100m", "0", "1Gi", ""), core.ContainerRestartPolicyAlways),
new: mkPodWithInitContainers(core.ResourceList{}, getResources("200m", "0", "1Gi", ""), core.ContainerRestartPolicyAlways),
err: "",
}, {
test: "storage limit change for sidecar containers",
old: mkPodWithInitCtrs(core.ResourceList{}, getResources("100m", "100Mi", "2Gi", "")),
new: mkPodWithInitCtrs(core.ResourceList{}, getResources("100m", "100Mi", "1Gi", "")),
isSideCarCtr: true,
err: "spec: Forbidden: cpu and memory resources for only sidecar containers are mutable",
test: "memory limit change for sidecar containers",
old: mkPodWithInitContainers(core.ResourceList{}, getResources("100m", "200Mi", "", ""), core.ContainerRestartPolicyAlways),
new: mkPodWithInitContainers(core.ResourceList{}, getResources("100m", "100Mi", "", ""), core.ContainerRestartPolicyAlways),
err: "",
}, {
test: "cpu request change for sidecar containers",
old: mkPodWithInitCtrs(getResources("200m", "0", "", ""), core.ResourceList{}),
new: mkPodWithInitCtrs(getResources("100m", "0", "", ""), core.ResourceList{}),
isSideCarCtr: true,
err: "",
test: "storage limit change for sidecar containers",
old: mkPodWithInitContainers(core.ResourceList{}, getResources("100m", "100Mi", "2Gi", ""), core.ContainerRestartPolicyAlways),
new: mkPodWithInitContainers(core.ResourceList{}, getResources("100m", "100Mi", "1Gi", ""), core.ContainerRestartPolicyAlways),
err: "spec: Forbidden: cpu and memory resources for only sidecar containers are mutable",
}, {
test: "memory request change for sidecar containers",
old: mkPodWithInitCtrs(getResources("0", "100Mi", "", ""), core.ResourceList{}),
new: mkPodWithInitCtrs(getResources("0", "200Mi", "", ""), core.ResourceList{}),
isSideCarCtr: true,
err: "",
test: "cpu request change for sidecar containers",
old: mkPodWithInitContainers(getResources("200m", "0", "", ""), core.ResourceList{}, core.ContainerRestartPolicyAlways),
new: mkPodWithInitContainers(getResources("100m", "0", "", ""), core.ResourceList{}, core.ContainerRestartPolicyAlways),
err: "",
}, {
test: "storage request change for sidecar containers",
old: mkPodWithInitCtrs(getResources("100m", "0", "1Gi", ""), core.ResourceList{}),
new: mkPodWithInitCtrs(getResources("100m", "0", "2Gi", ""), core.ResourceList{}),
isSideCarCtr: true,
err: "spec: Forbidden: cpu and memory resources for only sidecar containers are mutable",
test: "memory request change for sidecar containers",
old: mkPodWithInitContainers(getResources("0", "100Mi", "", ""), core.ResourceList{}, core.ContainerRestartPolicyAlways),
new: mkPodWithInitContainers(getResources("0", "200Mi", "", ""), core.ResourceList{}, core.ContainerRestartPolicyAlways),
err: "",
}, {
test: "storage request change for sidecar containers",
old: mkPodWithInitContainers(getResources("100m", "0", "1Gi", ""), core.ResourceList{}, core.ContainerRestartPolicyAlways),
new: mkPodWithInitContainers(getResources("100m", "0", "2Gi", ""), core.ResourceList{}, core.ContainerRestartPolicyAlways),
err: "spec: Forbidden: cpu and memory resources for only sidecar containers are mutable",
},
}
for _, test := range tests {
featuregatetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.SidecarContainers, test.isSideCarCtr)
test.new.ObjectMeta.ResourceVersion = "1"
test.old.ObjectMeta.ResourceVersion = "1"

View File

@ -3651,22 +3651,19 @@ func TestIsPodResizeInProgress(t *testing.T) {
name: "resizing cpu limit",
status: mkPodStatus(resizeCPULimitC1Status, steadyStateC2Status),
expectResize: true,
}, {
name: "resizing cpu request for restartable init container",
status: mkPodStatus(steadyStateC1Status, steadyStateC2Status, resizeCPUReqC3Status),
expectResize: true,
}, {
name: "resizing memory limit for restartable init container",
status: mkPodStatus(steadyStateC1Status, steadyStateC2Status, resizeMemLimitC3Status),
expectResize: true,
}, {
name: "non-restartable init container should be ignored",
status: mkPodStatus(steadyStateC1Status, steadyStateC2Status, steadyStateC3Status, resizeCPUReqC4Status),
expectResize: false,
},
{
name: "resizing cpu request for restartable init container",
status: mkPodStatus(steadyStateC1Status, steadyStateC2Status, resizeCPUReqC3Status),
expectResize: true,
},
{
name: "resizing memory limit for restartable init container",
status: mkPodStatus(steadyStateC1Status, steadyStateC2Status, resizeMemLimitC3Status),
expectResize: true,
},
{
name: "non-restartable init container should be ignored",
status: mkPodStatus(steadyStateC1Status, steadyStateC2Status, steadyStateC3Status, resizeCPUReqC4Status),
expectResize: false,
},
}
for _, test := range tests {