mirror of
https://github.com/grafana/grafana.git
synced 2025-02-14 17:43:35 -06:00
1c7f89c41b
* Add tests for service accounts metrics usage * Add service account store implementation * Add service account service implementation * Add tests for org metrics usage * Add org implementation * Add service implementation
36 lines
1.2 KiB
Go
36 lines
1.2 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/services/serviceaccounts"
|
|
)
|
|
|
|
func Test_UsageStats(t *testing.T) {
|
|
storeMock := newServiceAccountStoreFake()
|
|
svc := ServiceAccountsService{storeMock, log.New("test"), log.New("background-test"), &SecretsCheckerFake{}, true, 5}
|
|
err := svc.DeleteServiceAccount(context.Background(), 1, 1)
|
|
require.NoError(t, err)
|
|
|
|
storeMock.ExpectedStats = &serviceaccounts.Stats{
|
|
ServiceAccounts: 1,
|
|
ServiceAccountsWithNoRole: 1,
|
|
Tokens: 1,
|
|
ForcedExpiryEnabled: false,
|
|
}
|
|
stats, err := svc.getUsageMetrics(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
assert.Len(t, stats, 5, stats)
|
|
assert.Equal(t, int64(1), stats["stats.serviceaccounts.count"].(int64))
|
|
assert.Equal(t, int64(1), stats["stats.serviceaccounts.role_none.count"].(int64))
|
|
assert.Equal(t, int64(1), stats["stats.serviceaccounts.tokens.count"].(int64))
|
|
assert.Equal(t, int64(1), stats["stats.serviceaccounts.secret_scan.enabled.count"].(int64))
|
|
assert.Equal(t, int64(0), stats["stats.serviceaccounts.forced_expiry_enabled.count"].(int64))
|
|
}
|