mirror of
https://github.com/k3s-io/kubernetes.git
synced 2026-02-22 07:03:28 +00:00
add codes for drop disabled pod fields
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user