package gen import ( "testing" "github.com/filecoin-project/specs-actors/actors/abi" "github.com/filecoin-project/specs-actors/actors/abi/big" "github.com/filecoin-project/specs-actors/actors/builtin/miner" "github.com/filecoin-project/specs-actors/actors/builtin/power" "github.com/filecoin-project/specs-actors/actors/builtin/verifreg" _ "github.com/filecoin-project/lotus/lib/sigs/bls" _ "github.com/filecoin-project/lotus/lib/sigs/secp" ) func init() { miner.SupportedProofTypes = map[abi.RegisteredProof]struct{}{ abi.RegisteredProof_StackedDRG2KiBSeal: {}, } power.ConsensusMinerMinPower = big.NewInt(2048) verifreg.MinVerifiedDealSize = big.NewInt(256) } func testGeneration(t testing.TB, n int, msgs int, sectors int) { g, err := NewGeneratorWithSectors(sectors) if err != nil { t.Fatalf("%+v", err) } g.msgsPerBlock = msgs for i := 0; i < n; i++ { mts, err := g.NextTipSet() if err != nil { t.Fatalf("error at H:%d, %+v", i, err) } _ = mts } } func TestChainGeneration(t *testing.T) { testGeneration(t, 10, 20, 1) testGeneration(t, 10, 20, 25) } func BenchmarkChainGeneration(b *testing.B) { b.Run("0-messages", func(b *testing.B) { testGeneration(b, b.N, 0, 1) }) b.Run("10-messages", func(b *testing.B) { testGeneration(b, b.N, 10, 1) }) b.Run("100-messages", func(b *testing.B) { testGeneration(b, b.N, 100, 1) }) b.Run("1000-messages", func(b *testing.B) { testGeneration(b, b.N, 1000, 1) }) }