mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package activities_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
activities_model "github.com/hanzoai/git/models/activities"
|
|
"github.com/hanzoai/git/models/unittest"
|
|
user_model "github.com/hanzoai/git/models/user"
|
|
"github.com/hanzoai/git/modules/json"
|
|
"github.com/hanzoai/git/modules/timeutil"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetUserHeatmapDataByUser(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
userID int64
|
|
doerID int64
|
|
CountResult int
|
|
JSONResult string
|
|
}{
|
|
{
|
|
"self looks at action in private repo",
|
|
2, 2, 1, `[{"timestamp":1603227600,"contributions":1}]`,
|
|
},
|
|
{
|
|
"admin looks at action in private repo",
|
|
2, 1, 1, `[{"timestamp":1603227600,"contributions":1}]`,
|
|
},
|
|
{
|
|
"other user looks at action in private repo",
|
|
2, 3, 0, `[]`,
|
|
},
|
|
{
|
|
"nobody looks at action in private repo",
|
|
2, 0, 0, `[]`,
|
|
},
|
|
{
|
|
"collaborator looks at action in private repo",
|
|
16, 15, 1, `[{"timestamp":1603267200,"contributions":1}]`,
|
|
},
|
|
{
|
|
"no action action not performed by target user",
|
|
3, 3, 0, `[]`,
|
|
},
|
|
{
|
|
"multiple actions performed with two grouped together",
|
|
10, 10, 3, `[{"timestamp":1603009800,"contributions":1},{"timestamp":1603010700,"contributions":2}]`,
|
|
},
|
|
}
|
|
// Prepare
|
|
assert.NoError(t, unittest.PrepareTestDatabase())
|
|
|
|
// Mock time
|
|
timeutil.MockSet(time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC))
|
|
defer timeutil.MockUnset()
|
|
|
|
for _, tc := range testCases {
|
|
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: tc.userID})
|
|
|
|
var doer *user_model.User
|
|
if tc.doerID != 0 {
|
|
doer = unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: tc.doerID})
|
|
}
|
|
|
|
// get the action for comparison
|
|
actions, count, err := activities_model.GetFeeds(t.Context(), activities_model.GetFeedsOptions{
|
|
RequestedUser: user,
|
|
Actor: doer,
|
|
IncludePrivate: true,
|
|
OnlyPerformedBy: true,
|
|
IncludeDeleted: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
// Get the heatmap and compare
|
|
heatmap, err := activities_model.GetUserHeatmapDataByUser(t.Context(), user, doer)
|
|
var contributions int
|
|
for _, hm := range heatmap {
|
|
contributions += int(hm.Contributions)
|
|
}
|
|
assert.NoError(t, err)
|
|
assert.Len(t, actions, contributions, "invalid action count: did the test data became too old?")
|
|
assert.Equal(t, count, int64(contributions))
|
|
assert.Equal(t, tc.CountResult, contributions, "testcase '%s'", tc.desc)
|
|
|
|
// Test JSON rendering
|
|
jsonData, err := json.Marshal(heatmap)
|
|
assert.NoError(t, err)
|
|
assert.JSONEq(t, tc.JSONResult, string(jsonData))
|
|
}
|
|
}
|