mirror of
https://github.com/grafana/grafana.git
synced 2025-02-16 18:34:52 -06:00
4127db3d3a
* Remove Licensing as dependency Co-authored-by: Emil Tullstedt <emil.tullstedt@grafana.com>
105 lines
2.3 KiB
Go
105 lines
2.3 KiB
Go
package usagestats
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUsageStatsService_GetConcurrentUsersStats(t *testing.T) {
|
|
sqlStore := sqlstore.InitTestDB(t)
|
|
uss := &UsageStatsService{
|
|
Bus: bus.New(),
|
|
SQLStore: sqlStore,
|
|
}
|
|
|
|
createConcurrentTokens(t, sqlStore)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
t.Cleanup(func() {
|
|
cancel()
|
|
})
|
|
|
|
actualResult, err := uss.GetConcurrentUsersStats(ctx)
|
|
require.NoError(t, err)
|
|
|
|
expectedCachedResult := &concurrentUsersStats{
|
|
BucketLE3: 1,
|
|
BucketLE6: 2,
|
|
BucketLE9: 3,
|
|
BucketLE12: 4,
|
|
BucketLE15: 5,
|
|
BucketLEInf: 6,
|
|
}
|
|
assert.Equal(t, expectedCachedResult, actualResult)
|
|
|
|
createToken(t, 8, sqlStore)
|
|
require.NoError(t, err)
|
|
|
|
actualResult, err = uss.GetConcurrentUsersStats(ctx)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, expectedCachedResult, actualResult)
|
|
}
|
|
|
|
func createToken(t *testing.T, uID int, sqlStore *sqlstore.SQLStore) {
|
|
t.Helper()
|
|
token, err := util.RandomHex(16)
|
|
require.NoError(t, err)
|
|
|
|
tokenWithSecret := fmt.Sprintf("%ssecret%d", token, uID)
|
|
hashBytes := sha256.Sum256([]byte(tokenWithSecret))
|
|
hashedToken := hex.EncodeToString(hashBytes[:])
|
|
|
|
now := time.Now().Unix()
|
|
|
|
userAuthToken := userAuthToken{
|
|
UserID: int64(uID),
|
|
AuthToken: hashedToken,
|
|
PrevAuthToken: hashedToken,
|
|
ClientIP: "192.168.10.11",
|
|
UserAgent: "Mozilla",
|
|
RotatedAt: now,
|
|
CreatedAt: now,
|
|
UpdatedAt: now,
|
|
SeenAt: 0,
|
|
AuthTokenSeen: false,
|
|
}
|
|
|
|
err = sqlStore.WithDbSession(context.Background(), func(dbSession *sqlstore.DBSession) error {
|
|
_, err = dbSession.Insert(&userAuthToken)
|
|
return err
|
|
})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func createConcurrentTokens(t *testing.T, sqlStore *sqlstore.SQLStore) {
|
|
t.Helper()
|
|
for u := 1; u <= 6; u++ {
|
|
for tkn := 1; tkn <= u*3; tkn++ {
|
|
createToken(t, u, sqlStore)
|
|
}
|
|
}
|
|
}
|
|
|
|
type userAuthToken struct {
|
|
UserID int64 `xorm:"user_id"`
|
|
AuthToken string
|
|
PrevAuthToken string
|
|
UserAgent string
|
|
ClientIP string `xorm:"client_ip"`
|
|
AuthTokenSeen bool
|
|
SeenAt int64
|
|
RotatedAt int64
|
|
CreatedAt int64
|
|
UpdatedAt int64
|
|
}
|