From 897b26913957ac2560394f3a804e16cccb69e9f6 Mon Sep 17 00:00:00 2001
From: Aayush Rajasekaran <arajasek94@gmail.com>
Date: Thu, 30 Apr 2020 20:54:41 -0400
Subject: [PATCH] Test to demonstrate HAMT issue

---
 chain/gen/gen.go      | 10 +++++++---
 chain/gen/gen_test.go | 15 ++++++++-------
 2 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/chain/gen/gen.go b/chain/gen/gen.go
index 50c8632ce..97c70bb4b 100644
--- a/chain/gen/gen.go
+++ b/chain/gen/gen.go
@@ -87,7 +87,7 @@ func (m mybs) Get(c cid.Cid) (block.Block, error) {
 	return b, nil
 }
 
-func NewGenerator() (*ChainGen, error) {
+func NewGeneratorWithSectors(numSectors int) (*ChainGen, error) {
 	mr := repo.NewMemory(nil)
 	lr, err := mr.Lock(repo.StorageMiner)
 	if err != nil {
@@ -136,7 +136,7 @@ func NewGenerator() (*ChainGen, error) {
 		return nil, err
 	}
 
-	genm1, k1, err := seed.PreSeal(maddr1, abi.RegisteredProof_StackedDRG2KiBPoSt, 0, 1, m1temp, []byte("some randomness"), nil)
+	genm1, k1, err := seed.PreSeal(maddr1, abi.RegisteredProof_StackedDRG2KiBPoSt, 0, numSectors, m1temp, []byte("some randomness"), nil)
 	if err != nil {
 		return nil, err
 	}
@@ -148,7 +148,7 @@ func NewGenerator() (*ChainGen, error) {
 		return nil, err
 	}
 
-	genm2, k2, err := seed.PreSeal(maddr2, abi.RegisteredProof_StackedDRG2KiBPoSt, 0, 1, m2temp, []byte("some randomness"), nil)
+	genm2, k2, err := seed.PreSeal(maddr2, abi.RegisteredProof_StackedDRG2KiBPoSt, 0, numSectors, m2temp, []byte("some randomness"), nil)
 	if err != nil {
 		return nil, err
 	}
@@ -239,6 +239,10 @@ func NewGenerator() (*ChainGen, error) {
 	return gen, nil
 }
 
+func NewGenerator() (*ChainGen, error) {
+	return NewGeneratorWithSectors(1)
+}
+
 func (cg *ChainGen) SetStateManager(sm *stmgr.StateManager) {
 	cg.sm = sm
 }
diff --git a/chain/gen/gen_test.go b/chain/gen/gen_test.go
index b61a59e10..22381fa88 100644
--- a/chain/gen/gen_test.go
+++ b/chain/gen/gen_test.go
@@ -19,8 +19,8 @@ func init() {
 	power.ConsensusMinerMinPower = big.NewInt(2048)
 }
 
-func testGeneration(t testing.TB, n int, msgs int) {
-	g, err := NewGenerator()
+func testGeneration(t testing.TB, n int, msgs int, sectors int) {
+	g, err := NewGeneratorWithSectors(sectors)
 	if err != nil {
 		t.Fatalf("%+v", err)
 	}
@@ -37,23 +37,24 @@ func testGeneration(t testing.TB, n int, msgs int) {
 }
 
 func TestChainGeneration(t *testing.T) {
-	testGeneration(t, 10, 20)
+	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)
+		testGeneration(b, b.N, 0, 1)
 	})
 
 	b.Run("10-messages", func(b *testing.B) {
-		testGeneration(b, b.N, 10)
+		testGeneration(b, b.N, 10, 1)
 	})
 
 	b.Run("100-messages", func(b *testing.B) {
-		testGeneration(b, b.N, 100)
+		testGeneration(b, b.N, 100, 1)
 	})
 
 	b.Run("1000-messages", func(b *testing.B) {
-		testGeneration(b, b.N, 1000)
+		testGeneration(b, b.N, 1000, 1)
 	})
 }