77 lines
1.2 KiB
Go
77 lines
1.2 KiB
Go
|
package lsm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/fs"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/matryer/is"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEncoding(t *testing.T) {
|
||
|
|
is := is.New(t)
|
||
|
|
|
||
|
|
data := segment{entries: entries{
|
||
|
|
{"key-1", 1},
|
||
|
|
{"key-2", 2},
|
||
|
|
{"key-3", 3},
|
||
|
|
{"longerkey-4", 65535},
|
||
|
|
}}
|
||
|
|
|
||
|
|
b, err := data.MarshalBinary()
|
||
|
|
is.NoErr(err)
|
||
|
|
|
||
|
|
var got segment
|
||
|
|
err = got.UnmarshalBinary(b)
|
||
|
|
is.NoErr(err)
|
||
|
|
|
||
|
|
is.Equal(data, got)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestReverse(t *testing.T) {
|
||
|
|
is := is.New(t)
|
||
|
|
|
||
|
|
got := []byte("gnirts a si siht")
|
||
|
|
reverse(got)
|
||
|
|
|
||
|
|
is.Equal(got, []byte("this is a string"))
|
||
|
|
|
||
|
|
got = []byte("!gnirts a si siht")
|
||
|
|
reverse(got)
|
||
|
|
|
||
|
|
is.Equal(got, []byte("this is a string!"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFile(t *testing.T) {
|
||
|
|
is := is.New(t)
|
||
|
|
|
||
|
|
entries := entries {
|
||
|
|
{"key-1", 1},
|
||
|
|
{"key-2", 2},
|
||
|
|
{"key-3", 3},
|
||
|
|
{"longerkey-4", 65535},
|
||
|
|
}
|
||
|
|
|
||
|
|
f := basicFile(t, entries, entries, entries)
|
||
|
|
|
||
|
|
sf, err := ReadFile(f)
|
||
|
|
is.NoErr(err)
|
||
|
|
|
||
|
|
is.Equal(len(sf.segments), 3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func basicFile(t *testing.T, lis ...entries) fs.File {
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
segments := make([][]byte, len(lis))
|
||
|
|
var err error
|
||
|
|
for i, entries := range lis {
|
||
|
|
data := segment{entries: entries}
|
||
|
|
segments[i], err = data.MarshalBinary()
|
||
|
|
if err != nil {
|
||
|
|
t.Error(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return NewFile(segments...)
|
||
|
|
}
|