2021-11-24 10:49:20 +01:00
// Copyright 2017 The Gitea Authors. All rights reserved.
2022-11-27 19:20:29 +01:00
// SPDX-License-Identifier: MIT
2021-11-24 10:49:20 +01:00
2022-06-15 09:02:00 +02:00
package user_test
2021-11-24 10:49:20 +01:00
import (
2023-02-16 17:32:01 +01:00
"context"
2024-04-27 18:50:35 +02:00
"crypto/rand"
2023-04-19 15:40:42 +02:00
"fmt"
2021-11-24 10:49:20 +01:00
"strings"
"testing"
2023-02-16 17:32:01 +01:00
"time"
2021-11-24 10:49:20 +01:00
2022-01-02 14:12:35 +01:00
"code.gitea.io/gitea/models/auth"
2021-11-24 10:49:20 +01:00
"code.gitea.io/gitea/models/db"
"code.gitea.io/gitea/models/unittest"
2022-06-15 09:02:00 +02:00
user_model "code.gitea.io/gitea/models/user"
2023-02-19 08:35:20 +01:00
"code.gitea.io/gitea/modules/auth/password/hash"
2024-03-29 16:05:41 +01:00
"code.gitea.io/gitea/modules/container"
2024-02-29 19:52:49 +01:00
"code.gitea.io/gitea/modules/optional"
2021-11-24 10:49:20 +01:00
"code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/modules/structs"
2023-02-16 17:32:01 +01:00
"code.gitea.io/gitea/modules/timeutil"
2021-11-24 10:49:20 +01:00
"github.com/stretchr/testify/assert"
)
func TestOAuth2Application_LoadUser ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-08-16 04:22:25 +02:00
app := unittest . AssertExistsAndLoadBean ( t , & auth . OAuth2Application { ID : 1 } )
2022-12-03 03:48:26 +01:00
user , err := user_model . GetUserByID ( db . DefaultContext , app . UID )
2021-11-24 10:49:20 +01:00
assert . NoError ( t , err )
assert . NotNil ( t , user )
}
func TestGetUserEmailsByNames ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
// ignore none active user email
2023-04-22 23:56:27 +02:00
assert . ElementsMatch ( t , [ ] string { "user8@example.com" } , user_model . GetUserEmailsByNames ( db . DefaultContext , [ ] string { "user8" , "user9" } ) )
assert . ElementsMatch ( t , [ ] string { "user8@example.com" , "user5@example.com" } , user_model . GetUserEmailsByNames ( db . DefaultContext , [ ] string { "user8" , "user5" } ) )
2021-11-24 10:49:20 +01:00
2023-09-14 04:59:53 +02:00
assert . ElementsMatch ( t , [ ] string { "user8@example.com" } , user_model . GetUserEmailsByNames ( db . DefaultContext , [ ] string { "user8" , "org7" } ) )
2021-11-24 10:49:20 +01:00
}
func TestCanCreateOrganization ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-08-16 04:22:25 +02:00
admin := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } )
2021-11-24 10:49:20 +01:00
assert . True ( t , admin . CanCreateOrganization ( ) )
2022-08-16 04:22:25 +02:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
2021-11-24 10:49:20 +01:00
assert . True ( t , user . CanCreateOrganization ( ) )
// Disable user create organization permission.
user . AllowCreateOrganization = false
assert . False ( t , user . CanCreateOrganization ( ) )
setting . Admin . DisableRegularOrgCreation = true
user . AllowCreateOrganization = true
assert . True ( t , admin . CanCreateOrganization ( ) )
assert . False ( t , user . CanCreateOrganization ( ) )
}
func TestSearchUsers ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-06-15 09:02:00 +02:00
testSuccess := func ( opts * user_model . SearchUserOptions , expectedUserOrOrgIDs [ ] int64 ) {
2023-09-14 19:09:32 +02:00
users , _ , err := user_model . SearchUsers ( db . DefaultContext , opts )
2021-11-24 10:49:20 +01:00
assert . NoError ( t , err )
2023-04-19 15:40:42 +02:00
cassText := fmt . Sprintf ( "ids: %v, opts: %v" , expectedUserOrOrgIDs , opts )
if assert . Len ( t , users , len ( expectedUserOrOrgIDs ) , "case: %s" , cassText ) {
2021-11-24 10:49:20 +01:00
for i , expectedID := range expectedUserOrOrgIDs {
2023-04-19 15:40:42 +02:00
assert . EqualValues ( t , expectedID , users [ i ] . ID , "case: %s" , cassText )
2021-11-24 10:49:20 +01:00
}
}
}
// test orgs
2022-06-15 09:02:00 +02:00
testOrgSuccess := func ( opts * user_model . SearchUserOptions , expectedOrgIDs [ ] int64 ) {
opts . Type = user_model . UserTypeOrganization
2021-11-24 10:49:20 +01:00
testSuccess ( opts , expectedOrgIDs )
}
2022-06-15 09:02:00 +02:00
testOrgSuccess ( & user_model . SearchUserOptions { OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 1 , PageSize : 2 } } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 3 , 6 } )
2022-06-15 09:02:00 +02:00
testOrgSuccess ( & user_model . SearchUserOptions { OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 2 , PageSize : 2 } } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 7 , 17 } )
2022-06-15 09:02:00 +02:00
testOrgSuccess ( & user_model . SearchUserOptions { OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 3 , PageSize : 2 } } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 19 , 25 } )
2022-06-15 09:02:00 +02:00
testOrgSuccess ( & user_model . SearchUserOptions { OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 4 , PageSize : 2 } } ,
2024-02-24 13:38:43 +01:00
[ ] int64 { 26 , 41 } )
2021-11-24 10:49:20 +01:00
2024-10-12 01:35:04 +02:00
testOrgSuccess ( & user_model . SearchUserOptions { OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 5 , PageSize : 2 } } ,
[ ] int64 { 42 } )
testOrgSuccess ( & user_model . SearchUserOptions { ListOptions : db . ListOptions { Page : 6 , PageSize : 2 } } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { } )
// test users
2022-06-15 09:02:00 +02:00
testUserSuccess := func ( opts * user_model . SearchUserOptions , expectedUserIDs [ ] int64 ) {
opts . Type = user_model . UserTypeIndividual
2021-11-24 10:49:20 +01:00
testSuccess ( opts , expectedUserIDs )
}
2022-06-15 09:02:00 +02:00
testUserSuccess ( & user_model . SearchUserOptions { OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 1 } } ,
2024-02-24 13:38:43 +01:00
[ ] int64 { 1 , 2 , 4 , 5 , 8 , 9 , 10 , 11 , 12 , 13 , 14 , 15 , 16 , 18 , 20 , 21 , 24 , 27 , 28 , 29 , 30 , 32 , 34 , 37 , 38 , 39 , 40 } )
2021-11-24 10:49:20 +01:00
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { ListOptions : db . ListOptions { Page : 1 } , IsActive : optional . Some ( false ) } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 9 } )
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 1 } , IsActive : optional . Some ( true ) } ,
2024-02-24 13:38:43 +01:00
[ ] int64 { 1 , 2 , 4 , 5 , 8 , 10 , 11 , 12 , 13 , 14 , 15 , 16 , 18 , 20 , 21 , 24 , 27 , 28 , 29 , 30 , 32 , 34 , 37 , 38 , 39 , 40 } )
2021-11-24 10:49:20 +01:00
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { Keyword : "user1" , OrderBy : "id ASC" , ListOptions : db . ListOptions { Page : 1 } , IsActive : optional . Some ( true ) } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 1 , 10 , 11 , 12 , 13 , 14 , 15 , 16 , 18 } )
// order by name asc default
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { Keyword : "user1" , ListOptions : db . ListOptions { Page : 1 } , IsActive : optional . Some ( true ) } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 1 , 10 , 11 , 12 , 13 , 14 , 15 , 16 , 18 } )
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { ListOptions : db . ListOptions { Page : 1 } , IsAdmin : optional . Some ( true ) } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 1 } )
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { ListOptions : db . ListOptions { Page : 1 } , IsRestricted : optional . Some ( true ) } ,
2023-04-19 15:40:42 +02:00
[ ] int64 { 29 } )
2021-11-24 10:49:20 +01:00
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { ListOptions : db . ListOptions { Page : 1 } , IsProhibitLogin : optional . Some ( true ) } ,
2024-02-04 14:29:09 +01:00
[ ] int64 { 37 } )
2021-11-24 10:49:20 +01:00
2024-02-29 19:52:49 +01:00
testUserSuccess ( & user_model . SearchUserOptions { ListOptions : db . ListOptions { Page : 1 } , IsTwoFactorEnabled : optional . Some ( true ) } ,
2021-11-24 10:49:20 +01:00
[ ] int64 { 24 } )
}
func TestEmailNotificationPreferences ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
for _ , test := range [ ] struct {
expected string
userID int64
} {
2022-06-15 09:02:00 +02:00
{ user_model . EmailNotificationsEnabled , 1 } ,
{ user_model . EmailNotificationsEnabled , 2 } ,
{ user_model . EmailNotificationsOnMention , 3 } ,
{ user_model . EmailNotificationsOnMention , 4 } ,
{ user_model . EmailNotificationsEnabled , 5 } ,
{ user_model . EmailNotificationsEnabled , 6 } ,
{ user_model . EmailNotificationsDisabled , 7 } ,
{ user_model . EmailNotificationsEnabled , 8 } ,
{ user_model . EmailNotificationsOnMention , 9 } ,
2021-11-24 10:49:20 +01:00
} {
2022-08-16 04:22:25 +02:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : test . userID } )
2024-02-04 14:29:09 +01:00
assert . Equal ( t , test . expected , user . EmailNotificationsPreference )
2021-11-24 10:49:20 +01:00
}
}
func TestHashPasswordDeterministic ( t * testing . T ) {
b := make ( [ ] byte , 16 )
2022-06-15 09:02:00 +02:00
u := & user_model . User { }
2023-02-19 08:35:20 +01:00
algos := hash . RecommendedHashAlgorithms
2021-11-24 10:49:20 +01:00
for j := 0 ; j < len ( algos ) ; j ++ {
u . PasswdHashAlgo = algos [ j ]
for i := 0 ; i < 50 ; i ++ {
// generate a random password
rand . Read ( b )
pass := string ( b )
// save the current password in the user - hash it and store the result
u . SetPassword ( pass )
r1 := u . Passwd
// run again
u . SetPassword ( pass )
r2 := u . Passwd
assert . NotEqual ( t , r1 , r2 )
assert . True ( t , u . ValidatePassword ( pass ) )
}
}
}
func BenchmarkHashPassword ( b * testing . B ) {
// BenchmarkHashPassword ensures that it takes a reasonable amount of time
// to hash a password - in order to protect from brute-force attacks.
pass := "password1337"
2022-06-15 09:02:00 +02:00
u := & user_model . User { Passwd : pass }
2021-11-24 10:49:20 +01:00
b . ResetTimer ( )
for i := 0 ; i < b . N ; i ++ {
u . SetPassword ( pass )
}
}
func TestNewGitSig ( t * testing . T ) {
2022-06-15 09:02:00 +02:00
users := make ( [ ] * user_model . User , 0 , 20 )
2021-11-24 10:49:20 +01:00
err := db . GetEngine ( db . DefaultContext ) . Find ( & users )
assert . NoError ( t , err )
for _ , user := range users {
sig := user . NewGitSig ( )
assert . NotContains ( t , sig . Name , "<" )
assert . NotContains ( t , sig . Name , ">" )
assert . NotContains ( t , sig . Name , "\n" )
2024-12-15 11:41:29 +01:00
assert . NotEmpty ( t , strings . TrimSpace ( sig . Name ) )
2021-11-24 10:49:20 +01:00
}
}
func TestDisplayName ( t * testing . T ) {
2022-06-15 09:02:00 +02:00
users := make ( [ ] * user_model . User , 0 , 20 )
2021-11-24 10:49:20 +01:00
err := db . GetEngine ( db . DefaultContext ) . Find ( & users )
assert . NoError ( t , err )
for _ , user := range users {
displayName := user . DisplayName ( )
assert . Equal ( t , strings . TrimSpace ( displayName ) , displayName )
if len ( strings . TrimSpace ( user . FullName ) ) == 0 {
assert . Equal ( t , user . Name , displayName )
}
2024-12-15 11:41:29 +01:00
assert . NotEmpty ( t , strings . TrimSpace ( displayName ) )
2021-11-24 10:49:20 +01:00
}
}
func TestCreateUserInvalidEmail ( t * testing . T ) {
2022-06-15 09:02:00 +02:00
user := & user_model . User {
2021-11-24 10:49:20 +01:00
Name : "GiteaBot" ,
Email : "GiteaBot@gitea.io\r\n" ,
Passwd : ";p['////..-++']" ,
IsAdmin : false ,
Theme : setting . UI . DefaultTheme ,
MustChangePassword : false ,
}
2024-09-09 23:05:16 +02:00
err := user_model . CreateUser ( db . DefaultContext , user , & user_model . Meta { } )
2021-11-24 10:49:20 +01:00
assert . Error ( t , err )
2022-06-15 09:02:00 +02:00
assert . True ( t , user_model . IsErrEmailCharIsNotSupported ( err ) )
2021-11-24 10:49:20 +01:00
}
2022-02-18 10:33:00 +01:00
func TestCreateUserEmailAlreadyUsed ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-08-16 04:22:25 +02:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
2022-02-18 10:33:00 +01:00
// add new user with user2's email
user . Name = "testuser"
user . LowerName = strings . ToLower ( user . Name )
user . ID = 0
2024-09-09 23:05:16 +02:00
err := user_model . CreateUser ( db . DefaultContext , user , & user_model . Meta { } )
2022-02-18 10:33:00 +01:00
assert . Error ( t , err )
2022-06-15 09:02:00 +02:00
assert . True ( t , user_model . IsErrEmailAlreadyUsed ( err ) )
2022-02-18 10:33:00 +01:00
}
2023-02-16 17:32:01 +01:00
func TestCreateUserCustomTimestamps ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
// Add new user with a custom creation timestamp.
var creationTimestamp timeutil . TimeStamp = 12345
user . Name = "testuser"
user . LowerName = strings . ToLower ( user . Name )
user . ID = 0
user . Email = "unique@example.com"
user . CreatedUnix = creationTimestamp
2024-09-09 23:05:16 +02:00
err := user_model . CreateUser ( db . DefaultContext , user , & user_model . Meta { } )
2023-02-16 17:32:01 +01:00
assert . NoError ( t , err )
fetched , err := user_model . GetUserByID ( context . Background ( ) , user . ID )
assert . NoError ( t , err )
assert . Equal ( t , creationTimestamp , fetched . CreatedUnix )
assert . Equal ( t , creationTimestamp , fetched . UpdatedUnix )
}
func TestCreateUserWithoutCustomTimestamps ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
// There is no way to use a mocked time for the XORM auto-time functionality,
// so use the real clock to approximate the expected timestamp.
timestampStart := time . Now ( ) . Unix ( )
// Add new user without a custom creation timestamp.
user . Name = "Testuser"
user . LowerName = strings . ToLower ( user . Name )
user . ID = 0
user . Email = "unique@example.com"
user . CreatedUnix = 0
user . UpdatedUnix = 0
2024-09-09 23:05:16 +02:00
err := user_model . CreateUser ( db . DefaultContext , user , & user_model . Meta { } )
2023-02-16 17:32:01 +01:00
assert . NoError ( t , err )
timestampEnd := time . Now ( ) . Unix ( )
fetched , err := user_model . GetUserByID ( context . Background ( ) , user . ID )
assert . NoError ( t , err )
assert . LessOrEqual ( t , timestampStart , fetched . CreatedUnix )
assert . LessOrEqual ( t , fetched . CreatedUnix , timestampEnd )
assert . LessOrEqual ( t , timestampStart , fetched . UpdatedUnix )
assert . LessOrEqual ( t , fetched . UpdatedUnix , timestampEnd )
}
2021-11-24 10:49:20 +01:00
func TestGetUserIDsByNames ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
// ignore non existing
2022-11-19 09:12:33 +01:00
IDs , err := user_model . GetUserIDsByNames ( db . DefaultContext , [ ] string { "user1" , "user2" , "none_existing_user" } , true )
2021-11-24 10:49:20 +01:00
assert . NoError ( t , err )
assert . Equal ( t , [ ] int64 { 1 , 2 } , IDs )
// ignore non existing
2022-11-19 09:12:33 +01:00
IDs , err = user_model . GetUserIDsByNames ( db . DefaultContext , [ ] string { "user1" , "do_not_exist" } , false )
2021-11-24 10:49:20 +01:00
assert . Error ( t , err )
assert . Equal ( t , [ ] int64 ( nil ) , IDs )
}
func TestGetMaileableUsersByIDs ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-11-19 09:12:33 +01:00
results , err := user_model . GetMaileableUsersByIDs ( db . DefaultContext , [ ] int64 { 1 , 4 } , false )
2021-11-24 10:49:20 +01:00
assert . NoError ( t , err )
assert . Len ( t , results , 1 )
if len ( results ) > 1 {
2024-12-15 11:41:29 +01:00
assert . Equal ( t , 1 , results [ 0 ] . ID )
2021-11-24 10:49:20 +01:00
}
2022-11-19 09:12:33 +01:00
results , err = user_model . GetMaileableUsersByIDs ( db . DefaultContext , [ ] int64 { 1 , 4 } , true )
2021-11-24 10:49:20 +01:00
assert . NoError ( t , err )
assert . Len ( t , results , 2 )
if len ( results ) > 2 {
2024-12-15 11:41:29 +01:00
assert . Equal ( t , 1 , results [ 0 ] . ID )
assert . Equal ( t , 4 , results [ 1 ] . ID )
2021-11-24 10:49:20 +01:00
}
}
func TestNewUserRedirect ( t * testing . T ) {
// redirect to a completely new name
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-08-16 04:22:25 +02:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } )
2022-06-15 09:02:00 +02:00
assert . NoError ( t , user_model . NewUserRedirect ( db . DefaultContext , user . ID , user . Name , "newusername" ) )
2021-11-24 10:49:20 +01:00
2022-06-15 09:02:00 +02:00
unittest . AssertExistsAndLoadBean ( t , & user_model . Redirect {
2021-11-24 10:49:20 +01:00
LowerName : user . LowerName ,
RedirectUserID : user . ID ,
} )
2022-06-15 09:02:00 +02:00
unittest . AssertExistsAndLoadBean ( t , & user_model . Redirect {
2021-11-24 10:49:20 +01:00
LowerName : "olduser1" ,
RedirectUserID : user . ID ,
} )
}
func TestNewUserRedirect2 ( t * testing . T ) {
// redirect to previously used name
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-08-16 04:22:25 +02:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } )
2022-06-15 09:02:00 +02:00
assert . NoError ( t , user_model . NewUserRedirect ( db . DefaultContext , user . ID , user . Name , "olduser1" ) )
2021-11-24 10:49:20 +01:00
2022-06-15 09:02:00 +02:00
unittest . AssertExistsAndLoadBean ( t , & user_model . Redirect {
2021-11-24 10:49:20 +01:00
LowerName : user . LowerName ,
RedirectUserID : user . ID ,
} )
2022-06-15 09:02:00 +02:00
unittest . AssertNotExistsBean ( t , & user_model . Redirect {
2021-11-24 10:49:20 +01:00
LowerName : "olduser1" ,
RedirectUserID : user . ID ,
} )
}
func TestNewUserRedirect3 ( t * testing . T ) {
// redirect for a previously-unredirected user
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2022-08-16 04:22:25 +02:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
2022-06-15 09:02:00 +02:00
assert . NoError ( t , user_model . NewUserRedirect ( db . DefaultContext , user . ID , user . Name , "newusername" ) )
2021-11-24 10:49:20 +01:00
2022-06-15 09:02:00 +02:00
unittest . AssertExistsAndLoadBean ( t , & user_model . Redirect {
2021-11-24 10:49:20 +01:00
LowerName : user . LowerName ,
RedirectUserID : user . ID ,
} )
}
func TestGetUserByOpenID ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2023-09-14 19:09:32 +02:00
_ , err := user_model . GetUserByOpenID ( db . DefaultContext , "https://unknown" )
2021-11-24 10:49:20 +01:00
if assert . Error ( t , err ) {
2022-06-15 09:02:00 +02:00
assert . True ( t , user_model . IsErrUserNotExist ( err ) )
2021-11-24 10:49:20 +01:00
}
2023-09-14 19:09:32 +02:00
user , err := user_model . GetUserByOpenID ( db . DefaultContext , "https://user1.domain1.tld" )
2021-11-24 10:49:20 +01:00
if assert . NoError ( t , err ) {
assert . Equal ( t , int64 ( 1 ) , user . ID )
}
2023-09-14 19:09:32 +02:00
user , err = user_model . GetUserByOpenID ( db . DefaultContext , "https://domain1.tld/user2/" )
2021-11-24 10:49:20 +01:00
if assert . NoError ( t , err ) {
assert . Equal ( t , int64 ( 2 ) , user . ID )
}
}
2022-06-15 09:02:00 +02:00
func TestFollowUser ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2024-03-04 09:16:03 +01:00
testSuccess := func ( follower , followed * user_model . User ) {
assert . NoError ( t , user_model . FollowUser ( db . DefaultContext , follower , followed ) )
unittest . AssertExistsAndLoadBean ( t , & user_model . Follow { UserID : follower . ID , FollowID : followed . ID } )
2022-06-15 09:02:00 +02:00
}
2024-03-04 09:16:03 +01:00
user2 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
user4 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 4 } )
user5 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 5 } )
testSuccess ( user4 , user2 )
testSuccess ( user5 , user2 )
assert . NoError ( t , user_model . FollowUser ( db . DefaultContext , user2 , user2 ) )
2022-06-15 09:02:00 +02:00
unittest . CheckConsistencyFor ( t , & user_model . User { } )
}
func TestUnfollowUser ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
testSuccess := func ( followerID , followedID int64 ) {
2023-09-16 16:39:12 +02:00
assert . NoError ( t , user_model . UnfollowUser ( db . DefaultContext , followerID , followedID ) )
2022-06-15 09:02:00 +02:00
unittest . AssertNotExistsBean ( t , & user_model . Follow { UserID : followerID , FollowID : followedID } )
}
testSuccess ( 4 , 2 )
testSuccess ( 5 , 2 )
testSuccess ( 2 , 2 )
unittest . CheckConsistencyFor ( t , & user_model . User { } )
}
2022-09-20 09:59:20 +02:00
func TestIsUserVisibleToViewer ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
user1 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } ) // admin, public
user4 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 4 } ) // normal, public
user20 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 20 } ) // public, same team as user31
user29 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 29 } ) // public, is restricted
user31 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 31 } ) // private, same team as user20
user33 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 33 } ) // limited, follows 31
test := func ( u , viewer * user_model . User , expected bool ) {
name := func ( u * user_model . User ) string {
if u == nil {
return "<nil>"
}
return u . Name
}
assert . Equal ( t , expected , user_model . IsUserVisibleToViewer ( db . DefaultContext , u , viewer ) , "user %v should be visible to viewer %v: %v" , name ( u ) , name ( viewer ) , expected )
}
// admin viewer
test ( user1 , user1 , true )
test ( user20 , user1 , true )
test ( user31 , user1 , true )
test ( user33 , user1 , true )
// non admin viewer
test ( user4 , user4 , true )
test ( user20 , user4 , true )
test ( user31 , user4 , false )
test ( user33 , user4 , true )
test ( user4 , nil , true )
// public user
test ( user4 , user20 , true )
test ( user4 , user31 , true )
test ( user4 , user33 , true )
// limited user
test ( user33 , user33 , true )
test ( user33 , user4 , true )
test ( user33 , user29 , false )
test ( user33 , nil , false )
// private user
test ( user31 , user31 , true )
test ( user31 , user4 , false )
test ( user31 , user20 , true )
test ( user31 , user29 , false )
test ( user31 , user33 , true )
test ( user31 , nil , false )
}
2023-05-24 09:30:55 +02:00
func Test_ValidateUser ( t * testing . T ) {
oldSetting := setting . Service . AllowedUserVisibilityModesSlice
defer func ( ) {
setting . Service . AllowedUserVisibilityModesSlice = oldSetting
} ( )
setting . Service . AllowedUserVisibilityModesSlice = [ ] bool { true , false , true }
kases := map [ * user_model . User ] bool {
2024-02-04 14:29:09 +01:00
{ ID : 1 , Visibility : structs . VisibleTypePublic } : true ,
{ ID : 2 , Visibility : structs . VisibleTypeLimited } : false ,
{ ID : 2 , Visibility : structs . VisibleTypePrivate } : true ,
2023-05-24 09:30:55 +02:00
}
for kase , expected := range kases {
2024-12-15 11:41:29 +01:00
assert . EqualValues ( t , expected , nil == user_model . ValidateUser ( kase ) , "case: %+v" , kase )
2023-05-24 09:30:55 +02:00
}
}
2024-01-04 01:48:20 +01:00
func Test_NormalizeUserFromEmail ( t * testing . T ) {
testCases := [ ] struct {
Input string
Expected string
IsNormalizedValid bool
} {
2024-04-25 13:22:32 +02:00
{ "name@example.com" , "name" , true } ,
{ "test'`´ name" , "testname" , true } ,
2024-01-04 01:48:20 +01:00
{ "Sinéad.O'Connor" , "Sinead.OConnor" , true } ,
{ "Æsir" , "AEsir" , true } ,
2024-04-25 13:22:32 +02:00
{ "éé" , "ee" , true } , // \u00e9\u0065\u0301
2024-01-04 01:48:20 +01:00
{ "Awareness Hub" , "Awareness-Hub" , true } ,
{ "double__underscore" , "double__underscore" , false } , // We should consider squashing double non-alpha characters
{ ".bad." , ".bad." , false } ,
{ "new😀user" , "new😀user" , false } , // No plans to support
2024-04-25 13:22:32 +02:00
{ ` "quoted" ` , ` "quoted" ` , false } , // No plans to support
2024-01-04 01:48:20 +01:00
}
for _ , testCase := range testCases {
normalizedName , err := user_model . NormalizeUserName ( testCase . Input )
assert . NoError ( t , err )
assert . EqualValues ( t , testCase . Expected , normalizedName )
if testCase . IsNormalizedValid {
assert . NoError ( t , user_model . IsUsableUsername ( normalizedName ) )
} else {
assert . Error ( t , user_model . IsUsableUsername ( normalizedName ) )
}
}
}
2024-03-29 16:05:41 +01:00
2024-07-08 10:38:45 +02:00
func TestEmailTo ( t * testing . T ) {
testCases := [ ] struct {
fullName string
mail string
result string
} {
{ "Awareness Hub" , "awareness@hub.net" , "Awareness Hub <awareness@hub.net>" } ,
{ "name@example.com" , "name@example.com" , "name@example.com" } ,
{ "Hi Its <Mee>" , "ee@mail.box" , "Hi Its Mee <ee@mail.box>" } ,
{ "Sinéad.O'Connor" , "sinead.oconnor@gmail.com" , "=?utf-8?q?Sin=C3=A9ad.O'Connor?= <sinead.oconnor@gmail.com>" } ,
{ "Æsir" , "aesir@gmx.de" , "=?utf-8?q?=C3=86sir?= <aesir@gmx.de>" } ,
{ "new😀user" , "new.user@alo.com" , "=?utf-8?q?new=F0=9F=98=80user?= <new.user@alo.com>" } ,
{ ` "quoted" ` , "quoted@test.com" , "quoted <quoted@test.com>" } ,
}
for _ , testCase := range testCases {
t . Run ( testCase . result , func ( t * testing . T ) {
testUser := & user_model . User { FullName : testCase . fullName , Email : testCase . mail }
assert . EqualValues ( t , testCase . result , testUser . EmailTo ( ) )
} )
}
}
2024-03-29 16:05:41 +01:00
func TestDisabledUserFeatures ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
testValues := container . SetOf ( setting . UserFeatureDeletion ,
setting . UserFeatureManageSSHKeys ,
setting . UserFeatureManageGPGKeys )
oldSetting := setting . Admin . ExternalUserDisableFeatures
defer func ( ) {
setting . Admin . ExternalUserDisableFeatures = oldSetting
} ( )
setting . Admin . ExternalUserDisableFeatures = testValues
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } )
2024-12-15 11:41:29 +01:00
assert . Empty ( t , setting . Admin . UserDisabledFeatures . Values ( ) )
2024-03-29 16:05:41 +01:00
// no features should be disabled with a plain login type
assert . LessOrEqual ( t , user . LoginType , auth . Plain )
2024-12-15 11:41:29 +01:00
assert . Empty ( t , user_model . DisabledFeaturesWithLoginType ( user ) . Values ( ) )
2024-03-29 16:05:41 +01:00
for _ , f := range testValues . Values ( ) {
assert . False ( t , user_model . IsFeatureDisabledWithLoginType ( user , f ) )
}
// check disabled features with external login type
user . LoginType = auth . OAuth2
// all features should be disabled
assert . NotEmpty ( t , user_model . DisabledFeaturesWithLoginType ( user ) . Values ( ) )
for _ , f := range testValues . Values ( ) {
assert . True ( t , user_model . IsFeatureDisabledWithLoginType ( user , f ) )
}
}
2024-11-21 05:55:32 +01:00
func TestGetInactiveUsers ( t * testing . T ) {
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
// all inactive users
// user1's createdunix is 1730468968
users , err := user_model . GetInactiveUsers ( db . DefaultContext , 0 )
assert . NoError ( t , err )
assert . Len ( t , users , 1 )
interval := time . Now ( ) . Unix ( ) - 1730468968 + 3600 * 24
users , err = user_model . GetInactiveUsers ( db . DefaultContext , time . Duration ( interval * int64 ( time . Second ) ) )
assert . NoError ( t , err )
2024-12-15 11:41:29 +01:00
assert . Empty ( t , users )
2024-11-21 05:55:32 +01:00
}