mirror of
https://github.com/grafana/grafana.git
synced 2025-02-16 18:34:52 -06:00
6cfb640a0b
* Add encryption service * Add tests for encryption service * Inject encryption service into http server * Replace encryption global function usage in login tests * Migrate to Wire * Move Encryption bindings to OSS Wire set * Chore: Refactor securedata to remove global encryption calls from dashboard snapshots * Fix dashboard snapshot tests * Remove no longer user test * Add dashboard snapshots service tests * Refactor service initialization * Set up dashboard snapshots service as a background service Co-authored-by: Tania B <yalyna.ts@gmail.com> Co-authored-by: Emil Tullstedt <emil.tullstedt@grafana.com>
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package dashboardsnapshots
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/encryption/ossencryption"
|
|
"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)
|
|
|
|
s := &Service{
|
|
SQLStore: sqlStore,
|
|
EncryptionService: ossencryption.ProvideService(),
|
|
}
|
|
|
|
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) {
|
|
cmd := models.CreateDashboardSnapshotCommand{
|
|
Key: dashboardKey,
|
|
DeleteKey: dashboardKey,
|
|
Dashboard: dashboard,
|
|
}
|
|
|
|
err = s.CreateDashboardSnapshot(&cmd)
|
|
require.NoError(t, err)
|
|
|
|
decrypted, err := s.EncryptionService.Decrypt(cmd.Result.DashboardEncrypted, setting.SecretKey)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, rawDashboard, decrypted)
|
|
})
|
|
|
|
t.Run("get dashboard snapshot should return the dashboard decrypted", func(t *testing.T) {
|
|
query := models.GetDashboardSnapshotQuery{
|
|
Key: dashboardKey,
|
|
DeleteKey: dashboardKey,
|
|
}
|
|
|
|
err := s.GetDashboardSnapshot(&query)
|
|
require.NoError(t, err)
|
|
|
|
decrypted, err := query.Result.Dashboard.Encode()
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, rawDashboard, decrypted)
|
|
})
|
|
}
|