Emil Tullstedt
45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
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)
|
|
}
|