forked from NYANDEV/forgejo
dc9499bdf9
- Add the ability to block a user via their profile page. - This will unstar their repositories and visa versa. - Blocked users cannot create issues or pull requests on your the doer's repositories (mind that this is not the case for organizations). - Blocked users cannot comment on the doer's opened issues or pull requests. - Blocked users cannot add reactions to doer's comments. - Blocked users cannot cause a notification trough mentioning the doer. Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/540 (cherry picked from commit 687d852480388897db4d7b0cb397cf7135ab97b1) (cherry picked from commit 0c32a4fde531018f74e01d9db6520895fcfa10cc) (cherry picked from commit 1791130e3cb8470b9b39742e0004d5e4c7d1e64d) (cherry picked from commit 00f411819f62c02016d46602ab4daf49effe0550) (cherry picked from commit e0c039b0e899e787a8df1efdd6b47388d93e08fa) (cherry picked from commit b5a058ef0039e95be23893e6fefdcb62a7de071a) (cherry picked from commit 5ff5460d28a482526da7e77bffb18d08de14aaaa) (cherry picked from commit 97bc6e619d2970839b8692b7b025ff0ec1c96d12)
588 lines
18 KiB
Go
588 lines
18 KiB
Go
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package issues_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
issues_model "code.gitea.io/gitea/models/issues"
|
|
"code.gitea.io/gitea/models/organization"
|
|
repo_model "code.gitea.io/gitea/models/repo"
|
|
"code.gitea.io/gitea/models/unittest"
|
|
user_model "code.gitea.io/gitea/models/user"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"code.gitea.io/gitea/modules/util"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"xorm.io/builder"
|
|
)
|
|
|
|
func TestIssue_ReplaceLabels(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
testSuccess := func(issueID int64, labelIDs, expectedLabelIDs []int64) {
|
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: issueID})
|
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: issue.RepoID})
|
|
doer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID})
|
|
|
|
labels := make([]*issues_model.Label, len(labelIDs))
|
|
for i, labelID := range labelIDs {
|
|
labels[i] = unittest.AssertExistsAndLoadBean(t, &issues_model.Label{ID: labelID, RepoID: repo.ID})
|
|
}
|
|
assert.NoError(t, issues_model.ReplaceIssueLabels(issue, labels, doer))
|
|
unittest.AssertCount(t, &issues_model.IssueLabel{IssueID: issueID}, len(expectedLabelIDs))
|
|
for _, labelID := range expectedLabelIDs {
|
|
unittest.AssertExistsAndLoadBean(t, &issues_model.IssueLabel{IssueID: issueID, LabelID: labelID})
|
|
}
|
|
}
|
|
|
|
testSuccess(1, []int64{2}, []int64{2})
|
|
testSuccess(1, []int64{1, 2}, []int64{1, 2})
|
|
testSuccess(1, []int64{}, []int64{})
|
|
|
|
// mutually exclusive scoped labels 7 and 8
|
|
testSuccess(18, []int64{6, 7}, []int64{6, 7})
|
|
testSuccess(18, []int64{7, 8}, []int64{8})
|
|
testSuccess(18, []int64{6, 8, 7}, []int64{6, 7})
|
|
}
|
|
|
|
func Test_GetIssueIDsByRepoID(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
ids, err := issues_model.GetIssueIDsByRepoID(db.DefaultContext, 1)
|
|
assert.NoError(t, err)
|
|
assert.Len(t, ids, 5)
|
|
}
|
|
|
|
func TestIssueAPIURL(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 1})
|
|
err := issue.LoadAttributes(db.DefaultContext)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "https://try.gitea.io/api/v1/repos/user2/repo1/issues/1", issue.APIURL())
|
|
}
|
|
|
|
func TestGetIssuesByIDs(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
testSuccess := func(expectedIssueIDs, nonExistentIssueIDs []int64) {
|
|
issues, err := issues_model.GetIssuesByIDs(db.DefaultContext, append(expectedIssueIDs, nonExistentIssueIDs...))
|
|
assert.NoError(t, err)
|
|
actualIssueIDs := make([]int64, len(issues))
|
|
for i, issue := range issues {
|
|
actualIssueIDs[i] = issue.ID
|
|
}
|
|
assert.Equal(t, expectedIssueIDs, actualIssueIDs)
|
|
}
|
|
testSuccess([]int64{1, 2, 3}, []int64{})
|
|
testSuccess([]int64{1, 2, 3}, []int64{unittest.NonexistentID})
|
|
}
|
|
|
|
func TestGetParticipantIDsByIssue(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
checkParticipants := func(issueID int64, userIDs []int) {
|
|
issue, err := issues_model.GetIssueByID(db.DefaultContext, issueID)
|
|
assert.NoError(t, err)
|
|
participants, err := issue.GetParticipantIDsByIssue(db.DefaultContext)
|
|
if assert.NoError(t, err) {
|
|
participantsIDs := make([]int, len(participants))
|
|
for i, uid := range participants {
|
|
participantsIDs[i] = int(uid)
|
|
}
|
|
sort.Ints(participantsIDs)
|
|
sort.Ints(userIDs)
|
|
assert.Equal(t, userIDs, participantsIDs)
|
|
}
|
|
}
|
|
|
|
// User 1 is issue1 poster (see fixtures/issue.yml)
|
|
// User 2 only labeled issue1 (see fixtures/comment.yml)
|
|
// Users 3 and 5 made actual comments (see fixtures/comment.yml)
|
|
// User 3 is inactive, thus not active participant
|
|
checkParticipants(1, []int{1, 5})
|
|
}
|
|
|
|
func TestIssue_ClearLabels(t *testing.T) {
|
|
tests := []struct {
|
|
issueID int64
|
|
doerID int64
|
|
}{
|
|
{1, 2}, // non-pull-request, has labels
|
|
{2, 2}, // pull-request, has labels
|
|
{3, 2}, // pull-request, has no labels
|
|
}
|
|
for _, test := range tests {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: test.issueID})
|
|
doer := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: test.doerID})
|
|
assert.NoError(t, issues_model.ClearIssueLabels(issue, doer))
|
|
unittest.AssertNotExistsBean(t, &issues_model.IssueLabel{IssueID: test.issueID})
|
|
}
|
|
}
|
|
|
|
func TestUpdateIssueCols(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{})
|
|
|
|
const newTitle = "New Title for unit test"
|
|
issue.Title = newTitle
|
|
|
|
prevContent := issue.Content
|
|
issue.Content = "This should have no effect"
|
|
|
|
now := time.Now().Unix()
|
|
assert.NoError(t, issues_model.UpdateIssueCols(db.DefaultContext, issue, "name"))
|
|
then := time.Now().Unix()
|
|
|
|
updatedIssue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: issue.ID})
|
|
assert.EqualValues(t, newTitle, updatedIssue.Title)
|
|
assert.EqualValues(t, prevContent, updatedIssue.Content)
|
|
unittest.AssertInt64InRange(t, now, then, int64(updatedIssue.UpdatedUnix))
|
|
}
|
|
|
|
func TestIssues(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
for _, test := range []struct {
|
|
Opts issues_model.IssuesOptions
|
|
ExpectedIssueIDs []int64
|
|
}{
|
|
{
|
|
issues_model.IssuesOptions{
|
|
AssigneeID: 1,
|
|
SortType: "oldest",
|
|
},
|
|
[]int64{1, 6},
|
|
},
|
|
{
|
|
issues_model.IssuesOptions{
|
|
RepoCond: builder.In("repo_id", 1, 3),
|
|
SortType: "oldest",
|
|
ListOptions: db.ListOptions{
|
|
Page: 1,
|
|
PageSize: 4,
|
|
},
|
|
},
|
|
[]int64{1, 2, 3, 5},
|
|
},
|
|
{
|
|
issues_model.IssuesOptions{
|
|
LabelIDs: []int64{1},
|
|
ListOptions: db.ListOptions{
|
|
Page: 1,
|
|
PageSize: 4,
|
|
},
|
|
},
|
|
[]int64{2, 1},
|
|
},
|
|
{
|
|
issues_model.IssuesOptions{
|
|
LabelIDs: []int64{1, 2},
|
|
ListOptions: db.ListOptions{
|
|
Page: 1,
|
|
PageSize: 4,
|
|
},
|
|
},
|
|
[]int64{}, // issues with **both** label 1 and 2, none of these issues matches, TODO: add more tests
|
|
},
|
|
} {
|
|
issues, err := issues_model.Issues(db.DefaultContext, &test.Opts)
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, issues, len(test.ExpectedIssueIDs)) {
|
|
for i, issue := range issues {
|
|
assert.EqualValues(t, test.ExpectedIssueIDs[i], issue.ID)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetUserIssueStats(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
for _, test := range []struct {
|
|
FilterMode int
|
|
Opts issues_model.IssuesOptions
|
|
ExpectedIssueStats issues_model.IssueStats
|
|
}{
|
|
{
|
|
issues_model.FilterModeAll,
|
|
issues_model.IssuesOptions{
|
|
User: unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}),
|
|
RepoIDs: []int64{1},
|
|
IsPull: util.OptionalBoolFalse,
|
|
},
|
|
issues_model.IssueStats{
|
|
YourRepositoriesCount: 1, // 6
|
|
AssignCount: 1, // 6
|
|
CreateCount: 1, // 6
|
|
OpenCount: 1, // 6
|
|
ClosedCount: 1, // 1
|
|
},
|
|
},
|
|
{
|
|
issues_model.FilterModeAll,
|
|
issues_model.IssuesOptions{
|
|
User: unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}),
|
|
RepoIDs: []int64{1},
|
|
IsPull: util.OptionalBoolFalse,
|
|
IsClosed: util.OptionalBoolTrue,
|
|
},
|
|
issues_model.IssueStats{
|
|
YourRepositoriesCount: 1, // 6
|
|
AssignCount: 0,
|
|
CreateCount: 0,
|
|
OpenCount: 1, // 6
|
|
ClosedCount: 1, // 1
|
|
},
|
|
},
|
|
{
|
|
issues_model.FilterModeAssign,
|
|
issues_model.IssuesOptions{
|
|
User: unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}),
|
|
IsPull: util.OptionalBoolFalse,
|
|
},
|
|
issues_model.IssueStats{
|
|
YourRepositoriesCount: 1, // 6
|
|
AssignCount: 1, // 6
|
|
CreateCount: 1, // 6
|
|
OpenCount: 1, // 6
|
|
ClosedCount: 0,
|
|
},
|
|
},
|
|
{
|
|
issues_model.FilterModeCreate,
|
|
issues_model.IssuesOptions{
|
|
User: unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}),
|
|
IsPull: util.OptionalBoolFalse,
|
|
},
|
|
issues_model.IssueStats{
|
|
YourRepositoriesCount: 1, // 6
|
|
AssignCount: 1, // 6
|
|
CreateCount: 1, // 6
|
|
OpenCount: 1, // 6
|
|
ClosedCount: 0,
|
|
},
|
|
},
|
|
{
|
|
issues_model.FilterModeMention,
|
|
issues_model.IssuesOptions{
|
|
User: unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}),
|
|
IsPull: util.OptionalBoolFalse,
|
|
},
|
|
issues_model.IssueStats{
|
|
YourRepositoriesCount: 1, // 6
|
|
AssignCount: 1, // 6
|
|
CreateCount: 1, // 6
|
|
MentionCount: 0,
|
|
OpenCount: 0,
|
|
ClosedCount: 0,
|
|
},
|
|
},
|
|
{
|
|
issues_model.FilterModeCreate,
|
|
issues_model.IssuesOptions{
|
|
User: unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1}),
|
|
IssueIDs: []int64{1},
|
|
IsPull: util.OptionalBoolFalse,
|
|
},
|
|
issues_model.IssueStats{
|
|
YourRepositoriesCount: 1, // 1
|
|
AssignCount: 1, // 1
|
|
CreateCount: 1, // 1
|
|
OpenCount: 1, // 1
|
|
ClosedCount: 0,
|
|
},
|
|
},
|
|
{
|
|
issues_model.FilterModeAll,
|
|
issues_model.IssuesOptions{
|
|
User: unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2}),
|
|
Org: unittest.AssertExistsAndLoadBean(t, &organization.Organization{ID: 3}),
|
|
Team: unittest.AssertExistsAndLoadBean(t, &organization.Team{ID: 7}),
|
|
IsPull: util.OptionalBoolFalse,
|
|
},
|
|
issues_model.IssueStats{
|
|
YourRepositoriesCount: 2,
|
|
AssignCount: 1,
|
|
CreateCount: 1,
|
|
OpenCount: 2,
|
|
},
|
|
},
|
|
} {
|
|
t.Run(fmt.Sprintf("%#v", test.Opts), func(t *testing.T) {
|
|
stats, err := issues_model.GetUserIssueStats(test.FilterMode, test.Opts)
|
|
if !assert.NoError(t, err) {
|
|
return
|
|
}
|
|
assert.Equal(t, test.ExpectedIssueStats, *stats)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestIssue_loadTotalTimes(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
ms, err := issues_model.GetIssueByID(db.DefaultContext, 2)
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, ms.LoadTotalTimes(db.DefaultContext))
|
|
assert.Equal(t, int64(3682), ms.TotalTrackedTime)
|
|
}
|
|
|
|
func TestIssue_SearchIssueIDsByKeyword(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
total, ids, err := issues_model.SearchIssueIDsByKeyword(context.TODO(), "issue2", []int64{1}, 10, 0)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, total)
|
|
assert.EqualValues(t, []int64{2}, ids)
|
|
|
|
total, ids, err = issues_model.SearchIssueIDsByKeyword(context.TODO(), "first", []int64{1}, 10, 0)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, total)
|
|
assert.EqualValues(t, []int64{1}, ids)
|
|
|
|
total, ids, err = issues_model.SearchIssueIDsByKeyword(context.TODO(), "for", []int64{1}, 10, 0)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 5, total)
|
|
assert.ElementsMatch(t, []int64{1, 2, 3, 5, 11}, ids)
|
|
|
|
// issue1's comment id 2
|
|
total, ids, err = issues_model.SearchIssueIDsByKeyword(context.TODO(), "good", []int64{1}, 10, 0)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, total)
|
|
assert.EqualValues(t, []int64{1}, ids)
|
|
}
|
|
|
|
func TestGetRepoIDsForIssuesOptions(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
for _, test := range []struct {
|
|
Opts issues_model.IssuesOptions
|
|
ExpectedRepoIDs []int64
|
|
}{
|
|
{
|
|
issues_model.IssuesOptions{
|
|
AssigneeID: 2,
|
|
},
|
|
[]int64{3, 32},
|
|
},
|
|
{
|
|
issues_model.IssuesOptions{
|
|
RepoCond: builder.In("repo_id", 1, 2),
|
|
},
|
|
[]int64{1, 2},
|
|
},
|
|
} {
|
|
repoIDs, err := issues_model.GetRepoIDsForIssuesOptions(&test.Opts, user)
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, repoIDs, len(test.ExpectedRepoIDs)) {
|
|
for i, repoID := range repoIDs {
|
|
assert.EqualValues(t, test.ExpectedRepoIDs[i], repoID)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func testInsertIssue(t *testing.T, title, content string, expectIndex int64) *issues_model.Issue {
|
|
var newIssue issues_model.Issue
|
|
t.Run(title, func(t *testing.T) {
|
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
|
|
issue := issues_model.Issue{
|
|
RepoID: repo.ID,
|
|
PosterID: user.ID,
|
|
Poster: user,
|
|
Title: title,
|
|
Content: content,
|
|
}
|
|
err := issues_model.NewIssue(repo, &issue, nil, nil)
|
|
assert.NoError(t, err)
|
|
|
|
has, err := db.GetEngine(db.DefaultContext).ID(issue.ID).Get(&newIssue)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, issue.Title, newIssue.Title)
|
|
assert.EqualValues(t, issue.Content, newIssue.Content)
|
|
if expectIndex > 0 {
|
|
assert.EqualValues(t, expectIndex, newIssue.Index)
|
|
}
|
|
})
|
|
return &newIssue
|
|
}
|
|
|
|
func TestIssue_InsertIssue(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
// there are 5 issues and max index is 5 on repository 1, so this one should 6
|
|
issue := testInsertIssue(t, "my issue1", "special issue's comments?", 6)
|
|
_, err := db.GetEngine(db.DefaultContext).ID(issue.ID).Delete(new(issues_model.Issue))
|
|
assert.NoError(t, err)
|
|
|
|
issue = testInsertIssue(t, `my issue2, this is my son's love \n \r \ `, "special issue's '' comments?", 7)
|
|
_, err = db.GetEngine(db.DefaultContext).ID(issue.ID).Delete(new(issues_model.Issue))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestIssue_ResolveMentions(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
testSuccess := func(owner, repo, doer string, mentions []string, expected []int64) {
|
|
o := unittest.AssertExistsAndLoadBean(t, &user_model.User{LowerName: owner})
|
|
r := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{OwnerID: o.ID, LowerName: repo})
|
|
issue := &issues_model.Issue{RepoID: r.ID}
|
|
d := unittest.AssertExistsAndLoadBean(t, &user_model.User{LowerName: doer})
|
|
resolved, err := issues_model.ResolveIssueMentionsByVisibility(db.DefaultContext, issue, d, mentions)
|
|
assert.NoError(t, err)
|
|
ids := make([]int64, len(resolved))
|
|
for i, user := range resolved {
|
|
ids[i] = user.ID
|
|
}
|
|
sort.Slice(ids, func(i, j int) bool { return ids[i] < ids[j] })
|
|
assert.EqualValues(t, expected, ids)
|
|
}
|
|
|
|
// Public repo, existing user
|
|
testSuccess("user2", "repo1", "user1", []string{"user5"}, []int64{5})
|
|
// Public repo, non-existing user
|
|
testSuccess("user2", "repo1", "user1", []string{"nonexisting"}, []int64{})
|
|
// Public repo, doer
|
|
testSuccess("user2", "repo1", "user1", []string{"user1"}, []int64{})
|
|
// Public repo, blocked user
|
|
testSuccess("user2", "repo1", "user1", []string{"user4"}, []int64{})
|
|
// Private repo, team member
|
|
testSuccess("user17", "big_test_private_4", "user20", []string{"user2"}, []int64{2})
|
|
// Private repo, not a team member
|
|
testSuccess("user17", "big_test_private_4", "user20", []string{"user5"}, []int64{})
|
|
// Private repo, whole team
|
|
testSuccess("user17", "big_test_private_4", "user15", []string{"user17/owners"}, []int64{18})
|
|
}
|
|
|
|
func TestResourceIndex(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
var wg sync.WaitGroup
|
|
for i := 0; i < 100; i++ {
|
|
wg.Add(1)
|
|
go func(i int) {
|
|
testInsertIssue(t, fmt.Sprintf("issue %d", i+1), "my issue", 0)
|
|
wg.Done()
|
|
}(i)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func TestCorrectIssueStats(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
// Because the condition is to have chunked database look-ups,
|
|
// We have to more issues than `maxQueryParameters`, we will insert.
|
|
// maxQueryParameters + 10 issues into the testDatabase.
|
|
// Each new issues will have a constant description "Bugs are nasty"
|
|
// Which will be used later on.
|
|
|
|
issueAmount := issues_model.MaxQueryParameters + 10
|
|
|
|
var wg sync.WaitGroup
|
|
for i := 0; i < issueAmount; i++ {
|
|
wg.Add(1)
|
|
go func(i int) {
|
|
testInsertIssue(t, fmt.Sprintf("Issue %d", i+1), "Bugs are nasty", 0)
|
|
wg.Done()
|
|
}(i)
|
|
}
|
|
wg.Wait()
|
|
|
|
// Now we will get all issueID's that match the "Bugs are nasty" query.
|
|
total, ids, err := issues_model.SearchIssueIDsByKeyword(context.TODO(), "Bugs are nasty", []int64{1}, issueAmount, 0)
|
|
|
|
// Just to be sure.
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, issueAmount, total)
|
|
|
|
// Now we will call the GetIssueStats with these IDs and if working,
|
|
// get the correct stats back.
|
|
issueStats, err := issues_model.GetIssueStats(&issues_model.IssuesOptions{
|
|
RepoIDs: []int64{1},
|
|
IssueIDs: ids,
|
|
})
|
|
|
|
// Now check the values.
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, issueStats.OpenCount, issueAmount)
|
|
}
|
|
|
|
func TestMilestoneList_LoadTotalTrackedTimes(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
miles := issues_model.MilestoneList{
|
|
unittest.AssertExistsAndLoadBean(t, &issues_model.Milestone{ID: 1}),
|
|
}
|
|
|
|
assert.NoError(t, miles.LoadTotalTrackedTimes())
|
|
|
|
assert.Equal(t, int64(3682), miles[0].TotalTrackedTime)
|
|
}
|
|
|
|
func TestLoadTotalTrackedTime(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
milestone := unittest.AssertExistsAndLoadBean(t, &issues_model.Milestone{ID: 1})
|
|
|
|
assert.NoError(t, milestone.LoadTotalTrackedTime())
|
|
|
|
assert.Equal(t, int64(3682), milestone.TotalTrackedTime)
|
|
}
|
|
|
|
func TestCountIssues(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
count, err := issues_model.CountIssues(db.DefaultContext, &issues_model.IssuesOptions{})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 18, count)
|
|
}
|
|
|
|
func TestIssueLoadAttributes(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
setting.Service.EnableTimetracking = true
|
|
|
|
issueList := issues_model.IssueList{
|
|
unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 1}),
|
|
unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 4}),
|
|
}
|
|
|
|
for _, issue := range issueList {
|
|
assert.NoError(t, issue.LoadAttributes(db.DefaultContext))
|
|
assert.EqualValues(t, issue.RepoID, issue.Repo.ID)
|
|
for _, label := range issue.Labels {
|
|
assert.EqualValues(t, issue.RepoID, label.RepoID)
|
|
unittest.AssertExistsAndLoadBean(t, &issues_model.IssueLabel{IssueID: issue.ID, LabelID: label.ID})
|
|
}
|
|
if issue.PosterID > 0 {
|
|
assert.EqualValues(t, issue.PosterID, issue.Poster.ID)
|
|
}
|
|
if issue.AssigneeID > 0 {
|
|
assert.EqualValues(t, issue.AssigneeID, issue.Assignee.ID)
|
|
}
|
|
if issue.MilestoneID > 0 {
|
|
assert.EqualValues(t, issue.MilestoneID, issue.Milestone.ID)
|
|
}
|
|
if issue.IsPull {
|
|
assert.EqualValues(t, issue.ID, issue.PullRequest.IssueID)
|
|
}
|
|
for _, attachment := range issue.Attachments {
|
|
assert.EqualValues(t, issue.ID, attachment.IssueID)
|
|
}
|
|
for _, comment := range issue.Comments {
|
|
assert.EqualValues(t, issue.ID, comment.IssueID)
|
|
}
|
|
if issue.ID == int64(1) {
|
|
assert.Equal(t, int64(400), issue.TotalTrackedTime)
|
|
assert.NotNil(t, issue.Project)
|
|
assert.Equal(t, int64(1), issue.Project.ID)
|
|
} else {
|
|
assert.Nil(t, issue.Project)
|
|
}
|
|
}
|
|
}
|