mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
common "github.com/grafana/grafana/pkg/apimachinery/apis/common/v0alpha1"
|
|
dashboardsnapshot "github.com/grafana/grafana/pkg/apis/dashboardsnapshot/v0alpha1"
|
|
"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"
|
|
"github.com/grafana/grafana/pkg/tests/testsuite"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
testsuite.Run(m)
|
|
}
|
|
|
|
func TestDashboardSnapshotsService(t *testing.T) {
|
|
sqlStore := db.InitTestDB(t)
|
|
cfg := setting.NewCfg()
|
|
dsStore := dashsnapdb.ProvideStore(sqlStore, cfg)
|
|
secretsService := secretsManager.SetupTestService(t, database.ProvideSecretsStore(sqlStore))
|
|
s := ProvideService(dsStore, secretsService)
|
|
|
|
origSecret := cfg.SecretKey
|
|
cfg.SecretKey = "dashboard_snapshot_service_test"
|
|
t.Cleanup(func() {
|
|
cfg.SecretKey = origSecret
|
|
})
|
|
|
|
dashboardKey := "12345"
|
|
|
|
dashboard := &common.Unstructured{}
|
|
rawDashboard := []byte(`{"id":123}`)
|
|
err := json.Unmarshal(rawDashboard, dashboard)
|
|
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,
|
|
DashboardCreateCommand: dashboardsnapshot.DashboardCreateCommand{
|
|
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)
|
|
})
|
|
}
|