forgejo/services/mailer/mail_admin_new_user_test.go

// Copyright 2023 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package mailer

import (
	"context"
	"strconv"
	"testing"

	"code.gitea.io/gitea/models/db"
	user_model "code.gitea.io/gitea/models/user"
	"code.gitea.io/gitea/modules/setting"
	"code.gitea.io/gitea/modules/test"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	_ "github.com/mattn/go-sqlite3"
)

func getTestUsers(t *testing.T) []*user_model.User {
	t.Helper()
	admin := new(user_model.User)
	admin.Name = "testadmin"
	admin.IsAdmin = true
	admin.Language = "en_US"
	admin.Email = "admin@example.com"
	require.NoError(t, user_model.CreateUser(db.DefaultContext, admin))

	newUser := new(user_model.User)
	newUser.Name = "new_user"
	newUser.Language = "en_US"
	newUser.IsAdmin = false
	newUser.Email = "new_user@example.com"
	newUser.LastLoginUnix = 1693648327
	newUser.CreatedUnix = 1693648027
	require.NoError(t, user_model.CreateUser(db.DefaultContext, newUser))

	return []*user_model.User{admin, newUser}
}

func cleanUpUsers(ctx context.Context, users []*user_model.User) {
	for _, u := range users {
		db.DeleteByID[user_model.User](ctx, u.ID)
	}
}

func TestAdminNotificationMail_test(t *testing.T) {
	ctx := context.Background()

	users := getTestUsers(t)

	t.Run("SendNotificationEmailOnNewUser_true", func(t *testing.T) {
		defer test.MockVariableValue(&setting.Admin.SendNotificationEmailOnNewUser, true)()

		called := false
		defer mockMailSettings(func(msgs ...*Message) {
			assert.Len(t, msgs, 1, "Test provides only one admin user, so only one email must be sent")
			assert.Equal(t, msgs[0].To, users[0].Email, "checks if the recipient is the admin of the instance")
			manageUserURL := setting.AppURL + "admin/users/" + strconv.FormatInt(users[1].ID, 10)
			assert.Contains(t, msgs[0].Body, manageUserURL)
			assert.Contains(t, msgs[0].Body, users[1].HTMLURL())
			assert.Contains(t, msgs[0].Body, users[1].Name, "user name of the newly created user")
			assertTranslatedLocale(t, msgs[0].Body, "mail.admin", "admin.users")
			called = true
		})()
		MailNewUser(ctx, users[1])
		assert.True(t, called)
	})

	t.Run("SendNotificationEmailOnNewUser_false", func(t *testing.T) {
		defer test.MockVariableValue(&setting.Admin.SendNotificationEmailOnNewUser, false)()
		defer mockMailSettings(func(msgs ...*Message) {
			assert.Equal(t, 1, 0, "this shouldn't execute. MailNewUser must exit early since SEND_NOTIFICATION_EMAIL_ON_NEW_USER is disabled")
		})()
		MailNewUser(ctx, users[1])
	})

	cleanUpUsers(ctx, users)
}