Encryption: Improvements

pkg/services/sqlstore/data_keys_test.go

@@ -0,0 +1,44 @@
+package sqlstore
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/grafana/grafana/pkg/models"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDataKeys(t *testing.T) {
+	db := InitTestDB(t)
+	ctx := context.Background()
+
+	dataKey := models.DataKey{
+		Active:        true,
+		Name:          "Testing",
+		Provider:      "test",
+		EncryptedData: []byte{0x62, 0xAF, 0xA1, 0x1A},
+	}
+
+	res, err := db.GetDataKey(ctx, dataKey.Name)
+	assert.Equal(t, models.ErrDataKeyNotFound, err)
+	assert.Nil(t, res)
+
+	err = db.CreateDataKey(ctx, dataKey)
+	require.NoError(t, err)
+
+	res, err = db.GetDataKey(ctx, dataKey.Name)
+	require.NoError(t, err)
+	assert.Equal(t, dataKey.EncryptedData, res.EncryptedData)
+	assert.Equal(t, dataKey.Provider, res.Provider)
+	assert.Equal(t, dataKey.Name, res.Name)
+	assert.True(t, dataKey.Active)
+
+	err = db.DeleteDataKey(ctx, dataKey.Name)
+	require.NoError(t, err)
+
+	res, err = db.GetDataKey(ctx, dataKey.Name)
+	assert.Equal(t, models.ErrDataKeyNotFound, err)
+	assert.Nil(t, res)
+}