grafana/pkg/registry/apis/iam/user/mutate_test.go

package user

import (
	"context"
	"testing"

	iamv0alpha1 "github.com/grafana/grafana/apps/iam/pkg/apis/iam/v0alpha1"
	"github.com/stretchr/testify/require"
)

func TestMutateOnCreate_LoginEmail(t *testing.T) {
	testCases := []struct {
		name          string
		inputUser     *iamv0alpha1.User
		expectedLogin string
		expectedEmail string
	}{
		{
			name: "login and email provided with mixed case",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Login: "Test.User",
					Email: "Test.User@example.com",
				},
			},
			expectedLogin: "test.user",
			expectedEmail: "test.user@example.com",
		},
		{
			name: "only email provided",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Email: "Only.Email@example.com",
				},
			},
			expectedLogin: "only.email@example.com",
			expectedEmail: "only.email@example.com",
		},
		{
			name: "only login provided",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Login: "Only.Login",
				},
			},
			expectedLogin: "only.login",
		},
		{
			name: "login and email already lowercase",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Login: "already.lower",
					Email: "already.lower@example.com",
				},
			},
			expectedLogin: "already.lower",
			expectedEmail: "already.lower@example.com",
		},
		{
			name: "both login and email are empty",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{},
			},
			expectedLogin: "",
			expectedEmail: "",
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			err := MutateOnCreateAndUpdate(context.Background(), tc.inputUser)
			require.NoError(t, err)
			require.Equal(t, tc.expectedLogin, tc.inputUser.Spec.Login)
			require.Equal(t, tc.expectedEmail, tc.inputUser.Spec.Email)
		})
	}
}

func TestMutateOnCreate_Role(t *testing.T) {
	testCases := []struct {
		name         string
		inputUser    *iamv0alpha1.User
		expectedRole string
	}{
		{
			name: "role is lowercase",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Role: "admin",
				},
			},
			expectedRole: "Admin",
		},
		{
			name: "role is uppercase",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Role: "ADMIN",
				},
			},
			expectedRole: "Admin",
		},
		{
			name: "role is mixed case",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Role: "aDmIn",
				},
			},
			expectedRole: "Admin",
		},
		{
			name: "role is already title case",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Role: "Admin",
				},
			},
			expectedRole: "Admin",
		},
		{
			name: "role is empty",
			inputUser: &iamv0alpha1.User{
				Spec: iamv0alpha1.UserSpec{
					Role: "",
				},
			},
			expectedRole: "",
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			err := MutateOnCreateAndUpdate(context.Background(), tc.inputUser)
			require.NoError(t, err)
			require.Equal(t, tc.expectedRole, tc.inputUser.Spec.Role)
		})
	}
}