// Copyright 2023 The Gitea Authors. All rights reserved. // SPDX-License-Identifier: MIT package mailer import ( "context" "strconv" "strings" "testing" "code.gitea.io/gitea/models/db" user_model "code.gitea.io/gitea/models/user" "code.gitea.io/gitea/modules/setting" "github.com/stretchr/testify/assert" ) func getTestUsers() []*user_model.User { admin := new(user_model.User) admin.Name = "admin" admin.IsAdmin = true admin.Language = "en_US" admin.Email = "admin@example.com" 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 user_model.CreateUser(db.DefaultContext, admin) user_model.CreateUser(db.DefaultContext, newUser) users := make([]*user_model.User, 0) users = append(users, admin) users = append(users, newUser) return users } func cleanUpUsers(ctx context.Context, users []*user_model.User) { for _, u := range users { db.DeleteByID(ctx, u.ID, new(user_model.User)) } } func TestAdminNotificationMail_test(t *testing.T) { mailService := setting.Mailer{ From: "test@example.com", Protocol: "dummy", } setting.MailService = &mailService setting.Domain = "localhost" setting.AppSubURL = "http://localhost" // test with SEND_NOTIFICATION_EMAIL_ON_NEW_USER enabled setting.Admin.SendNotificationEmailOnNewUser = true ctx := context.Background() NewContext(ctx) users := getTestUsers() oldSendAsync := sa defer func() { sa = oldSendAsync cleanUpUsers(ctx, users) }() sa = func(msgs ...*Message) { assert.Equal(t, len(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 := "/admin/users/" + strconv.FormatInt(users[1].ID, 10) assert.True(t, strings.ContainsAny(msgs[0].Body, manageUserURL), "checks if the message contains the link to manage the newly created user from the admin panel") } MailNewUser(ctx, users[1]) // test with SEND_NOTIFICATION_EMAIL_ON_NEW_USER disabled; emails shouldn't be sent setting.Admin.SendNotificationEmailOnNewUser = false sa = 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]) }