mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-12-23 07:43:18 -05:00
044c754ea5
This PR adds a context parameter to a bunch of methods. Some helper `xxxCtx()` methods got replaced with the normal name now. Co-authored-by: delvh <dev.lh@web.de> Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
278 lines
8.7 KiB
Go
278 lines
8.7 KiB
Go
// Copyright 2020 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 activities_test
|
|
|
|
import (
|
|
"path"
|
|
"testing"
|
|
|
|
activities_model "code.gitea.io/gitea/models/activities"
|
|
"code.gitea.io/gitea/models/db"
|
|
issue_model "code.gitea.io/gitea/models/issues"
|
|
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"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAction_GetRepoPath(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
|
|
owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID})
|
|
action := &activities_model.Action{RepoID: repo.ID}
|
|
assert.Equal(t, path.Join(owner.Name, repo.Name), action.GetRepoPath())
|
|
}
|
|
|
|
func TestAction_GetRepoLink(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1})
|
|
owner := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: repo.OwnerID})
|
|
comment := unittest.AssertExistsAndLoadBean(t, &issue_model.Comment{ID: 2})
|
|
action := &activities_model.Action{RepoID: repo.ID, CommentID: comment.ID}
|
|
setting.AppSubURL = "/suburl"
|
|
expected := path.Join(setting.AppSubURL, owner.Name, repo.Name)
|
|
assert.Equal(t, expected, action.GetRepoLink())
|
|
assert.Equal(t, repo.HTMLURL(), action.GetRepoAbsoluteLink())
|
|
assert.Equal(t, comment.HTMLURL(), action.GetCommentLink())
|
|
}
|
|
|
|
func TestGetFeeds(t *testing.T) {
|
|
// test with an individual user
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
|
|
actions, err := activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedUser: user,
|
|
Actor: user,
|
|
IncludePrivate: true,
|
|
OnlyPerformedBy: false,
|
|
IncludeDeleted: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, actions, 1) {
|
|
assert.EqualValues(t, 1, actions[0].ID)
|
|
assert.EqualValues(t, user.ID, actions[0].UserID)
|
|
}
|
|
|
|
actions, err = activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedUser: user,
|
|
Actor: user,
|
|
IncludePrivate: false,
|
|
OnlyPerformedBy: false,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 0)
|
|
}
|
|
|
|
func TestGetFeedsForRepos(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
privRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 2})
|
|
pubRepo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 8})
|
|
|
|
// private repo & no login
|
|
actions, err := activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedRepo: privRepo,
|
|
IncludePrivate: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 0)
|
|
|
|
// public repo & no login
|
|
actions, err = activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedRepo: pubRepo,
|
|
IncludePrivate: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 1)
|
|
|
|
// private repo and login
|
|
actions, err = activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedRepo: privRepo,
|
|
IncludePrivate: true,
|
|
Actor: user,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 1)
|
|
|
|
// public repo & login
|
|
actions, err = activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedRepo: pubRepo,
|
|
IncludePrivate: true,
|
|
Actor: user,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 1)
|
|
}
|
|
|
|
func TestGetFeeds2(t *testing.T) {
|
|
// test with an organization user
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
org := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 3})
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
|
|
|
|
actions, err := activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedUser: org,
|
|
Actor: user,
|
|
IncludePrivate: true,
|
|
OnlyPerformedBy: false,
|
|
IncludeDeleted: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 1)
|
|
if assert.Len(t, actions, 1) {
|
|
assert.EqualValues(t, 2, actions[0].ID)
|
|
assert.EqualValues(t, org.ID, actions[0].UserID)
|
|
}
|
|
|
|
actions, err = activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedUser: org,
|
|
Actor: user,
|
|
IncludePrivate: false,
|
|
OnlyPerformedBy: false,
|
|
IncludeDeleted: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 0)
|
|
}
|
|
|
|
func TestActivityReadable(t *testing.T) {
|
|
tt := []struct {
|
|
desc string
|
|
user *user_model.User
|
|
doer *user_model.User
|
|
result bool
|
|
}{{
|
|
desc: "user should see own activity",
|
|
user: &user_model.User{ID: 1},
|
|
doer: &user_model.User{ID: 1},
|
|
result: true,
|
|
}, {
|
|
desc: "anon should see activity if public",
|
|
user: &user_model.User{ID: 1},
|
|
result: true,
|
|
}, {
|
|
desc: "anon should NOT see activity",
|
|
user: &user_model.User{ID: 1, KeepActivityPrivate: true},
|
|
result: false,
|
|
}, {
|
|
desc: "user should see own activity if private too",
|
|
user: &user_model.User{ID: 1, KeepActivityPrivate: true},
|
|
doer: &user_model.User{ID: 1},
|
|
result: true,
|
|
}, {
|
|
desc: "other user should NOT see activity",
|
|
user: &user_model.User{ID: 1, KeepActivityPrivate: true},
|
|
doer: &user_model.User{ID: 2},
|
|
result: false,
|
|
}, {
|
|
desc: "admin should see activity",
|
|
user: &user_model.User{ID: 1, KeepActivityPrivate: true},
|
|
doer: &user_model.User{ID: 2, IsAdmin: true},
|
|
result: true,
|
|
}}
|
|
for _, test := range tt {
|
|
assert.Equal(t, test.result, activities_model.ActivityReadable(test.user, test.doer), test.desc)
|
|
}
|
|
}
|
|
|
|
func TestNotifyWatchers(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
action := &activities_model.Action{
|
|
ActUserID: 8,
|
|
RepoID: 1,
|
|
OpType: activities_model.ActionStarRepo,
|
|
}
|
|
assert.NoError(t, activities_model.NotifyWatchers(db.DefaultContext, action))
|
|
|
|
// One watchers are inactive, thus action is only created for user 8, 1, 4, 11
|
|
unittest.AssertExistsAndLoadBean(t, &activities_model.Action{
|
|
ActUserID: action.ActUserID,
|
|
UserID: 8,
|
|
RepoID: action.RepoID,
|
|
OpType: action.OpType,
|
|
})
|
|
unittest.AssertExistsAndLoadBean(t, &activities_model.Action{
|
|
ActUserID: action.ActUserID,
|
|
UserID: 1,
|
|
RepoID: action.RepoID,
|
|
OpType: action.OpType,
|
|
})
|
|
unittest.AssertExistsAndLoadBean(t, &activities_model.Action{
|
|
ActUserID: action.ActUserID,
|
|
UserID: 4,
|
|
RepoID: action.RepoID,
|
|
OpType: action.OpType,
|
|
})
|
|
unittest.AssertExistsAndLoadBean(t, &activities_model.Action{
|
|
ActUserID: action.ActUserID,
|
|
UserID: 11,
|
|
RepoID: action.RepoID,
|
|
OpType: action.OpType,
|
|
})
|
|
}
|
|
|
|
func TestGetFeedsCorrupted(t *testing.T) {
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 1})
|
|
unittest.AssertExistsAndLoadBean(t, &activities_model.Action{
|
|
ID: 8,
|
|
RepoID: 1700,
|
|
})
|
|
|
|
actions, err := activities_model.GetFeeds(db.DefaultContext, activities_model.GetFeedsOptions{
|
|
RequestedUser: user,
|
|
Actor: user,
|
|
IncludePrivate: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, 0)
|
|
}
|
|
|
|
func TestConsistencyUpdateAction(t *testing.T) {
|
|
if !setting.Database.UseSQLite3 {
|
|
t.Skip("Test is only for SQLite database.")
|
|
}
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
id := 8
|
|
unittest.AssertExistsAndLoadBean(t, &activities_model.Action{
|
|
ID: int64(id),
|
|
})
|
|
_, err := db.GetEngine(db.DefaultContext).Exec(`UPDATE action SET created_unix = "" WHERE id = ?`, id)
|
|
assert.NoError(t, err)
|
|
actions := make([]*activities_model.Action, 0, 1)
|
|
//
|
|
// XORM returns an error when created_unix is a string
|
|
//
|
|
err = db.GetEngine(db.DefaultContext).Where("id = ?", id).Find(&actions)
|
|
if assert.Error(t, err) {
|
|
assert.Contains(t, err.Error(), "type string to a int64: invalid syntax")
|
|
}
|
|
//
|
|
// Get rid of incorrectly set created_unix
|
|
//
|
|
count, err := activities_model.CountActionCreatedUnixString(db.DefaultContext)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, count)
|
|
count, err = activities_model.FixActionCreatedUnixString(db.DefaultContext)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, count)
|
|
|
|
count, err = activities_model.CountActionCreatedUnixString(db.DefaultContext)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 0, count)
|
|
count, err = activities_model.FixActionCreatedUnixString(db.DefaultContext)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 0, count)
|
|
|
|
//
|
|
// XORM must be happy now
|
|
//
|
|
assert.NoError(t, db.GetEngine(db.DefaultContext).Where("id = ?", id).Find(&actions))
|
|
unittest.CheckConsistencyFor(t, &activities_model.Action{})
|
|
}
|