63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
package storage
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
)
|
|
|
|
type mock struct {
|
|
m map[string][]byte
|
|
}
|
|
|
|
func (mock *mock) Get(key string) ([]byte, error) {
|
|
v, ok := mock.m[key]
|
|
if ok {
|
|
return v, nil
|
|
}
|
|
return nil, ErrNotFound
|
|
}
|
|
|
|
func (mock *mock) Set(key string, value []byte) error {
|
|
mock.m[key] = value
|
|
return nil
|
|
}
|
|
|
|
func (mock *mock) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func TestImplementations(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "storage_tests_")
|
|
if err != nil {
|
|
t.Fatalf("cannot create temp dir: %v", err)
|
|
return
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
bolt, _ := NewBolt(path.Join(dir, "bolt"))
|
|
leveldb, _ := NewLevelDB(path.Join(dir, "leveldb"))
|
|
cases := []DB{
|
|
&mock{m: make(map[string][]byte)},
|
|
NewMap(),
|
|
bolt,
|
|
leveldb,
|
|
}
|
|
|
|
validKey := "key"
|
|
validValue := []byte("value")
|
|
|
|
for _, db := range cases {
|
|
if err := db.Set(validKey, validValue); err != nil {
|
|
t.Fatalf("cannot set %T: %v", db, err)
|
|
}
|
|
if v, err := db.Get(validKey); err != nil {
|
|
t.Fatalf("cannot get %T: %v", db, err)
|
|
} else if !bytes.Equal(v, validValue) {
|
|
t.Fatalf("wrong get %T: %q vs %q", db, v, validValue)
|
|
}
|
|
}
|
|
}
|