lotus/chain/gen/gen_test.go

63 lines
1.5 KiB
Go
Raw Normal View History

2019-07-25 22:15:33 +00:00
package gen
import (
"testing"
2019-12-03 01:55:10 +00:00
2020-09-07 03:49:10 +00:00
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/go-state-types/big"
2020-09-17 02:34:13 +00:00
v0miner "github.com/filecoin-project/specs-actors/actors/builtin/miner"
v0power "github.com/filecoin-project/specs-actors/actors/builtin/power"
2020-09-17 07:48:39 +00:00
v0verifreg "github.com/filecoin-project/specs-actors/actors/builtin/verifreg"
2020-02-14 14:14:39 +00:00
_ "github.com/filecoin-project/lotus/lib/sigs/bls"
_ "github.com/filecoin-project/lotus/lib/sigs/secp"
2019-07-25 22:15:33 +00:00
)
2019-12-03 01:55:10 +00:00
func init() {
2020-09-17 02:34:13 +00:00
v0miner.SupportedProofTypes = map[abi.RegisteredSealProof]struct{}{
2020-06-15 16:30:49 +00:00
abi.RegisteredSealProof_StackedDrg2KiBV1: {},
2020-05-12 17:58:12 +00:00
}
v0power.ConsensusMinerMinPower = big.NewInt(2048)
2020-09-17 07:48:39 +00:00
v0verifreg.MinVerifiedDealSize = big.NewInt(256)
2019-12-03 01:55:10 +00:00
}
2020-05-01 00:54:41 +00:00
func testGeneration(t testing.TB, n int, msgs int, sectors int) {
g, err := NewGeneratorWithSectors(sectors)
2019-07-25 22:15:33 +00:00
if err != nil {
2020-02-14 14:14:39 +00:00
t.Fatalf("%+v", err)
2019-07-25 22:15:33 +00:00
}
g.msgsPerBlock = msgs
2019-07-26 12:28:29 +00:00
for i := 0; i < n; i++ {
mts, err := g.NextTipSet()
2019-07-25 22:15:33 +00:00
if err != nil {
2020-02-19 19:39:00 +00:00
t.Fatalf("error at H:%d, %+v", i, err)
2019-07-25 22:15:33 +00:00
}
_ = mts
2019-07-25 22:15:33 +00:00
}
2019-07-26 12:28:29 +00:00
}
func TestChainGeneration(t *testing.T) {
t.Run("10-20-1", func(t *testing.T) { testGeneration(t, 10, 20, 1) })
t.Run("10-20-25", func(t *testing.T) { testGeneration(t, 10, 20, 25) })
2019-07-26 12:28:29 +00:00
}
2019-07-25 22:15:33 +00:00
2019-07-26 12:28:29 +00:00
func BenchmarkChainGeneration(b *testing.B) {
b.Run("0-messages", func(b *testing.B) {
2020-05-01 00:54:41 +00:00
testGeneration(b, b.N, 0, 1)
})
b.Run("10-messages", func(b *testing.B) {
2020-05-01 00:54:41 +00:00
testGeneration(b, b.N, 10, 1)
})
b.Run("100-messages", func(b *testing.B) {
2020-05-01 00:54:41 +00:00
testGeneration(b, b.N, 100, 1)
})
b.Run("1000-messages", func(b *testing.B) {
2020-05-01 00:54:41 +00:00
testGeneration(b, b.N, 1000, 1)
})
2019-07-25 22:15:33 +00:00
}