Lunny XiaoandGitHub
fd7d83ace6
* Move almost all functions' parameter db.Engine to context.Context * remove some unnecessary wrap functions
310 lines
9.6 KiB
Go
310 lines
9.6 KiB
Go
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package user
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
"code.gitea.io/gitea/models/unittest"
|
|
"code.gitea.io/gitea/modules/util"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetEmailAddresses(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
emails, _ := 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, _ = 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, _ := IsEmailUsed(db.DefaultContext, "")
|
|
assert.True(t, isExist)
|
|
isExist, _ = IsEmailUsed(db.DefaultContext, "[email protected]")
|
|
assert.True(t, isExist)
|
|
isExist, _ = IsEmailUsed(db.DefaultContext, "[email protected]")
|
|
assert.False(t, isExist)
|
|
}
|
|
|
|
func TestAddEmailAddress(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
assert.NoError(t, AddEmailAddress(db.DefaultContext, &EmailAddress{
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
IsPrimary: true,
|
|
IsActivated: true,
|
|
}))
|
|
|
|
// ErrEmailAlreadyUsed
|
|
err := AddEmailAddress(db.DefaultContext, &EmailAddress{
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
})
|
|
assert.Error(t, err)
|
|
assert.True(t, IsErrEmailAlreadyUsed(err))
|
|
}
|
|
|
|
func TestAddEmailAddresses(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
// insert multiple email address
|
|
emails := make([]*EmailAddress, 2)
|
|
emails[0] = &EmailAddress{
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
IsActivated: true,
|
|
}
|
|
emails[1] = &EmailAddress{
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
IsActivated: true,
|
|
}
|
|
assert.NoError(t, AddEmailAddresses(emails))
|
|
|
|
// ErrEmailAlreadyUsed
|
|
err := AddEmailAddresses(emails)
|
|
assert.Error(t, err)
|
|
assert.True(t, IsErrEmailAlreadyUsed(err))
|
|
}
|
|
|
|
func TestDeleteEmailAddress(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
assert.NoError(t, DeleteEmailAddress(&EmailAddress{
|
|
UID: int64(1),
|
|
ID: int64(33),
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
}))
|
|
|
|
assert.NoError(t, DeleteEmailAddress(&EmailAddress{
|
|
UID: int64(1),
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
}))
|
|
|
|
// Email address does not exist
|
|
err := DeleteEmailAddress(&EmailAddress{
|
|
UID: int64(1),
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
})
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestDeleteEmailAddresses(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
// delete multiple email address
|
|
emails := make([]*EmailAddress, 2)
|
|
emails[0] = &EmailAddress{
|
|
UID: int64(2),
|
|
ID: int64(3),
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
}
|
|
emails[1] = &EmailAddress{
|
|
UID: int64(2),
|
|
Email: "[email protected]",
|
|
LowerEmail: "[email protected]",
|
|
}
|
|
assert.NoError(t, DeleteEmailAddresses(emails))
|
|
|
|
// ErrEmailAlreadyUsed
|
|
err := DeleteEmailAddresses(emails)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestMakeEmailPrimary(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
email := &EmailAddress{
|
|
Email: "[email protected]",
|
|
}
|
|
err := MakeEmailPrimary(email)
|
|
assert.Error(t, err)
|
|
assert.EqualError(t, err, ErrEmailAddressNotExist{Email: email.Email}.Error())
|
|
|
|
email = &EmailAddress{
|
|
Email: "[email protected]",
|
|
}
|
|
err = MakeEmailPrimary(email)
|
|
assert.Error(t, err)
|
|
assert.EqualError(t, err, ErrEmailNotActivated.Error())
|
|
|
|
email = &EmailAddress{
|
|
Email: "[email protected]",
|
|
}
|
|
err = MakeEmailPrimary(email)
|
|
assert.Error(t, err)
|
|
assert.True(t, IsErrUserNotExist(err))
|
|
|
|
email = &EmailAddress{
|
|
Email: "[email protected]",
|
|
}
|
|
err = MakeEmailPrimary(email)
|
|
assert.NoError(t, err)
|
|
|
|
user, _ := GetUserByID(int64(10))
|
|
assert.Equal(t, "[email protected]", user.Email)
|
|
}
|
|
|
|
func TestActivate(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
email := &EmailAddress{
|
|
ID: int64(1),
|
|
UID: int64(1),
|
|
Email: "[email protected]",
|
|
}
|
|
assert.NoError(t, ActivateEmail(email))
|
|
|
|
emails, _ := 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 := &SearchEmailOptions{
|
|
ListOptions: db.ListOptions{
|
|
PageSize: 10000,
|
|
},
|
|
}
|
|
emails, count, err := SearchEmails(opts)
|
|
assert.NoError(t, err)
|
|
assert.NotEqual(t, int64(0), count)
|
|
assert.True(t, count > 5)
|
|
|
|
contains := func(match func(s *SearchEmailResult) bool) bool {
|
|
for _, v := range emails {
|
|
if match(v) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
assert.True(t, contains(func(s *SearchEmailResult) bool { return s.UID == 18 }))
|
|
// 'user3' is an organization
|
|
assert.False(t, contains(func(s *SearchEmailResult) bool { return s.UID == 3 }))
|
|
|
|
// Must find no records
|
|
opts = &SearchEmailOptions{Keyword: "NOTFOUND"}
|
|
emails, count, err = SearchEmails(opts)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(0), count)
|
|
|
|
// Must find users 'user2', 'user28', etc.
|
|
opts = &SearchEmailOptions{Keyword: "user2"}
|
|
emails, count, err = SearchEmails(opts)
|
|
assert.NoError(t, err)
|
|
assert.NotEqual(t, int64(0), count)
|
|
assert.True(t, contains(func(s *SearchEmailResult) bool { return s.UID == 2 }))
|
|
assert.True(t, contains(func(s *SearchEmailResult) bool { return s.UID == 27 }))
|
|
|
|
// Must find only primary addresses (i.e. from the `user` table)
|
|
opts = &SearchEmailOptions{IsPrimary: util.OptionalBoolTrue}
|
|
emails, _, err = SearchEmails(opts)
|
|
assert.NoError(t, err)
|
|
assert.True(t, contains(func(s *SearchEmailResult) bool { return s.IsPrimary }))
|
|
assert.False(t, contains(func(s *SearchEmailResult) bool { return !s.IsPrimary }))
|
|
|
|
// Must find only inactive addresses (i.e. not validated)
|
|
opts = &SearchEmailOptions{IsActivated: util.OptionalBoolFalse}
|
|
emails, _, err = SearchEmails(opts)
|
|
assert.NoError(t, err)
|
|
assert.True(t, contains(func(s *SearchEmailResult) bool { return !s.IsActivated }))
|
|
assert.False(t, contains(func(s *SearchEmailResult) bool { return s.IsActivated }))
|
|
|
|
// Must find more than one page, but retrieve only one
|
|
opts = &SearchEmailOptions{
|
|
ListOptions: db.ListOptions{
|
|
PageSize: 5,
|
|
Page: 1,
|
|
},
|
|
}
|
|
emails, count, err = 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{
|
|
"[email protected]": nil,
|
|
"[email protected]": nil,
|
|
"[email protected]": nil,
|
|
"[email protected]": nil,
|
|
"[email protected]": nil,
|
|
`[email protected]`: nil,
|
|
`[email protected]`: nil,
|
|
`first#[email protected]`: nil,
|
|
`[email protected]`: nil,
|
|
`first%[email protected]`: nil,
|
|
`first&[email protected]`: nil,
|
|
`first'[email protected]`: nil,
|
|
`first*[email protected]`: nil,
|
|
`[email protected]`: nil,
|
|
`first/[email protected]`: nil,
|
|
`[email protected]`: nil,
|
|
`[email protected]`: nil,
|
|
`first^[email protected]`: nil,
|
|
"first`[email protected]": nil,
|
|
`first{[email protected]`: nil,
|
|
`first|[email protected]`: nil,
|
|
`first}[email protected]`: nil,
|
|
`[email protected]`: nil,
|
|
`first;[email protected]`: ErrEmailCharIsNotSupported{`first;[email protected]`},
|
|
"[email protected]": ErrEmailInvalid{"[email protected]"},
|
|
"[email protected]": ErrEmailInvalid{"[email protected]"},
|
|
"#[email protected]": ErrEmailInvalid{"#[email protected]"},
|
|
"[email protected]": ErrEmailInvalid{"[email protected]"},
|
|
"%[email protected]": ErrEmailInvalid{"%[email protected]"},
|
|
"&[email protected]": ErrEmailInvalid{"&[email protected]"},
|
|
"'[email protected]": ErrEmailInvalid{"'[email protected]"},
|
|
"*[email protected]": ErrEmailInvalid{"*[email protected]"},
|
|
"[email protected]": ErrEmailInvalid{"[email protected]"},
|
|
"/[email protected]": ErrEmailInvalid{"/[email protected]"},
|
|
"[email protected]": ErrEmailInvalid{"[email protected]"},
|
|
"[email protected]": ErrEmailInvalid{"[email protected]"},
|
|
"^[email protected]": ErrEmailInvalid{"^[email protected]"},
|
|
"`[email protected]": ErrEmailInvalid{"`[email protected]"},
|
|
"{[email protected]": ErrEmailInvalid{"{[email protected]"},
|
|
"|[email protected]": ErrEmailInvalid{"|[email protected]"},
|
|
"}[email protected]": ErrEmailInvalid{"}[email protected]"},
|
|
"[email protected]": ErrEmailInvalid{"[email protected]"},
|
|
";[email protected]": ErrEmailCharIsNotSupported{";[email protected]"},
|
|
"Foo <[email protected]>": ErrEmailCharIsNotSupported{"Foo <[email protected]>"},
|
|
string([]byte{0xE2, 0x84, 0xAA}): ErrEmailCharIsNotSupported{string([]byte{0xE2, 0x84, 0xAA})},
|
|
}
|
|
for kase, err := range kases {
|
|
t.Run(kase, func(t *testing.T) {
|
|
assert.EqualValues(t, err, ValidateEmail(kase))
|
|
})
|
|
}
|
|
}
|