fix mocks

babylonlabs-io · Sep 16, 2024 · 6fcef59 · 6fcef59
1 parent 48e5d1c
commit 6fcef59
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/btcclient/query.go b/btcclient/query.go
@@ -12,8 +12,7 @@ import (
 	"github.com/babylonlabs-io/vigilante/types"
 )
 
-// GetBestBlock provides similar functionality with the btcd.rpcclient.GetBestBlock function
-// We implement this, because this function is only provided by btcd.
+// GetBestBlock returns the height of the best block
 func (c *Client) GetBestBlock() (uint64, error) {
 	height, err := c.getBlockCountWithRetry()
 	if err != nil {

diff --git a/monitor/btcscanner/btc_scanner_test.go b/monitor/btcscanner/btc_scanner_test.go
@@ -30,7 +30,7 @@ func FuzzBootStrap(f *testing.F) {
 		defer ctl.Finish()
 		mockBtcClient := mocks.NewMockBTCClient(ctl)
 		confirmedBlocks := chainIndexedBlocks[:numBlocks-k]
-		mockBtcClient.EXPECT().GetBestBlock().Return(nil, uint64(bestHeight), nil)
+		mockBtcClient.EXPECT().GetBestBlock().Return(uint64(bestHeight), nil)
 		for i := 0; i < int(numBlocks); i++ {
 			mockBtcClient.EXPECT().GetBlockByHeight(gomock.Eq(uint64(chainIndexedBlocks[i].Height))).
 				Return(chainIndexedBlocks[i], nil, nil).AnyTimes()