2017-12-15 13:11:02 -08:00
// 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 repo
import (
"net/http"
"strconv"
"testing"
2022-06-13 17:37:59 +08:00
issues_model "code.gitea.io/gitea/models/issues"
2021-11-12 22:36:47 +08:00
"code.gitea.io/gitea/models/unittest"
2017-12-15 13:11:02 -08:00
"code.gitea.io/gitea/modules/test"
2021-01-26 23:36:53 +08:00
"code.gitea.io/gitea/modules/web"
2021-04-06 20:44:05 +01:00
"code.gitea.io/gitea/services/forms"
2017-12-15 13:11:02 -08:00
"github.com/stretchr/testify/assert"
)
func int64SliceToCommaSeparated ( a [ ] int64 ) string {
s := ""
for i , n := range a {
if i > 0 {
s += ","
}
s += strconv . Itoa ( int ( n ) )
}
return s
}
func TestInitializeLabels ( t * testing . T ) {
2021-11-12 22:36:47 +08:00
unittest . PrepareTestEnv ( t )
2017-12-15 13:11:02 -08:00
ctx := test . MockContext ( t , "user2/repo1/labels/initialize" )
test . LoadUser ( t , ctx , 2 )
test . LoadRepo ( t , ctx , 2 )
2021-04-06 20:44:05 +01:00
web . SetForm ( ctx , & forms . InitializeLabelsForm { TemplateName : "Default" } )
2021-01-26 23:36:53 +08:00
InitializeLabels ( ctx )
2022-03-23 05:54:07 +01:00
assert . EqualValues ( t , http . StatusSeeOther , ctx . Resp . Status ( ) )
2022-06-13 17:37:59 +08:00
unittest . AssertExistsAndLoadBean ( t , & issues_model . Label {
2017-12-15 13:11:02 -08:00
RepoID : 2 ,
Name : "enhancement" ,
Color : "#84b6eb" ,
} )
assert . Equal ( t , "/user2/repo2/labels" , test . RedirectURL ( ctx . Resp ) )
}
func TestRetrieveLabels ( t * testing . T ) {
2021-11-12 22:36:47 +08:00
unittest . PrepareTestEnv ( t )
2017-12-15 13:11:02 -08:00
for _ , testCase := range [ ] struct {
RepoID int64
Sort string
ExpectedLabelIDs [ ] int64
} {
{ 1 , "" , [ ] int64 { 1 , 2 } } ,
{ 1 , "leastissues" , [ ] int64 { 2 , 1 } } ,
{ 2 , "" , [ ] int64 { } } ,
} {
ctx := test . MockContext ( t , "user/repo/issues" )
test . LoadUser ( t , ctx , 2 )
test . LoadRepo ( t , ctx , testCase . RepoID )
ctx . Req . Form . Set ( "sort" , testCase . Sort )
RetrieveLabels ( ctx )
assert . False ( t , ctx . Written ( ) )
2022-06-13 17:37:59 +08:00
labels , ok := ctx . Data [ "Labels" ] . ( [ ] * issues_model . Label )
2017-12-15 13:11:02 -08:00
assert . True ( t , ok )
if assert . Len ( t , labels , len ( testCase . ExpectedLabelIDs ) ) {
for i , label := range labels {
assert . EqualValues ( t , testCase . ExpectedLabelIDs [ i ] , label . ID )
}
}
}
}
func TestNewLabel ( t * testing . T ) {
2021-11-12 22:36:47 +08:00
unittest . PrepareTestEnv ( t )
2017-12-15 13:11:02 -08:00
ctx := test . MockContext ( t , "user2/repo1/labels/edit" )
test . LoadUser ( t , ctx , 2 )
test . LoadRepo ( t , ctx , 1 )
2021-04-06 20:44:05 +01:00
web . SetForm ( ctx , & forms . CreateLabelForm {
2017-12-15 13:11:02 -08:00
Title : "newlabel" ,
Color : "#abcdef" ,
} )
2021-01-26 23:36:53 +08:00
NewLabel ( ctx )
2022-03-23 05:54:07 +01:00
assert . EqualValues ( t , http . StatusSeeOther , ctx . Resp . Status ( ) )
2022-06-13 17:37:59 +08:00
unittest . AssertExistsAndLoadBean ( t , & issues_model . Label {
2017-12-15 13:11:02 -08:00
Name : "newlabel" ,
Color : "#abcdef" ,
} )
assert . Equal ( t , "/user2/repo1/labels" , test . RedirectURL ( ctx . Resp ) )
}
func TestUpdateLabel ( t * testing . T ) {
2021-11-12 22:36:47 +08:00
unittest . PrepareTestEnv ( t )
2017-12-15 13:11:02 -08:00
ctx := test . MockContext ( t , "user2/repo1/labels/edit" )
test . LoadUser ( t , ctx , 2 )
test . LoadRepo ( t , ctx , 1 )
2021-04-06 20:44:05 +01:00
web . SetForm ( ctx , & forms . CreateLabelForm {
2017-12-15 13:11:02 -08:00
ID : 2 ,
Title : "newnameforlabel" ,
Color : "#abcdef" ,
} )
2021-01-26 23:36:53 +08:00
UpdateLabel ( ctx )
2022-03-23 05:54:07 +01:00
assert . EqualValues ( t , http . StatusSeeOther , ctx . Resp . Status ( ) )
2022-06-13 17:37:59 +08:00
unittest . AssertExistsAndLoadBean ( t , & issues_model . Label {
2017-12-15 13:11:02 -08:00
ID : 2 ,
Name : "newnameforlabel" ,
Color : "#abcdef" ,
} )
assert . Equal ( t , "/user2/repo1/labels" , test . RedirectURL ( ctx . Resp ) )
}
func TestDeleteLabel ( t * testing . T ) {
2021-11-12 22:36:47 +08:00
unittest . PrepareTestEnv ( t )
2017-12-15 13:11:02 -08:00
ctx := test . MockContext ( t , "user2/repo1/labels/delete" )
test . LoadUser ( t , ctx , 2 )
test . LoadRepo ( t , ctx , 1 )
ctx . Req . Form . Set ( "id" , "2" )
DeleteLabel ( ctx )
assert . EqualValues ( t , http . StatusOK , ctx . Resp . Status ( ) )
2022-06-13 17:37:59 +08:00
unittest . AssertNotExistsBean ( t , & issues_model . Label { ID : 2 } )
unittest . AssertNotExistsBean ( t , & issues_model . IssueLabel { LabelID : 2 } )
2017-12-15 13:11:02 -08:00
assert . Equal ( t , ctx . Tr ( "repo.issues.label_deletion_success" ) , ctx . Flash . SuccessMsg )
}
func TestUpdateIssueLabel_Clear ( t * testing . T ) {
2021-11-12 22:36:47 +08:00
unittest . PrepareTestEnv ( t )
2017-12-15 13:11:02 -08:00
ctx := test . MockContext ( t , "user2/repo1/issues/labels" )
test . LoadUser ( t , ctx , 2 )
test . LoadRepo ( t , ctx , 1 )
ctx . Req . Form . Set ( "issue_ids" , "1,3" )
ctx . Req . Form . Set ( "action" , "clear" )
UpdateIssueLabel ( ctx )
assert . EqualValues ( t , http . StatusOK , ctx . Resp . Status ( ) )
2022-06-13 17:37:59 +08:00
unittest . AssertNotExistsBean ( t , & issues_model . IssueLabel { IssueID : 1 } )
unittest . AssertNotExistsBean ( t , & issues_model . IssueLabel { IssueID : 3 } )
unittest . CheckConsistencyFor ( t , & issues_model . Label { } )
2017-12-15 13:11:02 -08:00
}
func TestUpdateIssueLabel_Toggle ( t * testing . T ) {
for _ , testCase := range [ ] struct {
Action string
IssueIDs [ ] int64
LabelID int64
ExpectedAdd bool // whether we expect the label to be added to the issues
} {
{ "attach" , [ ] int64 { 1 , 3 } , 1 , true } ,
{ "detach" , [ ] int64 { 1 , 3 } , 1 , false } ,
{ "toggle" , [ ] int64 { 1 , 3 } , 1 , false } ,
{ "toggle" , [ ] int64 { 1 , 2 } , 2 , true } ,
} {
2021-11-12 22:36:47 +08:00
unittest . PrepareTestEnv ( t )
2017-12-15 13:11:02 -08:00
ctx := test . MockContext ( t , "user2/repo1/issues/labels" )
test . LoadUser ( t , ctx , 2 )
test . LoadRepo ( t , ctx , 1 )
ctx . Req . Form . Set ( "issue_ids" , int64SliceToCommaSeparated ( testCase . IssueIDs ) )
ctx . Req . Form . Set ( "action" , testCase . Action )
ctx . Req . Form . Set ( "id" , strconv . Itoa ( int ( testCase . LabelID ) ) )
UpdateIssueLabel ( ctx )
assert . EqualValues ( t , http . StatusOK , ctx . Resp . Status ( ) )
for _ , issueID := range testCase . IssueIDs {
2022-06-13 17:37:59 +08:00
unittest . AssertExistsIf ( t , testCase . ExpectedAdd , & issues_model . IssueLabel {
2017-12-15 13:11:02 -08:00
IssueID : issueID ,
LabelID : testCase . LabelID ,
} )
}
2022-06-13 17:37:59 +08:00
unittest . CheckConsistencyFor ( t , & issues_model . Label { } )
2017-12-15 13:11:02 -08:00
}
}