148 lines
2.4 KiB
Go
148 lines
2.4 KiB
Go
package splitstore
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/ipfs/go-cid"
|
|
"github.com/multiformats/go-multihash"
|
|
)
|
|
|
|
func TestCheckpoint(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "checkpoint.*")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
_ = os.RemoveAll(dir)
|
|
})
|
|
|
|
path := filepath.Join(dir, "checkpoint")
|
|
|
|
makeCid := func(key string) cid.Cid {
|
|
h, err := multihash.Sum([]byte(key), multihash.SHA2_256, -1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return cid.NewCidV1(cid.Raw, h)
|
|
}
|
|
|
|
k1 := makeCid("a")
|
|
k2 := makeCid("b")
|
|
k3 := makeCid("c")
|
|
k4 := makeCid("d")
|
|
|
|
cp, err := NewCheckpoint(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := cp.Set(k1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := cp.Set(k2); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := cp.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cp, start, err := OpenCheckpoint(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !start.Equals(k2) {
|
|
t.Fatalf("expected start to be %s; got %s", k2, start)
|
|
}
|
|
|
|
if err := cp.Set(k3); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := cp.Set(k4); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := cp.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cp, start, err = OpenCheckpoint(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !start.Equals(k4) {
|
|
t.Fatalf("expected start to be %s; got %s", k4, start)
|
|
}
|
|
|
|
if err := cp.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// also test correct operation with an empty checkpoint
|
|
cp, err = NewCheckpoint(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := cp.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cp, start, err = OpenCheckpoint(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if start.Defined() {
|
|
t.Fatal("expected start to be undefined")
|
|
}
|
|
|
|
if err := cp.Set(k1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := cp.Set(k2); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := cp.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cp, start, err = OpenCheckpoint(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !start.Equals(k2) {
|
|
t.Fatalf("expected start to be %s; got %s", k2, start)
|
|
}
|
|
|
|
if err := cp.Set(k3); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := cp.Set(k4); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := cp.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cp, start, err = OpenCheckpoint(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !start.Equals(k4) {
|
|
t.Fatalf("expected start to be %s; got %s", k4, start)
|
|
}
|
|
|
|
if err := cp.Close(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
}
|