mirror of
https://github.com/grafana/grafana.git
synced 2025-02-15 18:13:32 -06:00
2ce5dcfb91
* add secret scan service * add token batching * Apply suggestions from code review Co-authored-by: Victor Cinaglia <victor@grafana.com> * fix: finish constant renaming Co-authored-by: Victor Cinaglia <victor@grafana.com>
30 lines
985 B
Go
30 lines
985 B
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/services/serviceaccounts"
|
|
"github.com/grafana/grafana/pkg/services/serviceaccounts/tests"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_UsageStats(t *testing.T) {
|
|
storeMock := &tests.ServiceAccountsStoreMock{Calls: tests.Calls{}, Stats: &serviceaccounts.Stats{
|
|
ServiceAccounts: 1,
|
|
Tokens: 1,
|
|
}}
|
|
svc := ServiceAccountsService{store: storeMock, secretScanEnabled: true}
|
|
err := svc.DeleteServiceAccount(context.Background(), 1, 1)
|
|
require.NoError(t, err)
|
|
assert.Len(t, storeMock.Calls.DeleteServiceAccount, 1)
|
|
|
|
stats, err := svc.getUsageMetrics(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, int64(1), stats["stats.serviceaccounts.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))
|
|
}
|