b7a4dbb07f
And use the new CidBuilder from the spec actors. This patch does not switch over to inline CIDs by default, but paves the way.
81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
package store_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/filecoin-project/lotus/chain/gen"
|
|
"github.com/filecoin-project/lotus/chain/store"
|
|
"github.com/filecoin-project/lotus/chain/types/mock"
|
|
"github.com/filecoin-project/lotus/lib/blockstore"
|
|
"github.com/filecoin-project/specs-actors/actors/abi"
|
|
datastore "github.com/ipfs/go-datastore"
|
|
syncds "github.com/ipfs/go-datastore/sync"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestIndexSeeks(t *testing.T) {
|
|
cg, err := gen.NewGenerator()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
gencar, err := cg.GenesisCar()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
gen := cg.Genesis()
|
|
|
|
ctx := context.TODO()
|
|
|
|
nbs := blockstore.NewTemporarySync()
|
|
cs := store.NewChainStore(nbs, syncds.MutexWrap(datastore.NewMapDatastore()), nil)
|
|
|
|
_, err = cs.Import(bytes.NewReader(gencar))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cur := mock.TipSet(gen)
|
|
if err := cs.PutTipSet(ctx, mock.TipSet(gen)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
cs.SetGenesis(gen)
|
|
|
|
// Put 113 blocks from genesis
|
|
for i := 0; i < 113; i++ {
|
|
nextts := mock.TipSet(mock.MkBlock(cur, 1, 1))
|
|
|
|
if err := cs.PutTipSet(ctx, nextts); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
cur = nextts
|
|
}
|
|
|
|
// Put 50 null epochs + 1 block
|
|
skip := mock.MkBlock(cur, 1, 1)
|
|
skip.Height += 50
|
|
|
|
skipts := mock.TipSet(skip)
|
|
|
|
if err := cs.PutTipSet(ctx, skipts); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
ts, err := cs.GetTipsetByHeight(ctx, skip.Height-10, skipts, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, abi.ChainEpoch(164), ts.Height())
|
|
|
|
for i := 0; i <= 113; i++ {
|
|
ts3, err := cs.GetTipsetByHeight(ctx, abi.ChainEpoch(i), skipts, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, abi.ChainEpoch(i), ts3.Height())
|
|
}
|
|
}
|