micro/store/cache/cache_test.go
Dominic Wong 5967a68e78
cached file store (#1739)
* cached file store
2020-06-26 16:13:53 +01:00

103 lines
2.3 KiB
Go

package cache
import (
"os"
"path/filepath"
"testing"
"github.com/micro/go-micro/v2/store"
"github.com/micro/go-micro/v2/store/file"
"github.com/stretchr/testify/assert"
)
func cleanup(db string, s store.Store) {
s.Close()
dir := filepath.Join(file.DefaultDir, db+"/")
os.RemoveAll(dir)
}
func TestRead(t *testing.T) {
cf := NewStore(file.NewStore())
cf.Init()
cfInt := cf.(*cache)
defer cleanup(file.DefaultDatabase, cf)
_, err := cf.Read("key1")
assert.Error(t, err, "Unexpected record")
cfInt.b.Write(&store.Record{
Key: "key1",
Value: []byte("foo"),
})
recs, err := cf.Read("key1")
assert.NoError(t, err)
assert.Len(t, recs, 1, "Expected a record to be pulled from file store")
recs, err = cfInt.m.Read("key1")
assert.NoError(t, err)
assert.Len(t, recs, 1, "Expected a memory store to be populatedfrom file store")
}
func TestWrite(t *testing.T) {
cf := NewStore(file.NewStore())
cf.Init()
cfInt := cf.(*cache)
defer cleanup(file.DefaultDatabase, cf)
cf.Write(&store.Record{
Key: "key1",
Value: []byte("foo"),
})
recs, _ := cfInt.m.Read("key1")
assert.Len(t, recs, 1, "Expected a record in the memory store")
recs, _ = cfInt.b.Read("key1")
assert.Len(t, recs, 1, "Expected a record in the file store")
}
func TestDelete(t *testing.T) {
cf := NewStore(file.NewStore())
cf.Init()
cfInt := cf.(*cache)
defer cleanup(file.DefaultDatabase, cf)
cf.Write(&store.Record{
Key: "key1",
Value: []byte("foo"),
})
recs, _ := cfInt.m.Read("key1")
assert.Len(t, recs, 1, "Expected a record in the memory store")
recs, _ = cfInt.b.Read("key1")
assert.Len(t, recs, 1, "Expected a record in the file store")
cf.Delete("key1")
_, err := cfInt.m.Read("key1")
assert.Error(t, err, "Expected no records in memory store")
_, err = cfInt.b.Read("key1")
assert.Error(t, err, "Expected no records in file store")
}
func TestList(t *testing.T) {
cf := NewStore(file.NewStore())
cf.Init()
cfInt := cf.(*cache)
defer cleanup(file.DefaultDatabase, cf)
keys, err := cf.List()
assert.NoError(t, err)
assert.Len(t, keys, 0)
cfInt.b.Write(&store.Record{
Key: "key1",
Value: []byte("foo"),
})
cfInt.b.Write(&store.Record{
Key: "key2",
Value: []byte("foo"),
})
keys, err = cf.List()
assert.NoError(t, err)
assert.Len(t, keys, 2)
}