add get current blocknumber to ut

Layr-Labs · Apr 18, 2024 · 3d60bf6 · 3d60bf6
1 parent 7c2f690
commit 3d60bf6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/integration_test.go b/test/integration_test.go
@@ -401,6 +401,7 @@ func TestDispersalAndRetrieval(t *testing.T) {
 
 	txn := types.NewTransaction(0, gethcommon.Address{}, big.NewInt(0), 0, big.NewInt(0), nil)
 	dis.transactor.On("BuildConfirmBatchTxn").Return(txn, nil)
+	dis.transactor.On("GetCurrentBlockNumber").Return(uint(10), nil)
 	dis.txnManager.On("ProcessTransaction").Return(nil)
 
 	err = dis.batcher.HandleSingleBatch(ctx)