mirror of
https://github.com/helmfile/helmfile.git
synced 2026-10-08 04:33:02 +02:00
Eng Zer Jun
ce3b81e2a1
This commit replaces `os.Setenv` with `t.Setenv` in tests. The environment variable is automatically restored to its original value when the test and all its subtests complete. Reference: https://pkg.go.dev/testing#T.Setenv Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
49 lines
1.7 KiB
Go
49 lines
1.7 KiB
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/helmfile/helmfile/pkg/envvar"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestIsExplicitSelectorInheritanceEnabled tests the isExplicitSelectorInheritanceEnabled function
|
|
func TestIsExplicitSelectorInheritanceEnabled(t *testing.T) {
|
|
//env var ExperimentalEnvVar is not set
|
|
require.Empty(t, os.Getenv(envvar.Experimental))
|
|
require.False(t, isExplicitSelectorInheritanceEnabled())
|
|
|
|
//check for env var ExperimentalEnvVar set to true
|
|
t.Setenv(envvar.Experimental, "true")
|
|
require.True(t, isExplicitSelectorInheritanceEnabled())
|
|
|
|
//check for env var ExperimentalEnvVar set to anything
|
|
t.Setenv(envvar.Experimental, "anything")
|
|
require.False(t, isExplicitSelectorInheritanceEnabled())
|
|
|
|
//check for env var ExperimentalEnvVar set to ExperimentalSelectorExplicit
|
|
t.Setenv(envvar.Experimental, ExperimentalSelectorExplicit)
|
|
require.True(t, isExplicitSelectorInheritanceEnabled())
|
|
|
|
// check for env var ExperimentalEnvVar set to a string that contains ExperimentalSelectorExplicit and ExperimentalEnvVar set to true
|
|
t.Setenv(envvar.Experimental, fmt.Sprintf("%s-%s-%s", "a", ExperimentalSelectorExplicit, "b"))
|
|
require.True(t, isExplicitSelectorInheritanceEnabled())
|
|
}
|
|
|
|
// TestExperimentalModeEnabled tests the experimentalModeEnabled function
|
|
func TestExperimentalModeEnabled(t *testing.T) {
|
|
//env var ExperimentalEnvVar is not set
|
|
require.Empty(t, os.Getenv(envvar.Experimental))
|
|
require.False(t, experimentalModeEnabled())
|
|
|
|
//check for env var ExperimentalEnvVar set to anything
|
|
t.Setenv(envvar.Experimental, "anything")
|
|
require.False(t, experimentalModeEnabled())
|
|
|
|
//check for env var ExperimentalEnvVar set to true
|
|
t.Setenv(envvar.Experimental, "true")
|
|
require.True(t, experimentalModeEnabled())
|
|
}
|