add codes for drop disabled pod fields

This commit is contained in:
HirazawaUi
2025-07-02 08:09:29 +08:00
parent 8d65e1e98e
commit c35e4ad2b2
2 changed files with 97 additions and 0 deletions

View File

@@ -764,6 +764,11 @@ func dropDisabledFields(
}
}
if !utilfeature.DefaultFeatureGate.Enabled(features.HostnameOverride) && !setHostnameOverrideInUse(oldPodSpec) {
// Set HostnameOverride to nil only if feature is disabled and it is not used
podSpec.HostnameOverride = nil
}
dropFileKeyRefInUse(podSpec, oldPodSpec)
dropPodLifecycleSleepAction(podSpec, oldPodSpec)
dropImageVolumes(podSpec, oldPodSpec)
@@ -771,6 +776,14 @@ func dropDisabledFields(
dropContainerStopSignals(podSpec, oldPodSpec)
}
// setHostnameOverrideInUse returns true if any pod's spec defines HostnameOverride field.
func setHostnameOverrideInUse(podSpec *api.PodSpec) bool {
if podSpec == nil || podSpec.HostnameOverride == nil {
return false
}
return true
}
func dropFileKeyRefInUse(podSpec, oldPodSpec *api.PodSpec) {
if utilfeature.DefaultFeatureGate.Enabled(features.EnvFiles) || podFileKeyRefInUse(oldPodSpec) {
return

View File

@@ -5199,6 +5199,90 @@ func TestHasAPIReferences(t *testing.T) {
}
}
func TestDropHostnameOverride(t *testing.T) {
podWithoutHostnameOverride := func() *api.Pod {
return &api.Pod{
Spec: api.PodSpec{},
}
}
podWithHostnameOverride := func() *api.Pod {
return &api.Pod{
Spec: api.PodSpec{
HostnameOverride: ptr.To("custom-hostname"),
},
}
}
oldPodInfo := []struct {
description string
hasHostnameOverride bool
pod func() *api.Pod
}{
{
description: "with HostnameOverride=true",
hasHostnameOverride: true,
pod: podWithHostnameOverride,
},
{
description: "with HostnameOverride=nil",
hasHostnameOverride: false,
pod: podWithoutHostnameOverride,
},
}
newPodInfo := []struct {
description string
hasHostnameOverride bool
pod func() *api.Pod
}{
{
description: "with HostnameOverride=true",
hasHostnameOverride: true,
pod: podWithHostnameOverride,
},
{
description: "with HostnameOverride=nil",
hasHostnameOverride: false,
pod: podWithoutHostnameOverride,
},
}
for _, enabled := range []bool{true, false} {
for _, oldPodInfo := range oldPodInfo {
for _, newPodInfo := range newPodInfo {
oldPodHasHostnameOverride, oldPod := oldPodInfo.hasHostnameOverride, oldPodInfo.pod()
newPodHasHostnameOverride, newPod := newPodInfo.hasHostnameOverride, newPodInfo.pod()
t.Run(fmt.Sprintf("feature enabled=%v, old pod %v, new pod %v", enabled, oldPodInfo.description, newPodInfo.description), func(t *testing.T) {
featuregatetesting.SetFeatureGateDuringTest(t, utilfeature.DefaultFeatureGate, features.HostnameOverride, enabled)
DropDisabledPodFields(newPod, oldPod)
if !reflect.DeepEqual(oldPod, oldPodInfo.pod()) {
t.Errorf("old pod changed: %v", cmp.Diff(oldPod, oldPodInfo.pod()))
}
switch {
case enabled || oldPodHasHostnameOverride:
if !reflect.DeepEqual(newPod, newPodInfo.pod()) {
t.Errorf("new pod changed: %v", cmp.Diff(newPod, newPodInfo.pod()))
}
case newPodHasHostnameOverride:
if exp := podWithoutHostnameOverride(); !reflect.DeepEqual(newPod, exp) {
t.Errorf("new pod had HostnameOverride: %v", cmp.Diff(newPod, exp))
}
default:
if !reflect.DeepEqual(newPod, newPodInfo.pod()) {
t.Errorf("new pod changed: %v", cmp.Diff(newPod, newPodInfo.pod()))
}
}
})
}
}
}
}
func TestDropFileKeyRefInUse(t *testing.T) {
testCases := []struct {
name string