2018-05-10 01:29:04 +09:00
// Copyright 2018 The Gitea Authors. All rights reserved.
2022-11-28 03:20:29 +09:00
// SPDX-License-Identifier: MIT
2018-05-10 01:29:04 +09:00
2022-06-13 18:37:59 +09:00
package issues_test
2018-05-10 01:29:04 +09:00
import (
"testing"
2022-05-20 23:08:52 +09:00
"code.gitea.io/gitea/models/db"
2022-06-13 18:37:59 +09:00
issues_model "code.gitea.io/gitea/models/issues"
2021-11-12 23:36:47 +09:00
"code.gitea.io/gitea/models/unittest"
2021-11-24 18:49:20 +09:00
user_model "code.gitea.io/gitea/models/user"
2021-11-17 21:34:35 +09:00
2018-05-10 01:29:04 +09:00
"github.com/stretchr/testify/assert"
)
func TestUpdateAssignee ( t * testing . T ) {
2021-11-12 23:36:47 +09:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2018-05-10 01:29:04 +09:00
// Fake issue with assignees
2023-12-20 04:12:02 +09:00
issue , err := issues_model . GetIssueByID ( db . DefaultContext , 1 )
assert . NoError ( t , err )
err = issue . LoadAttributes ( db . DefaultContext )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
// Assign multiple users
2022-12-03 11:48:26 +09:00
user2 , err := user_model . GetUserByID ( db . DefaultContext , 2 )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
2023-04-15 03:18:28 +09:00
_ , _ , err = issues_model . ToggleIssueAssignee ( db . DefaultContext , issue , & user_model . User { ID : 1 } , user2 . ID )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
2023-09-14 11:59:53 +09:00
org3 , err := user_model . GetUserByID ( db . DefaultContext , 3 )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
2023-09-14 11:59:53 +09:00
_ , _ , err = issues_model . ToggleIssueAssignee ( db . DefaultContext , issue , & user_model . User { ID : 1 } , org3 . ID )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
2022-12-03 11:48:26 +09:00
user1 , err := user_model . GetUserByID ( db . DefaultContext , 1 ) // This user is already assigned (see the definition in fixtures), so running UpdateAssignee should unassign him
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
2023-04-15 03:18:28 +09:00
_ , _ , err = issues_model . ToggleIssueAssignee ( db . DefaultContext , issue , & user_model . User { ID : 1 } , user1 . ID )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
// Check if he got removed
2022-06-13 18:37:59 +09:00
isAssigned , err := issues_model . IsUserAssignedToIssue ( db . DefaultContext , issue , user1 )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
assert . False ( t , isAssigned )
// Check if they're all there
2022-05-20 23:08:52 +09:00
err = issue . LoadAssignees ( db . DefaultContext )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
2021-11-24 18:49:20 +09:00
var expectedAssignees [ ] * user_model . User
2023-09-14 11:59:53 +09:00
expectedAssignees = append ( expectedAssignees , user2 , org3 )
2018-05-10 01:29:04 +09:00
2022-05-20 23:08:52 +09:00
for in , assignee := range issue . Assignees {
2018-05-10 01:29:04 +09:00
assert . Equal ( t , assignee . ID , expectedAssignees [ in ] . ID )
}
// Check if the user is assigned
2022-06-13 18:37:59 +09:00
isAssigned , err = issues_model . IsUserAssignedToIssue ( db . DefaultContext , issue , user2 )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
assert . True ( t , isAssigned )
// This user should not be assigned
2022-06-13 18:37:59 +09:00
isAssigned , err = issues_model . IsUserAssignedToIssue ( db . DefaultContext , issue , & user_model . User { ID : 4 } )
2018-05-10 01:29:04 +09:00
assert . NoError ( t , err )
assert . False ( t , isAssigned )
}
2019-11-10 17:07:21 +09:00
func TestMakeIDsFromAPIAssigneesToAdd ( t * testing . T ) {
2021-11-12 23:36:47 +09:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2021-05-12 13:13:42 +09:00
2022-08-16 11:22:25 +09:00
_ = unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } )
_ = unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
2021-05-12 13:13:42 +09:00
2022-11-19 17:12:33 +09:00
IDs , err := issues_model . MakeIDsFromAPIAssigneesToAdd ( db . DefaultContext , "" , [ ] string { "" } )
2019-11-10 17:07:21 +09:00
assert . NoError ( t , err )
assert . Equal ( t , [ ] int64 { } , IDs )
2022-11-19 17:12:33 +09:00
_ , err = issues_model . MakeIDsFromAPIAssigneesToAdd ( db . DefaultContext , "" , [ ] string { "none_existing_user" } )
2019-11-10 17:07:21 +09:00
assert . Error ( t , err )
2022-11-19 17:12:33 +09:00
IDs , err = issues_model . MakeIDsFromAPIAssigneesToAdd ( db . DefaultContext , "user1" , [ ] string { "user1" } )
2019-11-10 17:07:21 +09:00
assert . NoError ( t , err )
assert . Equal ( t , [ ] int64 { 1 } , IDs )
2022-11-19 17:12:33 +09:00
IDs , err = issues_model . MakeIDsFromAPIAssigneesToAdd ( db . DefaultContext , "user2" , [ ] string { "" } )
2019-11-10 17:07:21 +09:00
assert . NoError ( t , err )
assert . Equal ( t , [ ] int64 { 2 } , IDs )
2022-11-19 17:12:33 +09:00
IDs , err = issues_model . MakeIDsFromAPIAssigneesToAdd ( db . DefaultContext , "" , [ ] string { "user1" , "user2" } )
2019-11-10 17:07:21 +09:00
assert . NoError ( t , err )
assert . Equal ( t , [ ] int64 { 1 , 2 } , IDs )
}