90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
// stm: #unit
|
|
package store_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/ipfs/go-datastore"
|
|
syncds "github.com/ipfs/go-datastore/sync"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
"github.com/filecoin-project/lotus/blockstore"
|
|
"github.com/filecoin-project/lotus/chain/consensus/filcns"
|
|
"github.com/filecoin-project/lotus/chain/gen"
|
|
"github.com/filecoin-project/lotus/chain/store"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/types/mock"
|
|
)
|
|
|
|
func TestIndexSeeks(t *testing.T) {
|
|
//stm: @CHAIN_STORE_IMPORT_001
|
|
//stm: @CHAIN_STORE_GET_TIPSET_BY_HEIGHT_001, @CHAIN_STORE_PUT_TIPSET_001, @CHAIN_STORE_SET_GENESIS_BLOCK_001
|
|
//stm: @CHAIN_STORE_CLOSE_001
|
|
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.NewMemorySync()
|
|
cs := store.NewChainStore(nbs, nbs, syncds.MutexWrap(datastore.NewMapDatastore()), filcns.Weight, nil)
|
|
defer cs.Close() //nolint:errcheck
|
|
|
|
_, err = cs.Import(ctx, bytes.NewReader(gencar))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cur := mock.TipSet(gen)
|
|
|
|
assert.NoError(t, cs.SetGenesis(ctx, gen))
|
|
|
|
// Put 113 blocks from genesis
|
|
for i := 0; i < 113; i++ {
|
|
nextBlk := mock.MkBlock(cur, 1, 1)
|
|
nextts := mock.TipSet(nextBlk)
|
|
assert.NoError(t, cs.PersistTipsets(ctx, []*types.TipSet{nextts}))
|
|
assert.NoError(t, cs.AddToTipSetTracker(ctx, nextBlk))
|
|
cur = nextts
|
|
}
|
|
|
|
assert.NoError(t, cs.RefreshHeaviestTipSet(ctx, cur.Height()))
|
|
|
|
// Put 50 null epochs + 1 block
|
|
skip := mock.MkBlock(cur, 1, 1)
|
|
skip.Height += 50
|
|
skipts := mock.TipSet(skip)
|
|
|
|
assert.NoError(t, cs.PersistTipsets(ctx, []*types.TipSet{skipts}))
|
|
assert.NoError(t, cs.AddToTipSetTracker(ctx, skip))
|
|
|
|
if err := cs.RefreshHeaviestTipSet(ctx, skip.Height); 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())
|
|
}
|
|
}
|