@@ -18,6 +18,7 @@ package pod
1818
1919import (
2020 "fmt"
21+ "os"
2122 "path/filepath"
2223 "strings"
2324 "testing"
@@ -813,14 +814,14 @@ func TestShouldOverrideHomeEnv(t *testing.T) {
813814 }{{
814815 description : "Default behaviour: A missing disable-home-env-overwrite flag should result in true" ,
815816 configMap : & corev1.ConfigMap {
816- ObjectMeta : metav1.ObjectMeta {Name : featureFlagConfigMapName , Namespace : system .GetNamespace ()},
817+ ObjectMeta : metav1.ObjectMeta {Name : GetFeatureFlagsConfigName () , Namespace : system .GetNamespace ()},
817818 Data : map [string ]string {},
818819 },
819820 expected : true ,
820821 }, {
821822 description : "Setting disable-home-env-overwrite to false should result in true" ,
822823 configMap : & corev1.ConfigMap {
823- ObjectMeta : metav1.ObjectMeta {Name : featureFlagConfigMapName , Namespace : system .GetNamespace ()},
824+ ObjectMeta : metav1.ObjectMeta {Name : GetFeatureFlagsConfigName () , Namespace : system .GetNamespace ()},
824825 Data : map [string ]string {
825826 featureFlagDisableHomeEnvKey : "false" ,
826827 },
@@ -829,7 +830,7 @@ func TestShouldOverrideHomeEnv(t *testing.T) {
829830 }, {
830831 description : "Setting disable-home-env-overwrite to true should result in false" ,
831832 configMap : & corev1.ConfigMap {
832- ObjectMeta : metav1.ObjectMeta {Name : featureFlagConfigMapName , Namespace : system .GetNamespace ()},
833+ ObjectMeta : metav1.ObjectMeta {Name : GetFeatureFlagsConfigName () , Namespace : system .GetNamespace ()},
833834 Data : map [string ]string {
834835 featureFlagDisableHomeEnvKey : "true" ,
835836 },
@@ -847,6 +848,37 @@ func TestShouldOverrideHomeEnv(t *testing.T) {
847848 }
848849}
849850
851+ func TestGetFeatureFlagsConfigName (t * testing.T ) {
852+ for _ , tc := range []struct {
853+ description string
854+ featureFlagEnvValue string
855+ expected string
856+ }{{
857+ description : "Feature flags config value not set" ,
858+ featureFlagEnvValue : "" ,
859+ expected : "feature-flags" ,
860+ }, {
861+ description : "Feature flags config value set" ,
862+ featureFlagEnvValue : "feature-flags-test" ,
863+ expected : "feature-flags-test" ,
864+ }} {
865+ t .Run (tc .description , func (t * testing.T ) {
866+ original := os .Getenv ("CONFIG_FEATURE_FLAGS_NAME" )
867+ defer t .Cleanup (func () {
868+ os .Setenv ("CONFIG_FEATURE_FLAGS_NAME" , original )
869+ })
870+ if tc .featureFlagEnvValue != "" {
871+ os .Setenv ("CONFIG_FEATURE_FLAGS_NAME" , tc .featureFlagEnvValue )
872+ }
873+ got := GetFeatureFlagsConfigName ()
874+ want := tc .expected
875+ if got != want {
876+ t .Errorf ("GetFeatureFlagsConfigName() = %s, want %s" , got , want )
877+ }
878+ })
879+ }
880+ }
881+
850882func TestShouldOverrideWorkingDir (t * testing.T ) {
851883 for _ , tc := range []struct {
852884 description string
@@ -855,14 +887,14 @@ func TestShouldOverrideWorkingDir(t *testing.T) {
855887 }{{
856888 description : "Default behaviour: A missing disable-working-directory-overwrite flag should result in true" ,
857889 configMap : & corev1.ConfigMap {
858- ObjectMeta : metav1.ObjectMeta {Name : featureFlagConfigMapName , Namespace : system .GetNamespace ()},
890+ ObjectMeta : metav1.ObjectMeta {Name : GetFeatureFlagsConfigName () , Namespace : system .GetNamespace ()},
859891 Data : map [string ]string {},
860892 },
861893 expected : true ,
862894 }, {
863895 description : "Setting disable-working-directory-overwrite to false should result in true" ,
864896 configMap : & corev1.ConfigMap {
865- ObjectMeta : metav1.ObjectMeta {Name : featureFlagConfigMapName , Namespace : system .GetNamespace ()},
897+ ObjectMeta : metav1.ObjectMeta {Name : GetFeatureFlagsConfigName () , Namespace : system .GetNamespace ()},
866898 Data : map [string ]string {
867899 featureFlagDisableWorkingDirKey : "false" ,
868900 },
@@ -871,7 +903,7 @@ func TestShouldOverrideWorkingDir(t *testing.T) {
871903 }, {
872904 description : "Setting disable-working-directory-overwrite to true should result in false" ,
873905 configMap : & corev1.ConfigMap {
874- ObjectMeta : metav1.ObjectMeta {Name : featureFlagConfigMapName , Namespace : system .GetNamespace ()},
906+ ObjectMeta : metav1.ObjectMeta {Name : GetFeatureFlagsConfigName () , Namespace : system .GetNamespace ()},
875907 Data : map [string ]string {
876908 featureFlagDisableWorkingDirKey : "true" ,
877909 },
0 commit comments