Unit tests for routers/repo/issue_label (#3198)
Ethan Koenig
7 years ago
committed by
Lauris BH
parent
bde0409433
commit
befa7445d2
@ -0,0 +1,14 @@ |
||||
// 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 test |
||||
|
||||
import ( |
||||
"net/http" |
||||
) |
||||
|
||||
// RedirectURL returns the redirect URL of a http response.
|
||||
func RedirectURL(resp http.ResponseWriter) string { |
||||
return resp.Header().Get("Location") |
||||
} |
||||
@ -0,0 +1,164 @@ |
||||
// 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" |
||||
|
||||
"code.gitea.io/gitea/models" |
||||
"code.gitea.io/gitea/modules/auth" |
||||
"code.gitea.io/gitea/modules/test" |
||||
|
||||
"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) { |
||||
models.PrepareTestEnv(t) |
||||
ctx := test.MockContext(t, "user2/repo1/labels/initialize") |
||||
test.LoadUser(t, ctx, 2) |
||||
test.LoadRepo(t, ctx, 2) |
||||
InitializeLabels(ctx, auth.InitializeLabelsForm{"Default"}) |
||||
assert.EqualValues(t, http.StatusFound, ctx.Resp.Status()) |
||||
models.AssertExistsAndLoadBean(t, &models.Label{ |
||||
RepoID: 2, |
||||
Name: "enhancement", |
||||
Color: "#84b6eb", |
||||
}) |
||||
assert.Equal(t, "/user2/repo2/labels", test.RedirectURL(ctx.Resp)) |
||||
} |
||||
|
||||
func TestRetrieveLabels(t *testing.T) { |
||||
models.PrepareTestEnv(t) |
||||
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()) |
||||
labels, ok := ctx.Data["Labels"].([]*models.Label) |
||||
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) { |
||||
models.PrepareTestEnv(t) |
||||
ctx := test.MockContext(t, "user2/repo1/labels/edit") |
||||
test.LoadUser(t, ctx, 2) |
||||
test.LoadRepo(t, ctx, 1) |
||||
NewLabel(ctx, auth.CreateLabelForm{ |
||||
Title: "newlabel", |
||||
Color: "#abcdef", |
||||
}) |
||||
assert.EqualValues(t, http.StatusFound, ctx.Resp.Status()) |
||||
models.AssertExistsAndLoadBean(t, &models.Label{ |
||||
Name: "newlabel", |
||||
Color: "#abcdef", |
||||
}) |
||||
assert.Equal(t, "/user2/repo1/labels", test.RedirectURL(ctx.Resp)) |
||||
} |
||||
|
||||
func TestUpdateLabel(t *testing.T) { |
||||
models.PrepareTestEnv(t) |
||||
ctx := test.MockContext(t, "user2/repo1/labels/edit") |
||||
test.LoadUser(t, ctx, 2) |
||||
test.LoadRepo(t, ctx, 1) |
||||
UpdateLabel(ctx, auth.CreateLabelForm{ |
||||
ID: 2, |
||||
Title: "newnameforlabel", |
||||
Color: "#abcdef", |
||||
}) |
||||
assert.EqualValues(t, http.StatusFound, ctx.Resp.Status()) |
||||
models.AssertExistsAndLoadBean(t, &models.Label{ |
||||
ID: 2, |
||||
Name: "newnameforlabel", |
||||
Color: "#abcdef", |
||||
}) |
||||
assert.Equal(t, "/user2/repo1/labels", test.RedirectURL(ctx.Resp)) |
||||
} |
||||
|
||||
func TestDeleteLabel(t *testing.T) { |
||||
models.PrepareTestEnv(t) |
||||
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()) |
||||
models.AssertNotExistsBean(t, &models.Label{ID: 2}) |
||||
models.AssertNotExistsBean(t, &models.IssueLabel{LabelID: 2}) |
||||
assert.Equal(t, ctx.Tr("repo.issues.label_deletion_success"), ctx.Flash.SuccessMsg) |
||||
} |
||||
|
||||
func TestUpdateIssueLabel_Clear(t *testing.T) { |
||||
models.PrepareTestEnv(t) |
||||
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()) |
||||
models.AssertNotExistsBean(t, &models.IssueLabel{IssueID: 1}) |
||||
models.AssertNotExistsBean(t, &models.IssueLabel{IssueID: 3}) |
||||
models.CheckConsistencyFor(t, &models.Label{}) |
||||
} |
||||
|
||||
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}, |
||||
} { |
||||
models.PrepareTestEnv(t) |
||||
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 { |
||||
models.AssertExistsIf(t, testCase.ExpectedAdd, &models.IssueLabel{ |
||||
IssueID: issueID, |
||||
LabelID: testCase.LabelID, |
||||
}) |
||||
} |
||||
models.CheckConsistencyFor(t, &models.Label{}) |
||||
} |
||||
} |
||||
Loading…
Reference in new issue