mirror of
https://github.com/grafana/grafana.git
synced 2024-11-25 18:30:41 -06:00
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
|
package notifier
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFileStore_FilepathFor(t *testing.T) {
|
||
|
|
store := newFakeKVStore(t)
|
||
|
|
workingDir := t.TempDir()
|
||
|
|
fs := NewFileStore(1, store, workingDir)
|
||
|
|
filekey := "silences"
|
||
|
|
filePath := filepath.Join(workingDir, filekey)
|
||
|
|
|
||
|
|
// With a file already on disk, it returns the existing file's filepath and no modification to the original file.
|
||
|
|
{
|
||
|
|
require.NoError(t, os.WriteFile(filePath, []byte("silence1,silence2"), 0644))
|
||
|
|
r, err := fs.FilepathFor(context.Background(), filekey)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, filePath, r)
|
||
|
|
f, err := ioutil.ReadFile(filepath.Clean(filePath))
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "silence1,silence2", string(f))
|
||
|
|
require.NoError(t, os.Remove(filePath))
|
||
|
|
}
|
||
|
|
|
||
|
|
// With a file already on the database, it writes the file to disk and returns the filepath.
|
||
|
|
{
|
||
|
|
require.NoError(t, store.Set(context.Background(), 1, KVNamespace, filekey, encode([]byte("silence1,silence3"))))
|
||
|
|
r, err := fs.FilepathFor(context.Background(), filekey)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, filePath, r)
|
||
|
|
f, err := ioutil.ReadFile(filepath.Clean(filePath))
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "silence1,silence3", string(f))
|
||
|
|
require.NoError(t, os.Remove(filePath))
|
||
|
|
require.NoError(t, store.Del(context.Background(), 1, KVNamespace, filekey))
|
||
|
|
}
|
||
|
|
|
||
|
|
// With no file on disk or database, it returns the original filepath.
|
||
|
|
{
|
||
|
|
r, err := fs.FilepathFor(context.Background(), filekey)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, filePath, r)
|
||
|
|
_, err = ioutil.ReadFile(filepath.Clean(filePath))
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFileStore_Persist(t *testing.T) {
|
||
|
|
store := newFakeKVStore(t)
|
||
|
|
state := &fakeState{data: "something to marshal"}
|
||
|
|
workingDir := t.TempDir()
|
||
|
|
fs := NewFileStore(1, store, workingDir)
|
||
|
|
filekey := "silences"
|
||
|
|
|
||
|
|
size, err := fs.Persist(context.Background(), filekey, state)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, int64(20), size)
|
||
|
|
store.mtx.Lock()
|
||
|
|
require.Len(t, store.store, 1)
|
||
|
|
store.mtx.Unlock()
|
||
|
|
v, ok, err := store.Get(context.Background(), 1, KVNamespace, filekey)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, ok)
|
||
|
|
b, err := decode(v)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "something to marshal", string(b))
|
||
|
|
}
|