123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311 |
- // Copyright 2017 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package user_test
-
- import (
- "testing"
-
- "code.gitea.io/gitea/models/db"
- "code.gitea.io/gitea/models/unittest"
- user_model "code.gitea.io/gitea/models/user"
- "code.gitea.io/gitea/modules/util"
-
- "github.com/stretchr/testify/assert"
- )
-
- func TestGetEmailAddresses(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- emails, _ := user_model.GetEmailAddresses(int64(1))
- if assert.Len(t, emails, 3) {
- assert.True(t, emails[0].IsPrimary)
- assert.True(t, emails[2].IsActivated)
- assert.False(t, emails[2].IsPrimary)
- }
-
- emails, _ = user_model.GetEmailAddresses(int64(2))
- if assert.Len(t, emails, 2) {
- assert.True(t, emails[0].IsPrimary)
- assert.True(t, emails[0].IsActivated)
- }
- }
-
- func TestIsEmailUsed(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- isExist, _ := user_model.IsEmailUsed(db.DefaultContext, "")
- assert.True(t, isExist)
- isExist, _ = user_model.IsEmailUsed(db.DefaultContext, "user11@example.com")
- assert.True(t, isExist)
- isExist, _ = user_model.IsEmailUsed(db.DefaultContext, "user1234567890@example.com")
- assert.False(t, isExist)
- }
-
- func TestAddEmailAddress(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- assert.NoError(t, user_model.AddEmailAddress(db.DefaultContext, &user_model.EmailAddress{
- Email: "user1234567890@example.com",
- LowerEmail: "user1234567890@example.com",
- IsPrimary: true,
- IsActivated: true,
- }))
-
- // ErrEmailAlreadyUsed
- err := user_model.AddEmailAddress(db.DefaultContext, &user_model.EmailAddress{
- Email: "user1234567890@example.com",
- LowerEmail: "user1234567890@example.com",
- })
- assert.Error(t, err)
- assert.True(t, user_model.IsErrEmailAlreadyUsed(err))
- }
-
- func TestAddEmailAddresses(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- // insert multiple email address
- emails := make([]*user_model.EmailAddress, 2)
- emails[0] = &user_model.EmailAddress{
- Email: "user1234@example.com",
- LowerEmail: "user1234@example.com",
- IsActivated: true,
- }
- emails[1] = &user_model.EmailAddress{
- Email: "user5678@example.com",
- LowerEmail: "user5678@example.com",
- IsActivated: true,
- }
- assert.NoError(t, user_model.AddEmailAddresses(emails))
-
- // ErrEmailAlreadyUsed
- err := user_model.AddEmailAddresses(emails)
- assert.Error(t, err)
- assert.True(t, user_model.IsErrEmailAlreadyUsed(err))
- }
-
- func TestDeleteEmailAddress(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- assert.NoError(t, user_model.DeleteEmailAddress(&user_model.EmailAddress{
- UID: int64(1),
- ID: int64(33),
- Email: "user1-2@example.com",
- LowerEmail: "user1-2@example.com",
- }))
-
- assert.NoError(t, user_model.DeleteEmailAddress(&user_model.EmailAddress{
- UID: int64(1),
- Email: "user1-3@example.com",
- LowerEmail: "user1-3@example.com",
- }))
-
- // Email address does not exist
- err := user_model.DeleteEmailAddress(&user_model.EmailAddress{
- UID: int64(1),
- Email: "user1234567890@example.com",
- LowerEmail: "user1234567890@example.com",
- })
- assert.Error(t, err)
- }
-
- func TestDeleteEmailAddresses(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- // delete multiple email address
- emails := make([]*user_model.EmailAddress, 2)
- emails[0] = &user_model.EmailAddress{
- UID: int64(2),
- ID: int64(3),
- Email: "user2@example.com",
- LowerEmail: "user2@example.com",
- }
- emails[1] = &user_model.EmailAddress{
- UID: int64(2),
- Email: "user2-2@example.com",
- LowerEmail: "user2-2@example.com",
- }
- assert.NoError(t, user_model.DeleteEmailAddresses(emails))
-
- // ErrEmailAlreadyUsed
- err := user_model.DeleteEmailAddresses(emails)
- assert.Error(t, err)
- }
-
- func TestMakeEmailPrimary(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- email := &user_model.EmailAddress{
- Email: "user567890@example.com",
- }
- err := user_model.MakeEmailPrimary(email)
- assert.Error(t, err)
- assert.EqualError(t, err, user_model.ErrEmailAddressNotExist{Email: email.Email}.Error())
-
- email = &user_model.EmailAddress{
- Email: "user11@example.com",
- }
- err = user_model.MakeEmailPrimary(email)
- assert.Error(t, err)
- assert.EqualError(t, err, user_model.ErrEmailNotActivated.Error())
-
- email = &user_model.EmailAddress{
- Email: "user9999999@example.com",
- }
- err = user_model.MakeEmailPrimary(email)
- assert.Error(t, err)
- assert.True(t, user_model.IsErrUserNotExist(err))
-
- email = &user_model.EmailAddress{
- Email: "user101@example.com",
- }
- err = user_model.MakeEmailPrimary(email)
- assert.NoError(t, err)
-
- user, _ := user_model.GetUserByID(db.DefaultContext, int64(10))
- assert.Equal(t, "user101@example.com", user.Email)
- }
-
- func TestActivate(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- email := &user_model.EmailAddress{
- ID: int64(1),
- UID: int64(1),
- Email: "user11@example.com",
- }
- assert.NoError(t, user_model.ActivateEmail(email))
-
- emails, _ := user_model.GetEmailAddresses(int64(1))
- assert.Len(t, emails, 3)
- assert.True(t, emails[0].IsActivated)
- assert.True(t, emails[0].IsPrimary)
- assert.False(t, emails[1].IsPrimary)
- assert.True(t, emails[2].IsActivated)
- assert.False(t, emails[2].IsPrimary)
- }
-
- func TestListEmails(t *testing.T) {
- assert.NoError(t, unittest.PrepareTestDatabase())
-
- // Must find all users and their emails
- opts := &user_model.SearchEmailOptions{
- ListOptions: db.ListOptions{
- PageSize: 10000,
- },
- }
- emails, count, err := user_model.SearchEmails(opts)
- assert.NoError(t, err)
- assert.NotEqual(t, int64(0), count)
- assert.True(t, count > 5)
-
- contains := func(match func(s *user_model.SearchEmailResult) bool) bool {
- for _, v := range emails {
- if match(v) {
- return true
- }
- }
- return false
- }
-
- assert.True(t, contains(func(s *user_model.SearchEmailResult) bool { return s.UID == 18 }))
- // 'user3' is an organization
- assert.False(t, contains(func(s *user_model.SearchEmailResult) bool { return s.UID == 3 }))
-
- // Must find no records
- opts = &user_model.SearchEmailOptions{Keyword: "NOTFOUND"}
- emails, count, err = user_model.SearchEmails(opts)
- assert.NoError(t, err)
- assert.Equal(t, int64(0), count)
-
- // Must find users 'user2', 'user28', etc.
- opts = &user_model.SearchEmailOptions{Keyword: "user2"}
- emails, count, err = user_model.SearchEmails(opts)
- assert.NoError(t, err)
- assert.NotEqual(t, int64(0), count)
- assert.True(t, contains(func(s *user_model.SearchEmailResult) bool { return s.UID == 2 }))
- assert.True(t, contains(func(s *user_model.SearchEmailResult) bool { return s.UID == 27 }))
-
- // Must find only primary addresses (i.e. from the `user` table)
- opts = &user_model.SearchEmailOptions{IsPrimary: util.OptionalBoolTrue}
- emails, _, err = user_model.SearchEmails(opts)
- assert.NoError(t, err)
- assert.True(t, contains(func(s *user_model.SearchEmailResult) bool { return s.IsPrimary }))
- assert.False(t, contains(func(s *user_model.SearchEmailResult) bool { return !s.IsPrimary }))
-
- // Must find only inactive addresses (i.e. not validated)
- opts = &user_model.SearchEmailOptions{IsActivated: util.OptionalBoolFalse}
- emails, _, err = user_model.SearchEmails(opts)
- assert.NoError(t, err)
- assert.True(t, contains(func(s *user_model.SearchEmailResult) bool { return !s.IsActivated }))
- assert.False(t, contains(func(s *user_model.SearchEmailResult) bool { return s.IsActivated }))
-
- // Must find more than one page, but retrieve only one
- opts = &user_model.SearchEmailOptions{
- ListOptions: db.ListOptions{
- PageSize: 5,
- Page: 1,
- },
- }
- emails, count, err = user_model.SearchEmails(opts)
- assert.NoError(t, err)
- assert.Len(t, emails, 5)
- assert.Greater(t, count, int64(len(emails)))
- }
-
- func TestEmailAddressValidate(t *testing.T) {
- kases := map[string]error{
- "abc@gmail.com": nil,
- "132@hotmail.com": nil,
- "1-3-2@test.org": nil,
- "1.3.2@test.org": nil,
- "a_123@test.org.cn": nil,
- `first.last@iana.org`: nil,
- `first!last@iana.org`: nil,
- `first#last@iana.org`: nil,
- `first$last@iana.org`: nil,
- `first%last@iana.org`: nil,
- `first&last@iana.org`: nil,
- `first'last@iana.org`: nil,
- `first*last@iana.org`: nil,
- `first+last@iana.org`: nil,
- `first/last@iana.org`: nil,
- `first=last@iana.org`: nil,
- `first?last@iana.org`: nil,
- `first^last@iana.org`: nil,
- "first`last@iana.org": nil,
- `first{last@iana.org`: nil,
- `first|last@iana.org`: nil,
- `first}last@iana.org`: nil,
- `first~last@iana.org`: nil,
- `first;last@iana.org`: user_model.ErrEmailCharIsNotSupported{`first;last@iana.org`},
- ".233@qq.com": user_model.ErrEmailInvalid{".233@qq.com"},
- "!233@qq.com": nil,
- "#233@qq.com": nil,
- "$233@qq.com": nil,
- "%233@qq.com": nil,
- "&233@qq.com": nil,
- "'233@qq.com": nil,
- "*233@qq.com": nil,
- "+233@qq.com": nil,
- "-233@qq.com": user_model.ErrEmailInvalid{"-233@qq.com"},
- "/233@qq.com": nil,
- "=233@qq.com": nil,
- "?233@qq.com": nil,
- "^233@qq.com": nil,
- "_233@qq.com": nil,
- "`233@qq.com": nil,
- "{233@qq.com": nil,
- "|233@qq.com": nil,
- "}233@qq.com": nil,
- "~233@qq.com": nil,
- ";233@qq.com": user_model.ErrEmailCharIsNotSupported{";233@qq.com"},
- "Foo <foo@bar.com>": user_model.ErrEmailCharIsNotSupported{"Foo <foo@bar.com>"},
- string([]byte{0xE2, 0x84, 0xAA}): user_model.ErrEmailCharIsNotSupported{string([]byte{0xE2, 0x84, 0xAA})},
- }
- for kase, err := range kases {
- t.Run(kase, func(t *testing.T) {
- assert.EqualValues(t, err, user_model.ValidateEmail(kase))
- })
- }
- }
|