73 lines
2.0 KiB
Go
73 lines
2.0 KiB
Go
package store
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/types/mock"
|
|
)
|
|
|
|
func TestHeadChangeCoalescer(t *testing.T) {
|
|
notif := make(chan headChange, 1)
|
|
c := NewHeadChangeCoalescer(func(revert, apply []*types.TipSet) error {
|
|
notif <- headChange{apply: apply, revert: revert}
|
|
return nil
|
|
},
|
|
100*time.Millisecond,
|
|
200*time.Millisecond,
|
|
10*time.Millisecond,
|
|
)
|
|
defer c.Close() //nolint
|
|
|
|
b0 := mock.MkBlock(nil, 0, 0)
|
|
root := mock.TipSet(b0)
|
|
bA := mock.MkBlock(root, 1, 1)
|
|
tA := mock.TipSet(bA)
|
|
bB := mock.MkBlock(root, 1, 2)
|
|
tB := mock.TipSet(bB)
|
|
tAB := mock.TipSet(bA, bB)
|
|
bC := mock.MkBlock(root, 1, 3)
|
|
tABC := mock.TipSet(bA, bB, bC)
|
|
bD := mock.MkBlock(root, 1, 4)
|
|
tABCD := mock.TipSet(bA, bB, bC, bD)
|
|
bE := mock.MkBlock(root, 1, 5)
|
|
tABCDE := mock.TipSet(bA, bB, bC, bD, bE)
|
|
|
|
c.HeadChange(nil, []*types.TipSet{tA}) //nolint
|
|
c.HeadChange(nil, []*types.TipSet{tB}) //nolint
|
|
c.HeadChange([]*types.TipSet{tA, tB}, []*types.TipSet{tAB}) //nolint
|
|
c.HeadChange([]*types.TipSet{tAB}, []*types.TipSet{tABC}) //nolint
|
|
|
|
change := <-notif
|
|
|
|
if len(change.revert) != 0 {
|
|
t.Fatalf("expected empty revert set but got %d elements", len(change.revert))
|
|
}
|
|
if len(change.apply) != 1 {
|
|
t.Fatalf("expected single element apply set but got %d elements", len(change.apply))
|
|
}
|
|
if change.apply[0] != tABC {
|
|
t.Fatalf("expected to apply tABC")
|
|
}
|
|
|
|
c.HeadChange([]*types.TipSet{tABC}, []*types.TipSet{tABCD}) //nolint
|
|
c.HeadChange([]*types.TipSet{tABCD}, []*types.TipSet{tABCDE}) //nolint
|
|
|
|
change = <-notif
|
|
|
|
if len(change.revert) != 1 {
|
|
t.Fatalf("expected single element revert set but got %d elements", len(change.revert))
|
|
}
|
|
if change.revert[0] != tABC {
|
|
t.Fatalf("expected to revert tABC")
|
|
}
|
|
if len(change.apply) != 1 {
|
|
t.Fatalf("expected single element apply set but got %d elements", len(change.apply))
|
|
}
|
|
if change.apply[0] != tABCDE {
|
|
t.Fatalf("expected to revert tABC")
|
|
}
|
|
|
|
}
|