You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.8 KiB
72 lines
1.8 KiB
package dashboardsnapshots
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/services/secrets/database"
|
|
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
secretsManager "github.com/grafana/grafana/pkg/services/secrets/manager"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDashboardSnapshotsService(t *testing.T) {
|
|
sqlStore := sqlstore.InitTestDB(t)
|
|
secretsService := secretsManager.SetupTestService(t, database.ProvideSecretsStore(sqlStore))
|
|
|
|
s := &Service{
|
|
SQLStore: sqlStore,
|
|
SecretsService: 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 := models.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 := models.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)
|
|
})
|
|
}
|
|
|