lotus/chain/events/tscache_test.go

123 lines
2.6 KiB
Go
Raw Normal View History

2019-09-30 22:38:07 +00:00
package events
import (
"context"
"testing"
2019-10-01 21:34:53 +00:00
"github.com/filecoin-project/specs-actors/actors/abi"
2020-02-23 07:49:15 +00:00
"github.com/filecoin-project/specs-actors/actors/crypto"
"github.com/stretchr/testify/require"
"github.com/filecoin-project/go-address"
"github.com/filecoin-project/lotus/chain/types"
2019-09-30 22:38:07 +00:00
)
func TestTsCache(t *testing.T) {
tsc := newTSCache(50, func(context.Context, abi.ChainEpoch, types.TipSetKey) (*types.TipSet, error) {
2019-09-30 22:38:07 +00:00
t.Fatal("storage call")
return &types.TipSet{}, nil
})
h := abi.ChainEpoch(75)
2019-09-30 22:38:07 +00:00
a, _ := address.NewFromString("t00")
2019-09-30 22:38:07 +00:00
add := func() {
ts, err := types.NewTipSet([]*types.BlockHeader{{
Miner: a,
2019-10-01 21:34:53 +00:00
Height: h,
ParentStateRoot: dummyCid,
Messages: dummyCid,
ParentMessageReceipts: dummyCid,
2020-02-26 09:05:22 +00:00
BlockSig: &crypto.Signature{Type: crypto.SigTypeBLS},
2020-03-21 22:25:00 +00:00
BLSAggregate: &crypto.Signature{Type: crypto.SigTypeBLS},
2019-09-30 22:38:07 +00:00
}})
if err != nil {
t.Fatal(err)
}
if err := tsc.add(ts); err != nil {
t.Fatal(err)
}
h++
}
for i := 0; i < 9000; i++ {
if i%90 > 60 {
if err := tsc.revert(tsc.best()); err != nil {
t.Fatal(err, "; i:", i)
return
}
h--
} else {
add()
}
}
}
2019-10-04 22:43:04 +00:00
func TestTsCacheNulls(t *testing.T) {
tsc := newTSCache(50, func(context.Context, abi.ChainEpoch, types.TipSetKey) (*types.TipSet, error) {
2019-10-04 22:43:04 +00:00
t.Fatal("storage call")
return &types.TipSet{}, nil
})
h := abi.ChainEpoch(75)
2019-10-04 22:43:04 +00:00
a, _ := address.NewFromString("t00")
2019-10-04 22:43:04 +00:00
add := func() {
ts, err := types.NewTipSet([]*types.BlockHeader{{
Miner: a,
2019-10-04 22:43:04 +00:00
Height: h,
ParentStateRoot: dummyCid,
Messages: dummyCid,
ParentMessageReceipts: dummyCid,
2020-02-26 09:05:22 +00:00
BlockSig: &crypto.Signature{Type: crypto.SigTypeBLS},
2020-03-21 22:25:00 +00:00
BLSAggregate: &crypto.Signature{Type: crypto.SigTypeBLS},
2019-10-04 22:43:04 +00:00
}})
if err != nil {
t.Fatal(err)
}
if err := tsc.add(ts); err != nil {
t.Fatal(err)
}
h++
}
add()
add()
add()
h += 5
add()
add()
require.Equal(t, h-1, tsc.best().Height())
ts, err := tsc.get(h - 1)
require.NoError(t, err)
require.Equal(t, h-1, ts.Height())
ts, err = tsc.get(h - 2)
require.NoError(t, err)
require.Equal(t, h-2, ts.Height())
ts, err = tsc.get(h - 3)
require.NoError(t, err)
require.Nil(t, ts)
ts, err = tsc.get(h - 8)
require.NoError(t, err)
require.Equal(t, h-8, ts.Height())
require.NoError(t, tsc.revert(tsc.best()))
require.NoError(t, tsc.revert(tsc.best()))
require.Equal(t, h-8, tsc.best().Height())
h += 50
add()
ts, err = tsc.get(h - 1)
require.NoError(t, err)
require.Equal(t, h-1, ts.Height())
2019-10-04 22:43:04 +00:00
}