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