f000b76879
* [FIX] Formatting and code-style - gofmt - import resort - prealloc slices - simplify code * fix vet
57 lines
881 B
Go
57 lines
881 B
Go
package storage
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetPut(t *testing.T) {
|
|
var (
|
|
s = NewMemoryStore()
|
|
key = []byte("sparse")
|
|
value = []byte("rocks")
|
|
)
|
|
|
|
if err := s.Put(key, value); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
newVal, err := s.Get(key)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, value, newVal)
|
|
}
|
|
|
|
func TestKeyNotExist(t *testing.T) {
|
|
var (
|
|
s = NewMemoryStore()
|
|
key = []byte("sparse")
|
|
)
|
|
|
|
_, err := s.Get(key)
|
|
assert.NotNil(t, err)
|
|
assert.Equal(t, err.Error(), "key not found")
|
|
}
|
|
|
|
func TestPutBatch(t *testing.T) {
|
|
var (
|
|
s = NewMemoryStore()
|
|
key = []byte("sparse")
|
|
value = []byte("rocks")
|
|
batch = s.Batch()
|
|
)
|
|
|
|
batch.Put(key, value)
|
|
|
|
if err := s.PutBatch(batch); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
newVal, err := s.Get(key)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, value, newVal)
|
|
}
|