mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
faa1244518
* replace sqlstore with db interface in a few packages * remove from stats * remove sqlstore in admin test * remove sqlstore from api plugin tests * fix another createUser * remove sqlstore in publicdashboards * remove sqlstore from orgs * clean up orguser test * more clean up in sso * clean up service accounts * further cleanup * more cleanup in accesscontrol * last cleanup in accesscontrol * clean up teams * more removals * split cfg from db in testenv * few remaining fixes * fix test with bus * pass cfg for testing inside db as an option * set query retries when no opts provided * revert golden test data * rebase and rollback
52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/components/satokengen"
|
|
"github.com/grafana/grafana/pkg/services/org"
|
|
"github.com/grafana/grafana/pkg/services/serviceaccounts"
|
|
"github.com/grafana/grafana/pkg/services/serviceaccounts/tests"
|
|
)
|
|
|
|
func TestIntegrationStore_UsageStats(t *testing.T) {
|
|
saToCreate := tests.TestUser{Login: "servicetestwithTeam@admin", IsServiceAccount: true}
|
|
db, store := setupTestDatabase(t)
|
|
sa := tests.SetupUserServiceAccount(t, db, store.cfg, saToCreate)
|
|
|
|
store.cfg.SATokenExpirationDayLimit = 4
|
|
|
|
keyName := t.Name()
|
|
key, err := satokengen.New(keyName)
|
|
require.NoError(t, err)
|
|
|
|
cmd := serviceaccounts.AddServiceAccountTokenCommand{
|
|
Name: keyName,
|
|
OrgId: sa.OrgID,
|
|
Key: key.HashedKey,
|
|
SecondsToLive: 0,
|
|
}
|
|
|
|
_, err = store.AddServiceAccountToken(context.Background(), sa.ID, &cmd)
|
|
require.NoError(t, err)
|
|
|
|
role := org.RoleNone
|
|
form := serviceaccounts.UpdateServiceAccountForm{
|
|
Role: &role,
|
|
}
|
|
_, err = store.UpdateServiceAccount(context.Background(), sa.OrgID, sa.ID, &form)
|
|
require.NoError(t, err)
|
|
|
|
stats, err := store.GetUsageMetrics(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, int64(1), stats.ServiceAccounts)
|
|
assert.Equal(t, int64(1), stats.ServiceAccountsWithNoRole)
|
|
assert.Equal(t, int64(1), stats.Tokens)
|
|
assert.Equal(t, true, stats.ForcedExpiryEnabled)
|
|
}
|