mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
39a3bb8a1c
* Alerting: Persist notification log and silences to the database This removes the dependency of having persistent disk to run grafana alerting. Instead of regularly flushing the notification log and silences to disk we now flush the binary content of those files to the database encoded as a base64 string.
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))
|
|
}
|