123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277 |
- // Copyright 2020 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- 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{})
- }
|