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