mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
05709ce411
* chore: add alias for InitTestDB and Session Adds an alias for the sqlstore InitTestDB and Session, and updates tests using these to reduce dependencies on the sqlstore.Store. * next pass of removing sqlstore imports * last little bit * remove mockstore where possible
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/services/dashboardsnapshots"
|
|
dashsnapdb "github.com/grafana/grafana/pkg/services/dashboardsnapshots/database"
|
|
"github.com/grafana/grafana/pkg/services/secrets/database"
|
|
secretsManager "github.com/grafana/grafana/pkg/services/secrets/manager"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
func TestDashboardSnapshotsService(t *testing.T) {
|
|
sqlStore := db.InitTestDB(t)
|
|
dsStore := dashsnapdb.ProvideStore(sqlStore)
|
|
secretsService := secretsManager.SetupTestService(t, database.ProvideSecretsStore(sqlStore))
|
|
s := ProvideService(dsStore, secretsService)
|
|
|
|
origSecret := setting.SecretKey
|
|
setting.SecretKey = "dashboard_snapshot_service_test"
|
|
t.Cleanup(func() {
|
|
setting.SecretKey = origSecret
|
|
})
|
|
|
|
dashboardKey := "12345"
|
|
|
|
rawDashboard := []byte(`{"id":123}`)
|
|
dashboard, err := simplejson.NewJson(rawDashboard)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("create dashboard snapshot should encrypt the dashboard", func(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
cmd := dashboardsnapshots.CreateDashboardSnapshotCommand{
|
|
Key: dashboardKey,
|
|
DeleteKey: dashboardKey,
|
|
Dashboard: dashboard,
|
|
}
|
|
|
|
err = s.CreateDashboardSnapshot(ctx, &cmd)
|
|
require.NoError(t, err)
|
|
|
|
decrypted, err := s.secretsService.Decrypt(ctx, cmd.Result.DashboardEncrypted)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, rawDashboard, decrypted)
|
|
})
|
|
|
|
t.Run("get dashboard snapshot should return the dashboard decrypted", func(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
query := dashboardsnapshots.GetDashboardSnapshotQuery{
|
|
Key: dashboardKey,
|
|
DeleteKey: dashboardKey,
|
|
}
|
|
|
|
err := s.GetDashboardSnapshot(ctx, &query)
|
|
require.NoError(t, err)
|
|
|
|
decrypted, err := query.Result.Dashboard.Encode()
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, rawDashboard, decrypted)
|
|
})
|
|
}
|