lotus/chain/events/tscache_test.go

121 lines
2.4 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/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, uint64, *types.TipSet) (*types.TipSet, error) {
t.Fatal("storage call")
return &types.TipSet{}, nil
})
h := uint64(75)
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,
2019-11-19 20:07:16 +00:00
BlockSig: &types.Signature{Type: types.KTBLS},
2019-11-19 15:51:12 +00:00
BLSAggregate: types.Signature{Type: types.KTBLS},
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, uint64, *types.TipSet) (*types.TipSet, error) {
t.Fatal("storage call")
return &types.TipSet{}, nil
})
h := uint64(75)
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,
2019-11-19 20:07:16 +00:00
BlockSig: &types.Signature{Type: types.KTBLS},
2019-11-19 15:51:12 +00:00
BLSAggregate: types.Signature{Type: types.KTBLS},
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
}