90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
|
package store_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/lotus/chain/gen"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestChainCheckpoint(t *testing.T) {
|
||
|
|
cg, err := gen.NewGenerator()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Let the first miner mine some blocks.
|
||
|
|
last := cg.CurTipset.TipSet()
|
||
|
|
for i := 0; i < 4; i++ {
|
||
|
|
ts, err := cg.NextTipSetFromMiners(last, cg.Miners[:1])
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
last = ts.TipSet.TipSet()
|
||
|
|
}
|
||
|
|
|
||
|
|
cs := cg.ChainStore()
|
||
|
|
|
||
|
|
checkpoint := last
|
||
|
|
checkpointParents, err := cs.GetTipSetFromKey(checkpoint.Parents())
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Set the head to the block before the checkpoint.
|
||
|
|
err = cs.SetHead(checkpointParents)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Verify it worked.
|
||
|
|
head := cs.GetHeaviestTipSet()
|
||
|
|
require.True(t, head.Equals(checkpointParents))
|
||
|
|
|
||
|
|
// Try to set the checkpoint in the future, it should fail.
|
||
|
|
err = cs.SetCheckpoint(checkpoint)
|
||
|
|
require.Error(t, err)
|
||
|
|
|
||
|
|
// Then move the head back.
|
||
|
|
err = cs.SetHead(checkpoint)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Verify it worked.
|
||
|
|
head = cs.GetHeaviestTipSet()
|
||
|
|
require.True(t, head.Equals(checkpoint))
|
||
|
|
|
||
|
|
// And checkpoint it.
|
||
|
|
err = cs.SetCheckpoint(checkpoint)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Let the second miner miner mine a fork
|
||
|
|
last = checkpointParents
|
||
|
|
for i := 0; i < 4; i++ {
|
||
|
|
ts, err := cg.NextTipSetFromMiners(last, cg.Miners[1:])
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
last = ts.TipSet.TipSet()
|
||
|
|
}
|
||
|
|
|
||
|
|
// See if the chain will take the fork, it shouldn't.
|
||
|
|
err = cs.MaybeTakeHeavierTipSet(context.Background(), last)
|
||
|
|
require.NoError(t, err)
|
||
|
|
head = cs.GetHeaviestTipSet()
|
||
|
|
require.True(t, head.Equals(checkpoint))
|
||
|
|
|
||
|
|
// Remove the checkpoint.
|
||
|
|
err = cs.RemoveCheckpoint()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Now switch to the other fork.
|
||
|
|
err = cs.MaybeTakeHeavierTipSet(context.Background(), last)
|
||
|
|
require.NoError(t, err)
|
||
|
|
head = cs.GetHeaviestTipSet()
|
||
|
|
require.True(t, head.Equals(last))
|
||
|
|
|
||
|
|
// Setting a checkpoint on the other fork should fail.
|
||
|
|
err = cs.SetCheckpoint(checkpoint)
|
||
|
|
require.Error(t, err)
|
||
|
|
|
||
|
|
// Setting a checkpoint on this fork should succeed.
|
||
|
|
err = cs.SetCheckpoint(checkpointParents)
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|