fix benchmark

babylonchain · Feb 21, 2024 · e1a40d2 · e1a40d2
1 parent d335bbc
commit e1a40d2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/x/btcstaking/keeper/bench_test.go b/x/btcstaking/keeper/bench_test.go
@@ -9,6 +9,7 @@ import (
 	"time"
 
 	"github.com/babylonchain/babylon/testutil/datagen"
+	btclctypes "github.com/babylonchain/babylon/x/btclightclient/types"
 	bsmodule "github.com/babylonchain/babylon/x/btcstaking"
 	"github.com/babylonchain/babylon/x/btcstaking/types"
 	"github.com/golang/mock/gomock"
@@ -67,6 +68,9 @@ func benchBeginBlock(b *testing.B, numFPs int, numDelsUnderFP int) {
 		}
 	}
 
+	// mock stuff
+	h.BTCLightClientKeeper.EXPECT().GetTipInfo(gomock.Eq(h.Ctx)).Return(&btclctypes.BTCHeaderInfo{Height: 30}).AnyTimes()
+
 	// Start the CPU profiler
 	cpuProfileFile := fmt.Sprintf("/tmp/btcstaking-beginblock-%d-%d-cpu.pprof", numFPs, numDelsUnderFP)
 	f, err := os.Create(cpuProfileFile)