fix: lint errors

babylonlabs-io · Aug 6, 2024 · 6fe90ea · 6fe90ea
1 parent 04a6c73
commit 6fe90ea
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 11 deletions.
diff --git a/db/bbolt_test.go b/db/bbolt_test.go
@@ -38,7 +38,10 @@ func setupDB(t *testing.T) (*BBoltHandler, func()) {
 	// Cleanup function to close DB and remove temp file
 	cleanup := func() {
 		db.Close()
-		db.DeleteDB()
+		err := db.DeleteDB()
+		if err != nil {
+			t.Fatalf("Failed to delete DB: %v", err)
+		}
 	}
 
 	return db, cleanup

diff --git a/finalitygadget/finalitygadget.go b/finalitygadget/finalitygadget.go
@@ -33,13 +33,12 @@ type FinalityGadget struct {
 	cwClient  cwclient.ICosmWasmClient
 	l2Client  ethl2client.IEthL2Client
 
-	db    db.IDatabaseHandler
-	mutex sync.Mutex
+	db     db.IDatabaseHandler
+	logger *zap.Logger
+	mutex  sync.Mutex
 
 	pollInterval time.Duration
 	currHeight   uint64
-
-	logger *zap.Logger
 }
 
 //////////////////////////////

diff --git a/finalitygadget/finalitygadget_test.go b/finalitygadget/finalitygadget_test.go
@@ -428,16 +428,12 @@ func TestGetBlockByHashForNonExistentBlock(t *testing.T) {
 }
 
 func TestGetBlockStatusByHeight(t *testing.T) {
-	block := &types.Block{
-		BlockHeight:    1,
-		BlockHash:      "0x123",
-		BlockTimestamp: 1000,
-	}
+	blockHeight := uint64(1)
 
 	// mock db and finality gadget
 	ctl := gomock.NewController(t)
 	mockDbHandler := mocks.NewMockIDatabaseHandler(ctl)
-	mockDbHandler.EXPECT().GetBlockStatusByHeight(block.BlockHeight).Return(true, nil).Times(1)
+	mockDbHandler.EXPECT().GetBlockStatusByHeight(blockHeight).Return(true, nil).Times(1)
 
 	mockFinalityGadget := &FinalityGadget{
 		db: mockDbHandler,