diff options
author | Norwin <noerw@users.noreply.github.com> | 2020-12-22 02:53:37 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-12-22 03:53:37 +0100 |
commit | f6bec8529697bdb89ebcd0901ba093f06aa9ac46 (patch) | |
tree | d742367805a8296411211d5157c505f252f24dbf /models/user_heatmap_test.go | |
parent | 2c9dd71140474b1f83f068bece40b25e94240ab4 (diff) | |
download | gitea-f6bec8529697bdb89ebcd0901ba093f06aa9ac46.tar.gz gitea-f6bec8529697bdb89ebcd0901ba093f06aa9ac46.zip |
rework heatmap permissions (#14080)
* now uses the same permission model as for the activity feed:
only include activities in repos, that the doer has access to.
this might be somewhat slower.
* also improves handling of user.KeepActivityPrivate (still shows
the heatmap to self & admins)
* extend tests
* adjust integration test to new behaviour
* add access to actions for admins
* extend heatmap unit tests
Diffstat (limited to 'models/user_heatmap_test.go')
-rw-r--r-- | models/user_heatmap_test.go | 29 |
1 files changed, 20 insertions, 9 deletions
diff --git a/models/user_heatmap_test.go b/models/user_heatmap_test.go index c9d33db29b..d98c4c63e4 100644 --- a/models/user_heatmap_test.go +++ b/models/user_heatmap_test.go @@ -6,6 +6,7 @@ package models import ( "encoding/json" + "fmt" "testing" "github.com/stretchr/testify/assert" @@ -14,35 +15,45 @@ import ( func TestGetUserHeatmapDataByUser(t *testing.T) { testCases := []struct { userID int64 + doerID int64 CountResult int JSONResult string }{ - {2, 1, `[{"timestamp":1603152000,"contributions":1}]`}, - {3, 0, `[]`}, + {2, 2, 1, `[{"timestamp":1603152000,"contributions":1}]`}, // self looks at action in private repo + {2, 1, 1, `[{"timestamp":1603152000,"contributions":1}]`}, // admin looks at action in private repo + {2, 3, 0, `[]`}, // other user looks at action in private repo + {2, 0, 0, `[]`}, // nobody looks at action in private repo + {16, 15, 1, `[{"timestamp":1603238400,"contributions":1}]`}, // collaborator looks at action in private repo + {3, 3, 0, `[]`}, // no action action not performed by target user } // Prepare assert.NoError(t, PrepareTestDatabase()) - for _, tc := range testCases { - - // Insert some action + for i, tc := range testCases { user := AssertExistsAndLoadBean(t, &User{ID: tc.userID}).(*User) + doer := &User{ID: tc.doerID} + _, err := loadBeanIfExists(doer) + assert.NoError(t, err) + if tc.doerID == 0 { + doer = nil + } + // get the action for comparison actions, err := GetFeeds(GetFeedsOptions{ RequestedUser: user, - Actor: user, + Actor: doer, IncludePrivate: true, - OnlyPerformedBy: false, + OnlyPerformedBy: true, IncludeDeleted: true, }) assert.NoError(t, err) // Get the heatmap and compare - heatmap, err := GetUserHeatmapDataByUser(user) + heatmap, err := GetUserHeatmapDataByUser(user, doer) assert.NoError(t, err) assert.Equal(t, len(actions), len(heatmap), "invalid action count: did the test data became too old?") - assert.Equal(t, tc.CountResult, len(heatmap)) + assert.Equal(t, tc.CountResult, len(heatmap), fmt.Sprintf("testcase %d", i)) //Test JSON rendering jsonData, err := json.Marshal(heatmap) |