argo-cd

1
package common
2

3
import (
4
	"fmt"
5
	"os"
6
	"testing"
7
	"time"
8

9
	"github.com/stretchr/testify/assert"
10
)
11

12
// Test env var not set for EnvGRPCKeepAliveMin
13
func Test_GRPCKeepAliveMinNotSet(t *testing.T) {
14
	grpcKeepAliveMin := GetGRPCKeepAliveEnforcementMinimum()
15
	grpcKeepAliveExpectedMin := defaultGRPCKeepAliveEnforcementMinimum
16
	assert.Equal(t, grpcKeepAliveExpectedMin, grpcKeepAliveMin)
17

18
	grpcKeepAliveTime := GetGRPCKeepAliveTime()
19
	assert.Equal(t, 2*grpcKeepAliveExpectedMin, grpcKeepAliveTime)
20
}
21

22
// Test valid env var set for EnvGRPCKeepAliveMin
23
func Test_GRPCKeepAliveMinIsSet(t *testing.T) {
24
	numSeconds := 15
25
	os.Setenv(EnvGRPCKeepAliveMin, fmt.Sprintf("%ds", numSeconds))
26

27
	grpcKeepAliveMin := GetGRPCKeepAliveEnforcementMinimum()
28
	grpcKeepAliveExpectedMin := time.Duration(numSeconds) * time.Second
29
	assert.Equal(t, grpcKeepAliveExpectedMin, grpcKeepAliveMin)
30

31
	grpcKeepAliveTime := GetGRPCKeepAliveTime()
32
	assert.Equal(t, 2*grpcKeepAliveExpectedMin, grpcKeepAliveTime)
33
}
34

35
// Test invalid env var set for EnvGRPCKeepAliveMin
36
func Test_GRPCKeepAliveMinIncorrectlySet(t *testing.T) {
37
	numSeconds := 15
38
	os.Setenv(EnvGRPCKeepAliveMin, fmt.Sprintf("%d", numSeconds))
39

40
	grpcKeepAliveMin := GetGRPCKeepAliveEnforcementMinimum()
41
	grpcKeepAliveExpectedMin := defaultGRPCKeepAliveEnforcementMinimum
42
	assert.Equal(t, grpcKeepAliveExpectedMin, grpcKeepAliveMin)
43

44
	grpcKeepAliveTime := GetGRPCKeepAliveTime()
45
	assert.Equal(t, 2*grpcKeepAliveExpectedMin, grpcKeepAliveTime)
46
}
47