fix: ut

bnb-chain · Jun 19, 2024 · 51e0f31 · 51e0f31
1 parent e757d28
commit 51e0f31
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modular/receiver/receive_task_test.go b/modular/receiver/receive_task_test.go
@@ -3,6 +3,7 @@ package receiver
 import (
 	"context"
 	"fmt"
+	"gorm.io/gorm"
 	"testing"
 
 	sdkmath "cosmossdk.io/math"
@@ -236,6 +237,7 @@ func TestHandleDoneReceivePieceTask_PieceCountMismatch(t *testing.T) {
 	mockSPDB := spdb.NewMockSPDB(ctrl)
 	r.baseApp.SetGfSpDB(mockSPDB)
 	mockSPDB.EXPECT().GetAllReplicatePieceChecksumOptimized(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, nil).Times(1)
+	mockSPDB.EXPECT().GetObjectIntegrity(gomock.Any(), gomock.Any()).Return(nil, gorm.ErrRecordNotFound).AnyTimes()
 	_, err := r.HandleDoneReceivePieceTask(context.TODO(), mockTask)
 	assert.NotNil(t, err)
 }