da50be7360
Currently 'userxx' and 'orgxx' are both used as username in test files when the user type is org, which is confusing. This PR replaces all 'userxx' with 'orgxx' when the user type is org(`user.type==1`). Some non-trivial changes 1. Rename `user3` dir to `org3` in `tests/git-repositories-meta` 2. Change `end` in `issue reference` because 'org3' is one char shorter than 'user3'  3. Change the search result number of `user/repo2` because `user3/repo21` can't be searched now  4. Change the first org name getting from API because the result is ordered by alphabet asc and now `org 17` is before `org25`   Other modifications are just find all and replace all. Unit tests with SQLite are all passed. --------- Co-authored-by: caicandong <1290147055@qq.com>
118 lines
3.7 KiB
Go
118 lines
3.7 KiB
Go
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package issues_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
issues_model "code.gitea.io/gitea/models/issues"
|
|
"code.gitea.io/gitea/models/unittest"
|
|
user_model "code.gitea.io/gitea/models/user"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAddTime(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
org3, err := user_model.GetUserByID(db.DefaultContext, 3)
|
|
assert.NoError(t, err)
|
|
|
|
issue1, err := issues_model.GetIssueByID(db.DefaultContext, 1)
|
|
assert.NoError(t, err)
|
|
|
|
// 3661 = 1h 1min 1s
|
|
trackedTime, err := issues_model.AddTime(db.DefaultContext, org3, issue1, 3661, time.Now())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(3), trackedTime.UserID)
|
|
assert.Equal(t, int64(1), trackedTime.IssueID)
|
|
assert.Equal(t, int64(3661), trackedTime.Time)
|
|
|
|
tt := unittest.AssertExistsAndLoadBean(t, &issues_model.TrackedTime{UserID: 3, IssueID: 1})
|
|
assert.Equal(t, int64(3661), tt.Time)
|
|
|
|
comment := unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{Type: issues_model.CommentTypeAddTimeManual, PosterID: 3, IssueID: 1})
|
|
assert.Equal(t, "|3661", comment.Content)
|
|
}
|
|
|
|
func TestGetTrackedTimes(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
// by Issue
|
|
times, err := issues_model.GetTrackedTimes(db.DefaultContext, &issues_model.FindTrackedTimesOptions{IssueID: 1})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, times, 1)
|
|
assert.Equal(t, int64(400), times[0].Time)
|
|
|
|
times, err = issues_model.GetTrackedTimes(db.DefaultContext, &issues_model.FindTrackedTimesOptions{IssueID: -1})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, times, 0)
|
|
|
|
// by User
|
|
times, err = issues_model.GetTrackedTimes(db.DefaultContext, &issues_model.FindTrackedTimesOptions{UserID: 1})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, times, 3)
|
|
assert.Equal(t, int64(400), times[0].Time)
|
|
|
|
times, err = issues_model.GetTrackedTimes(db.DefaultContext, &issues_model.FindTrackedTimesOptions{UserID: 3})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, times, 0)
|
|
|
|
// by Repo
|
|
times, err = issues_model.GetTrackedTimes(db.DefaultContext, &issues_model.FindTrackedTimesOptions{RepositoryID: 2})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, times, 3)
|
|
assert.Equal(t, int64(1), times[0].Time)
|
|
issue, err := issues_model.GetIssueByID(db.DefaultContext, times[0].IssueID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, issue.RepoID, int64(2))
|
|
|
|
times, err = issues_model.GetTrackedTimes(db.DefaultContext, &issues_model.FindTrackedTimesOptions{RepositoryID: 1})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, times, 5)
|
|
|
|
times, err = issues_model.GetTrackedTimes(db.DefaultContext, &issues_model.FindTrackedTimesOptions{RepositoryID: 10})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, times, 0)
|
|
}
|
|
|
|
func TestTotalTimes(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
total, err := issues_model.TotalTimes(&issues_model.FindTrackedTimesOptions{IssueID: 1})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, total, 1)
|
|
for user, time := range total {
|
|
assert.EqualValues(t, 1, user.ID)
|
|
assert.EqualValues(t, 400, time)
|
|
}
|
|
|
|
total, err = issues_model.TotalTimes(&issues_model.FindTrackedTimesOptions{IssueID: 2})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, total, 2)
|
|
for user, time := range total {
|
|
if user.ID == 2 {
|
|
assert.EqualValues(t, 3662, time)
|
|
} else if user.ID == 1 {
|
|
assert.EqualValues(t, 20, time)
|
|
} else {
|
|
assert.Error(t, assert.AnError)
|
|
}
|
|
}
|
|
|
|
total, err = issues_model.TotalTimes(&issues_model.FindTrackedTimesOptions{IssueID: 5})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, total, 1)
|
|
for user, time := range total {
|
|
assert.EqualValues(t, 2, user.ID)
|
|
assert.EqualValues(t, 1, time)
|
|
}
|
|
|
|
total, err = issues_model.TotalTimes(&issues_model.FindTrackedTimesOptions{IssueID: 4})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, total, 2)
|
|
}
|