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]>
130 lines
3.7 KiB
Go
130 lines
3.7 KiB
Go
package settings
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
corev1 "k8s.io/api/core/v1"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
|
|
"github.com/hanzoai/cd/common"
|
|
"github.com/hanzoai/cd/util/settings"
|
|
)
|
|
|
|
const (
|
|
testNamespace = "default"
|
|
resourceOverrides = `{
|
|
"jsonPointers": [
|
|
""
|
|
],
|
|
"jqPathExpressions": [
|
|
""
|
|
],
|
|
"managedFieldsManagers": [
|
|
""
|
|
]
|
|
}`
|
|
)
|
|
|
|
func fixtures(ctx context.Context, data map[string]string) (*fake.Clientset, *settings.SettingsManager) {
|
|
kubeClient := fake.NewClientset(&corev1.ConfigMap{
|
|
Name: common.ConfigMapName,
|
|
Namespace: testNamespace,
|
|
Labels: map[string]string{
|
|
"app.kubernetes.io/part-of": "cd",
|
|
},
|
|
Data: data,
|
|
}, &corev1.Secret{
|
|
Name: "cd-secret",
|
|
Namespace: testNamespace,
|
|
Data: map[string][]byte{
|
|
"admin.password": []byte("test"),
|
|
"server.secretkey": []byte("test"),
|
|
},
|
|
})
|
|
settingsManager := settings.NewSettingsManager(ctx, kubeClient, testNamespace)
|
|
return kubeClient, settingsManager
|
|
}
|
|
|
|
func TestSettingsServer(t *testing.T) {
|
|
t.Parallel()
|
|
newServer := func(data map[string]string) *Server {
|
|
_, settingsMgr := fixtures(t.Context(), data)
|
|
return NewServer(settingsMgr, nil, nil, false, false, false, false)
|
|
}
|
|
|
|
t.Run("TestGetInstallationID", func(t *testing.T) {
|
|
t.Parallel()
|
|
settingsServer := newServer(map[string]string{
|
|
"installationID": "1234567890",
|
|
})
|
|
resp, err := settingsServer.Get(t.Context(), nil)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "1234567890", resp.InstallationID)
|
|
})
|
|
|
|
t.Run("TestGetInstallationIDNotSet", func(t *testing.T) {
|
|
t.Parallel()
|
|
settingsServer := newServer(map[string]string{})
|
|
resp, err := settingsServer.Get(t.Context(), nil)
|
|
require.NoError(t, err)
|
|
assert.Empty(t, resp.InstallationID)
|
|
})
|
|
|
|
t.Run("TestGetTrackingMethod", func(t *testing.T) {
|
|
t.Parallel()
|
|
settingsServer := newServer(map[string]string{
|
|
"application.resourceTrackingMethod": "annotation+label",
|
|
})
|
|
resp, err := settingsServer.Get(t.Context(), nil)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "annotation+label", resp.TrackingMethod)
|
|
})
|
|
|
|
t.Run("TestGetAppLabelKey", func(t *testing.T) {
|
|
t.Parallel()
|
|
settingsServer := newServer(map[string]string{
|
|
"application.instanceLabelKey": "instance",
|
|
})
|
|
resp, err := settingsServer.Get(t.Context(), nil)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "instance", resp.AppLabelKey)
|
|
})
|
|
|
|
t.Run("TestGetLoginButtonTextNotLoggedIn", func(t *testing.T) {
|
|
t.Parallel()
|
|
settingsServer := newServer(map[string]string{
|
|
"ui.loginButtonText": "Sign in with SSO",
|
|
})
|
|
resp, err := settingsServer.Get(t.Context(), nil)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "Sign in with SSO", resp.UiLoginButtonText)
|
|
})
|
|
|
|
t.Run("TestGetResourceOverridesNotLoggedIn", func(t *testing.T) {
|
|
t.Parallel()
|
|
settingsServer := newServer(map[string]string{
|
|
"resource.customizations.ignoreResourceUpdates.all": resourceOverrides,
|
|
})
|
|
resp, err := settingsServer.Get(t.Context(), nil)
|
|
require.NoError(t, err)
|
|
assert.Nil(t, resp.ResourceOverrides)
|
|
})
|
|
|
|
t.Run("TestGetResourceOverridesLoggedIn", func(t *testing.T) {
|
|
t.Parallel()
|
|
//nolint:staticcheck // it's ok to use built-in type string as key for value for testing purposes
|
|
loggedInContext := context.WithValue(t.Context(), "claims", &jwt.MapClaims{"iss": "qux", "sub": "foo", "email": "bar", "groups": []string{"baz"}})
|
|
settingsServer := newServer(map[string]string{
|
|
"resource.customizations.ignoreResourceUpdates.all": resourceOverrides,
|
|
})
|
|
resp, err := settingsServer.Get(loggedInContext, nil)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, resp.ResourceOverrides)
|
|
assert.NotEmpty(t, resp.ResourceOverrides["*/*"])
|
|
})
|
|
}
|