summaryrefslogtreecommitdiffstats
path: root/models/user_heatmap_test.go
blob: aeea427499d9c92fb9855ad66fb3a910db2779c8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
// Copyright 2018 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 models

package models

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestGetUserHeatmapDataByUser(t *testing.T) {
	// Prepare
	assert.NoError(t, PrepareTestDatabase())

	// Insert some action
	user := AssertExistsAndLoadBean(t, &User{ID: 2}).(*User)

	// get the action for comparison
	actions, err := GetFeeds(GetFeedsOptions{
		RequestedUser:    user,
		RequestingUserID: user.ID,
		IncludePrivate:   true,
		OnlyPerformedBy:  false,
		IncludeDeleted:   true,
	})
	assert.NoError(t, err)

	// Get the heatmap and compare
	heatmap, err := GetUserHeatmapDataByUser(user)
	assert.NoError(t, err)
	assert.Equal(t, len(actions), len(heatmap))
}