71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package types
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestKVStore(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
//stores to be tested
|
|
ms := NewMemKVStore()
|
|
store := NewMemKVStore()
|
|
kvc := NewKVCache(store)
|
|
|
|
//key value pairs to be tested within the system
|
|
var keyvalue = []struct {
|
|
key string
|
|
value string
|
|
}{
|
|
{"foo", "snake"},
|
|
{"bar", "mouse"},
|
|
}
|
|
|
|
//set the kvc to have all the key value pairs
|
|
setRecords := func(kv KVStore) {
|
|
for _, n := range keyvalue {
|
|
kv.Set([]byte(n.key), []byte(n.value))
|
|
}
|
|
}
|
|
|
|
//store has all the key value pairs
|
|
storeHasAll := func(kv KVStore) bool {
|
|
for _, n := range keyvalue {
|
|
if !bytes.Equal(kv.Get([]byte(n.key)), []byte(n.value)) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
//test read/write for MemKVStore
|
|
setRecords(ms)
|
|
assert.True(storeHasAll(ms), "MemKVStore doesn't retrieve after Set")
|
|
|
|
//test read/write for KVCache
|
|
setRecords(kvc)
|
|
assert.True(storeHasAll(kvc), "KVCache doesn't retrieve after Set")
|
|
|
|
//test reset
|
|
kvc.Reset()
|
|
assert.False(storeHasAll(kvc), "KVCache retrieving after reset")
|
|
|
|
//test sync
|
|
setRecords(kvc)
|
|
assert.False(storeHasAll(store), "store retrieving before synced")
|
|
kvc.Sync()
|
|
assert.True(storeHasAll(store), "store isn't retrieving after synced")
|
|
|
|
//test logging
|
|
assert.Zero(len(kvc.GetLogLines()), "logging events existed before using SetLogging")
|
|
kvc.SetLogging()
|
|
setRecords(kvc)
|
|
assert.Equal(len(kvc.GetLogLines()), 2, "incorrect number of logging events recorded")
|
|
kvc.ClearLogLines()
|
|
assert.Zero(len(kvc.GetLogLines()), "logging events still exists after ClearLogLines")
|
|
|
|
}
|