mirror of
https://github.com/argoproj/argo-cd
synced 2026-05-04 15:18:20 +00:00
f533ed17f0
feat: simplify parametrization of Argo CD server processes (#6328) Signed-off-by: Ishita Sequeira <isequeir@redhat.com>
122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
package env
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
util "github.com/argoproj/argo-cd/v2/util/io"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// nolint:unparam
|
|
func setEnv(t *testing.T, env string, val string) io.Closer {
|
|
assert.NoError(t, os.Setenv(env, val))
|
|
return util.NewCloser(func() error {
|
|
assert.NoError(t, os.Setenv(env, ""))
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func TestParseNumFromEnv_NoEnvVariable(t *testing.T) {
|
|
num := ParseNumFromEnv("test", 10, 0, 100)
|
|
|
|
assert.Equal(t, 10, num)
|
|
}
|
|
|
|
func TestParseNumFromEnv_CorrectValueSet(t *testing.T) {
|
|
closer := setEnv(t, "test", "15")
|
|
defer util.Close(closer)
|
|
|
|
num := ParseNumFromEnv("test", 10, 0, 100)
|
|
|
|
assert.Equal(t, 15, num)
|
|
}
|
|
|
|
func TestParseNumFromEnv_NonIntValueSet(t *testing.T) {
|
|
closer := setEnv(t, "test", "wrong")
|
|
defer util.Close(closer)
|
|
|
|
num := ParseNumFromEnv("test", 10, 0, 100)
|
|
|
|
assert.Equal(t, 10, num)
|
|
}
|
|
|
|
func TestParseNumFromEnv_NegativeValueSet(t *testing.T) {
|
|
closer := setEnv(t, "test", "-1")
|
|
defer util.Close(closer)
|
|
|
|
num := ParseNumFromEnv("test", 10, 0, 100)
|
|
|
|
assert.Equal(t, 10, num)
|
|
}
|
|
|
|
func TestParseNumFromEnv_OutOfRangeValueSet(t *testing.T) {
|
|
closer := setEnv(t, "test", "1000")
|
|
defer util.Close(closer)
|
|
|
|
num := ParseNumFromEnv("test", 10, 0, 100)
|
|
|
|
assert.Equal(t, 10, num)
|
|
}
|
|
|
|
func TestParseDurationFromEnv(t *testing.T) {
|
|
testKey := "key"
|
|
defaultVal := 2 * time.Second
|
|
min := 1 * time.Second
|
|
max := 3 * time.Second
|
|
|
|
testCases := []struct {
|
|
name string
|
|
env string
|
|
expected time.Duration
|
|
}{{
|
|
name: "EnvNotSet",
|
|
expected: defaultVal,
|
|
}, {
|
|
name: "ValidValueSet",
|
|
env: "2s",
|
|
expected: time.Second * 2,
|
|
}, {
|
|
name: "MoreThanMaxSet",
|
|
env: "5s",
|
|
expected: defaultVal,
|
|
}, {
|
|
name: "LessThanMinSet",
|
|
env: "-1s",
|
|
expected: defaultVal,
|
|
}, {
|
|
name: "InvalidSet",
|
|
env: "hello",
|
|
expected: defaultVal,
|
|
}}
|
|
|
|
for i, tc := range testCases {
|
|
t.Run(testCases[i].name, func(t *testing.T) {
|
|
tc = testCases[i]
|
|
setEnv(t, testKey, tc.env)
|
|
|
|
val := ParseDurationFromEnv(testKey, defaultVal, min, max)
|
|
assert.Equal(t, tc.expected, val)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_ParseBoolFromEnv(t *testing.T) {
|
|
t.Run("Get 'true' value from existing env var", func(t *testing.T) {
|
|
_ = os.Setenv("TEST_BOOL_VAL", "true")
|
|
defer os.Setenv("TEST_BOOL_VAL", "")
|
|
assert.True(t, ParseBoolFromEnv("TEST_BOOL_VAL", false))
|
|
})
|
|
t.Run("Get 'false' value from existing env var", func(t *testing.T) {
|
|
_ = os.Setenv("TEST_BOOL_VAL", "false")
|
|
defer os.Setenv("TEST_BOOL_VAL", "")
|
|
assert.False(t, ParseBoolFromEnv("TEST_BOOL_VAL", true))
|
|
})
|
|
t.Run("Get default value from non-existing env var", func(t *testing.T) {
|
|
_ = os.Setenv("TEST_BOOL_VAL", "")
|
|
assert.True(t, ParseBoolFromEnv("TEST_BOOL_VAL", true))
|
|
})
|
|
}
|