mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
7d8ec6199d
* Add config to remove Snapshot functionality (frontend is hidden and validation in the backend) * Add test cases * Remove unused mock on the test * Moving Snapshot config from globar variables to settings.Cfg * Removing warnings on code
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, setting.NewCfg())
|
|
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,
|
|
}
|
|
|
|
result, err := s.CreateDashboardSnapshot(ctx, &cmd)
|
|
require.NoError(t, err)
|
|
|
|
decrypted, err := s.secretsService.Decrypt(ctx, 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,
|
|
}
|
|
|
|
queryResult, err := s.GetDashboardSnapshot(ctx, &query)
|
|
require.NoError(t, err)
|
|
|
|
decrypted, err := queryResult.Dashboard.Encode()
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, rawDashboard, decrypted)
|
|
})
|
|
}
|