121 lines
2.4 KiB
Go
121 lines
2.4 KiB
Go
package events
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
)
|
|
|
|
func TestTsCache(t *testing.T) {
|
|
tsc := newTSCache(50, func(context.Context, uint64, types.TipSetKey) (*types.TipSet, error) {
|
|
t.Fatal("storage call")
|
|
return &types.TipSet{}, nil
|
|
})
|
|
|
|
h := uint64(75)
|
|
|
|
a, _ := address.NewFromString("t00")
|
|
|
|
add := func() {
|
|
ts, err := types.NewTipSet([]*types.BlockHeader{{
|
|
Miner: a,
|
|
Height: h,
|
|
ParentStateRoot: dummyCid,
|
|
Messages: dummyCid,
|
|
ParentMessageReceipts: dummyCid,
|
|
BlockSig: &types.Signature{Type: types.KTBLS},
|
|
BLSAggregate: types.Signature{Type: types.KTBLS},
|
|
}})
|
|
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()
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func TestTsCacheNulls(t *testing.T) {
|
|
tsc := newTSCache(50, func(context.Context, uint64, types.TipSetKey) (*types.TipSet, error) {
|
|
t.Fatal("storage call")
|
|
return &types.TipSet{}, nil
|
|
})
|
|
|
|
h := uint64(75)
|
|
|
|
a, _ := address.NewFromString("t00")
|
|
add := func() {
|
|
ts, err := types.NewTipSet([]*types.BlockHeader{{
|
|
Miner: a,
|
|
Height: h,
|
|
ParentStateRoot: dummyCid,
|
|
Messages: dummyCid,
|
|
ParentMessageReceipts: dummyCid,
|
|
BlockSig: &types.Signature{Type: types.KTBLS},
|
|
BLSAggregate: types.Signature{Type: types.KTBLS},
|
|
}})
|
|
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())
|
|
}
|