mirror of
https://github.com/hanzoai/cd.git
synced 2026-10-30 14:56:45 +00:00
zeekayandHanzo Dev
cbf668a980
go fix applied the modernizers: promoted-field keys in struct literals,
strings.Cut, SplitSeq, new(expr), slices and maps helpers, wg.Go, reflect.TypeFor
and errors.AsType. omitzero is left out, since omitting a zero struct such as
time.Time{} is a per-field wire decision and not a mechanical one.
Two changes the modernizers could not make for themselves. AsType needs its type
argument to be an error, so the git exit-status reader embeds error in the
interface it asks for. And an error built by concatenating a tag list into its
own format string now says so with a verb, which is what vet was pointing at.
Co-authored-by: Hanzo Dev <[email protected]>
109 lines
3.4 KiB
Go
109 lines
3.4 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hanzokv/go/v9"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
kubefake "k8s.io/client-go/kubernetes/fake"
|
|
)
|
|
|
|
// Test env var not set for EnvGRPCKeepAliveMin
|
|
func Test_GRPCKeepAliveMinNotSet(t *testing.T) {
|
|
grpcKeepAliveMin := GetGRPCKeepAliveEnforcementMinimum()
|
|
grpcKeepAliveExpectedMin := defaultGRPCKeepAliveEnforcementMinimum
|
|
assert.Equal(t, grpcKeepAliveExpectedMin, grpcKeepAliveMin)
|
|
|
|
grpcKeepAliveTime := GetGRPCKeepAliveTime()
|
|
assert.Equal(t, 2*grpcKeepAliveExpectedMin, grpcKeepAliveTime)
|
|
}
|
|
|
|
// Test valid env var set for EnvGRPCKeepAliveMin
|
|
func Test_GRPCKeepAliveMinIsSet(t *testing.T) {
|
|
numSeconds := 15
|
|
t.Setenv(EnvGRPCKeepAliveMin, fmt.Sprintf("%ds", numSeconds))
|
|
|
|
grpcKeepAliveMin := GetGRPCKeepAliveEnforcementMinimum()
|
|
grpcKeepAliveExpectedMin := time.Duration(numSeconds) * time.Second
|
|
assert.Equal(t, grpcKeepAliveExpectedMin, grpcKeepAliveMin)
|
|
|
|
grpcKeepAliveTime := GetGRPCKeepAliveTime()
|
|
assert.Equal(t, 2*grpcKeepAliveExpectedMin, grpcKeepAliveTime)
|
|
}
|
|
|
|
// Test invalid env var set for EnvGRPCKeepAliveMin
|
|
func Test_GRPCKeepAliveMinIncorrectlySet(t *testing.T) {
|
|
numSeconds := 15
|
|
t.Setenv(EnvGRPCKeepAliveMin, strconv.Itoa(numSeconds))
|
|
|
|
grpcKeepAliveMin := GetGRPCKeepAliveEnforcementMinimum()
|
|
grpcKeepAliveExpectedMin := defaultGRPCKeepAliveEnforcementMinimum
|
|
assert.Equal(t, grpcKeepAliveExpectedMin, grpcKeepAliveMin)
|
|
|
|
grpcKeepAliveTime := GetGRPCKeepAliveTime()
|
|
assert.Equal(t, 2*grpcKeepAliveExpectedMin, grpcKeepAliveTime)
|
|
}
|
|
|
|
func TestSetOptionalRedisPasswordFromKubeConfig(t *testing.T) {
|
|
t.Parallel()
|
|
testCases := []struct {
|
|
name, namespace, expectedPassword, expectedErr string
|
|
secret *corev1.Secret
|
|
}{
|
|
{
|
|
name: "Secret exists with correct key",
|
|
namespace: "default",
|
|
expectedPassword: "password123",
|
|
expectedErr: "",
|
|
secret: &corev1.Secret{
|
|
Name: RedisInitialCredentials,
|
|
Data: map[string][]byte{RedisInitialCredentialsKey: []byte("password123")},
|
|
},
|
|
},
|
|
{
|
|
name: "Secret does not exist",
|
|
namespace: "default",
|
|
expectedPassword: "",
|
|
expectedErr: "failed to get secret default/" + RedisInitialCredentials,
|
|
secret: nil,
|
|
},
|
|
{
|
|
name: "Secret exists without correct key",
|
|
namespace: "default",
|
|
expectedPassword: "",
|
|
expectedErr: fmt.Sprintf("secret default/%s does not contain key %s", RedisInitialCredentials, RedisInitialCredentialsKey),
|
|
secret: &corev1.Secret{
|
|
Name: RedisInitialCredentials,
|
|
Data: map[string][]byte{},
|
|
},
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
var (
|
|
ctx = t.Context()
|
|
kubeClient = kubefake.NewClientset()
|
|
redisOptions = &kv.Options{}
|
|
)
|
|
if tc.secret != nil {
|
|
_, err := kubeClient.CoreV1().Secrets(tc.namespace).Create(ctx, tc.secret, metav1.CreateOptions{})
|
|
require.NoErrorf(t, err, "Failed to create secret")
|
|
}
|
|
err := SetOptionalRedisPasswordFromKubeConfig(ctx, kubeClient, tc.namespace, redisOptions)
|
|
if tc.expectedErr != "" {
|
|
require.ErrorContains(t, err, tc.expectedErr)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
require.Equal(t, tc.expectedPassword, redisOptions.Password)
|
|
})
|
|
}
|
|
}
|