初始提交: Gitea 项目代码

services/user/email_test.go

@@ -0,0 +1,76 @@
+// Copyright 2024 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package user
+
+import (
+	"testing"
+
+	"gitea.dev/models/unittest"
+	user_model "gitea.dev/models/user"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestUserEmail(t *testing.T) {
+	assert.NoError(t, unittest.PrepareTestDatabase())
+
+	t.Run("PrimaryEmailAddress", func(t *testing.T) {
+		user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 13})
+
+		emails, err := user_model.GetEmailAddresses(t.Context(), user.ID)
+		assert.NoError(t, err)
+		assert.Len(t, emails, 1)
+
+		primary, err := user_model.GetPrimaryEmailAddressOfUser(t.Context(), user.ID)
+		assert.NoError(t, err)
+		assert.NotEqual(t, "primary-13@example.com", primary.Email)
+		assert.Equal(t, user.Email, primary.Email)
+
+		assert.NoError(t, ReplacePrimaryEmailAddress(t.Context(), user, "primary-13@example.com"))
+
+		primary, err = user_model.GetPrimaryEmailAddressOfUser(t.Context(), user.ID)
+		assert.NoError(t, err)
+		assert.Equal(t, "primary-13@example.com", primary.Email)
+		assert.Equal(t, user.Email, primary.Email)
+
+		emails, err = user_model.GetEmailAddresses(t.Context(), user.ID)
+		assert.NoError(t, err)
+		assert.Len(t, emails, 1)
+
+		assert.NoError(t, ReplacePrimaryEmailAddress(t.Context(), user, "primary-13@example.com"))
+	})
+
+	t.Run("AddEmailAddresses", func(t *testing.T) {
+		user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
+
+		assert.Error(t, AddEmailAddresses(t.Context(), user, []string{" invalid email "}))
+
+		emails := []string{"user1234@example.com", "user5678@example.com"}
+
+		assert.NoError(t, AddEmailAddresses(t.Context(), user, emails))
+
+		err := AddEmailAddresses(t.Context(), user, emails)
+		assert.Error(t, err)
+		assert.True(t, user_model.IsErrEmailAlreadyUsed(err))
+	})
+
+	t.Run("DeleteEmailAddresses", func(t *testing.T) {
+		user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
+
+		emails := []string{"user2-2@example.com"}
+
+		err := DeleteEmailAddresses(t.Context(), user, emails)
+		assert.NoError(t, err)
+
+		err = DeleteEmailAddresses(t.Context(), user, emails)
+		assert.Error(t, err)
+		assert.True(t, user_model.IsErrEmailAddressNotExist(err))
+
+		emails = []string{"user2@example.com"}
+
+		err = DeleteEmailAddresses(t.Context(), user, emails)
+		assert.Error(t, err)
+		assert.True(t, user_model.IsErrPrimaryEmailCannotDelete(err))
+	})
+}