package store_test import ( "bytes" "context" "testing" "github.com/filecoin-project/go-state-types/abi" "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" 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) } assert.NoError(t, 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()) } }