Test to demonstrate HAMT issue

This commit is contained in:
Aayush Rajasekaran 2020-04-30 20:54:41 -04:00
parent a6bb8ec688
commit 897b269139
2 changed files with 15 additions and 10 deletions

View File

@ -87,7 +87,7 @@ func (m mybs) Get(c cid.Cid) (block.Block, error) {
return b, nil return b, nil
} }
func NewGenerator() (*ChainGen, error) { func NewGeneratorWithSectors(numSectors int) (*ChainGen, error) {
mr := repo.NewMemory(nil) mr := repo.NewMemory(nil)
lr, err := mr.Lock(repo.StorageMiner) lr, err := mr.Lock(repo.StorageMiner)
if err != nil { if err != nil {
@ -136,7 +136,7 @@ func NewGenerator() (*ChainGen, error) {
return nil, err 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 { if err != nil {
return nil, err return nil, err
} }
@ -148,7 +148,7 @@ func NewGenerator() (*ChainGen, error) {
return nil, err 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 { if err != nil {
return nil, err return nil, err
} }
@ -239,6 +239,10 @@ func NewGenerator() (*ChainGen, error) {
return gen, nil return gen, nil
} }
func NewGenerator() (*ChainGen, error) {
return NewGeneratorWithSectors(1)
}
func (cg *ChainGen) SetStateManager(sm *stmgr.StateManager) { func (cg *ChainGen) SetStateManager(sm *stmgr.StateManager) {
cg.sm = sm cg.sm = sm
} }

View File

@ -19,8 +19,8 @@ func init() {
power.ConsensusMinerMinPower = big.NewInt(2048) power.ConsensusMinerMinPower = big.NewInt(2048)
} }
func testGeneration(t testing.TB, n int, msgs int) { func testGeneration(t testing.TB, n int, msgs int, sectors int) {
g, err := NewGenerator() g, err := NewGeneratorWithSectors(sectors)
if err != nil { if err != nil {
t.Fatalf("%+v", err) t.Fatalf("%+v", err)
} }
@ -37,23 +37,24 @@ func testGeneration(t testing.TB, n int, msgs int) {
} }
func TestChainGeneration(t *testing.T) { func TestChainGeneration(t *testing.T) {
testGeneration(t, 10, 20) testGeneration(t, 10, 20, 1)
testGeneration(t, 10, 20, 25)
} }
func BenchmarkChainGeneration(b *testing.B) { func BenchmarkChainGeneration(b *testing.B) {
b.Run("0-messages", func(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) { 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) { 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) { b.Run("1000-messages", func(b *testing.B) {
testGeneration(b, b.N, 1000) testGeneration(b, b.N, 1000, 1)
}) })
} }