fix featureflags init in go tests

fix/EE-6459/CVEs
Matt Hook 2023-12-27 08:33:02 +13:00
parent e3c5cd063b
commit 7ecb8a6a44
1 changed files with 4 additions and 4 deletions

View File

@ -14,7 +14,7 @@ func Test_enableFeaturesFromFlags(t *testing.T) {
supportedFeatures := []Feature{"supported", "supported2", "supported3", "supported4", "supported5"}
t.Run("supported features should be supported", func(t *testing.T) {
Init(supportedFeatures)
initSupportedFeatures(supportedFeatures)
for _, featureFlag := range supportedFeatures {
is.True(IsSupported(featureFlag))
@ -22,7 +22,7 @@ func Test_enableFeaturesFromFlags(t *testing.T) {
})
t.Run("unsupported features should not be supported", func(t *testing.T) {
Init(supportedFeatures)
initSupportedFeatures(supportedFeatures)
is.False(IsSupported("unsupported"))
})
@ -35,13 +35,13 @@ func Test_enableFeaturesFromFlags(t *testing.T) {
}
for _, test := range tests {
Init(supportedFeatures)
initSupportedFeatures(supportedFeatures)
os.Unsetenv("PORTAINER_FEATURE_FLAGS")
os.Setenv("PORTAINER_FEATURE_FLAGS", strings.Join(test.envFeatureFlags, ","))
t.Run("testing", func(t *testing.T) {
Parse(test.cliFeatureFlags)
Parse(test.cliFeatureFlags, supportedFeatures)
supported := toFeatureMap(test.cliFeatureFlags, test.envFeatureFlags)
// add env flags to supported flags