@@ -136,11 +136,11 @@ func TestProcessNextDAHeader_Success_SingleHeaderAndData(t *testing.T) {
136136 blockDataBytes , err := signedData .MarshalBinary ()
137137 require .NoError (t , err )
138138 // -----------------------------------------------------------
139- mockDAClient .On ("GetIDs" , mock .Anything , daHeight ).Return (& coreda.GetIDsResult {
139+ mockDAClient .On ("GetIDs" , mock .Anything , daHeight , mock . Anything ).Return (& coreda.GetIDsResult {
140140 IDs : []coreda.ID {[]byte ("dummy-id" )},
141141 Timestamp : time .Now (),
142142 }, nil ).Once ()
143- mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}).Return (
143+ mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}, mock . Anything ).Return (
144144 []coreda.Blob {headerBytes , blockDataBytes }, nil ,
145145 ).Once ()
146146
@@ -248,11 +248,11 @@ func TestProcessNextDAHeader_MultipleHeadersAndData(t *testing.T) {
248248 // Add a few more invalid blobs at the end
249249 blobs = append (blobs , invalidBlob , []byte {})
250250
251- mockDAClient .On ("GetIDs" , mock .Anything , daHeight ).Return (& coreda.GetIDsResult {
251+ mockDAClient .On ("GetIDs" , mock .Anything , daHeight , mock . Anything ).Return (& coreda.GetIDsResult {
252252 IDs : []coreda.ID {[]byte ("dummy-id" )},
253253 Timestamp : time .Now (),
254254 }, nil ).Once ()
255- mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}).Return (
255+ mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}, mock . Anything ).Return (
256256 blobs , nil ,
257257 ).Once ()
258258
@@ -313,7 +313,7 @@ func TestProcessNextDAHeaderAndData_NotFound(t *testing.T) {
313313
314314 // Mock GetIDs to return empty IDs to simulate "not found" scenario
315315 // Example updates needed for one instance:
316- mockDAClient .On ("GetIDs" , mock .Anything , daHeight ).Return (& coreda.GetIDsResult {
316+ mockDAClient .On ("GetIDs" , mock .Anything , daHeight , mock . Anything ).Return (& coreda.GetIDsResult {
317317 IDs : []coreda.ID {},
318318 Timestamp : time .Now (),
319319 }, coreda .ErrBlobNotFound ).Once ()
@@ -346,13 +346,13 @@ func TestProcessNextDAHeaderAndData_UnmarshalHeaderError(t *testing.T) {
346346 invalidBytes := []byte ("this is not a valid protobuf message" )
347347
348348 // Mock GetIDs to return success with dummy ID
349- mockDAClient .On ("GetIDs" , mock .Anything , daHeight ).Return (& coreda.GetIDsResult {
349+ mockDAClient .On ("GetIDs" , mock .Anything , daHeight , mock . Anything ).Return (& coreda.GetIDsResult {
350350 IDs : []coreda.ID {[]byte ("dummy-id" )},
351351 Timestamp : time .Now (),
352352 }, nil ).Once ()
353353
354354 // Mock Get to return invalid bytes
355- mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}).Return (
355+ mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}, mock . Anything ).Return (
356356 []coreda.Blob {invalidBytes }, nil ,
357357 ).Once ()
358358
@@ -405,13 +405,13 @@ func TestProcessNextDAHeader_UnexpectedSequencer(t *testing.T) {
405405 require .NoError (t , err )
406406
407407 // Mock GetIDs to return success with dummy ID
408- mockDAClient .On ("GetIDs" , mock .Anything , daHeight ).Return (& coreda.GetIDsResult {
408+ mockDAClient .On ("GetIDs" , mock .Anything , daHeight , mock . Anything ).Return (& coreda.GetIDsResult {
409409 IDs : []coreda.ID {[]byte ("dummy-id" )},
410410 Timestamp : time .Now (),
411411 }, nil ).Once ()
412412
413413 // Mock Get to return header bytes
414- mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}).Return (
414+ mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}, mock . Anything ).Return (
415415 []coreda.Blob {headerBytes }, nil ,
416416 ).Once ()
417417
@@ -449,7 +449,7 @@ func TestProcessNextDAHeader_FetchError_RetryFailure(t *testing.T) {
449449 fetchErr := errors .New ("persistent DA connection error" )
450450
451451 // Mock GetIDs to return error for all retries
452- mockDAClient .On ("GetIDs" , mock .Anything , daHeight ).Return (
452+ mockDAClient .On ("GetIDs" , mock .Anything , daHeight , mock . Anything ).Return (
453453 nil , fetchErr ,
454454 ).Times (dAFetcherRetries )
455455
@@ -539,7 +539,7 @@ func TestProcessNextDAHeader_HeaderAndDataAlreadySeen(t *testing.T) {
539539 Timestamp : time .Now (),
540540 }, nil ).Once ()
541541
542- mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}).Return (
542+ mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}, mock . Anything ).Return (
543543 []coreda.Blob {headerBytes , blockDataBytes }, nil ,
544544 ).Once ()
545545
@@ -577,12 +577,12 @@ func TestRetrieveLoop_ProcessError_HeightFromFuture(t *testing.T) {
577577 futureErr := fmt .Errorf ("some error wrapping: %w" , ErrHeightFromFutureStr )
578578
579579 // Mock GetIDs to return future error for all retries
580- mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight ).Return (
580+ mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight , mock . Anything ).Return (
581581 nil , futureErr ,
582582 ).Once ()
583583
584584 // Optional: Mock for the next height if needed
585- mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight + 1 ).Return (
585+ mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight + 1 , mock . Anything ).Return (
586586 & coreda.GetIDsResult {IDs : []coreda.ID {}}, coreda .ErrBlobNotFound ,
587587 ).Maybe ()
588588
@@ -625,7 +625,7 @@ func TestRetrieveLoop_ProcessError_Other(t *testing.T) {
625625 otherErr := errors .New ("some other DA error" )
626626
627627 // Mock GetIDs to return error for all retries
628- mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight ).Return (
628+ mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight , mock . Anything ).Return (
629629 nil , otherErr ,
630630 ).Times (dAFetcherRetries )
631631
@@ -709,11 +709,11 @@ func TestProcessNextDAHeader_WithNoTxs(t *testing.T) {
709709 emptyDataBytes , err := emptySignedData .MarshalBinary ()
710710 require .NoError (t , err )
711711
712- mockDAClient .On ("GetIDs" , mock .Anything , daHeight ).Return (& coreda.GetIDsResult {
712+ mockDAClient .On ("GetIDs" , mock .Anything , daHeight , mock . Anything ).Return (& coreda.GetIDsResult {
713713 IDs : []coreda.ID {[]byte ("dummy-id" )},
714714 Timestamp : time .Now (),
715715 }, nil ).Once ()
716- mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}).Return (
716+ mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}, mock . Anything ).Return (
717717 []coreda.Blob {headerBytes , emptyDataBytes }, nil ,
718718 ).Once ()
719719
@@ -762,23 +762,23 @@ func TestRetrieveLoop_DAHeightIncrementsOnlyOnSuccess(t *testing.T) {
762762 require .NoError (t , err )
763763
764764 // 1. First call: success (header)
765- mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight ).Return (& coreda.GetIDsResult {
765+ mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight , mock . Anything ).Return (& coreda.GetIDsResult {
766766 IDs : []coreda.ID {[]byte ("dummy-id" )},
767767 Timestamp : time .Now (),
768768 }, nil ).Once ()
769- mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}).Return (
769+ mockDAClient .On ("Get" , mock .Anything , []coreda.ID {[]byte ("dummy-id" )}, mock . Anything ).Return (
770770 []coreda.Blob {headerBytes }, nil ,
771771 ).Once ()
772772
773773 // 2. Second call: NotFound
774- mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight + 1 ).Return (& coreda.GetIDsResult {
774+ mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight + 1 , mock . Anything ).Return (& coreda.GetIDsResult {
775775 IDs : nil ,
776776 Timestamp : time .Now (),
777777 }, nil ).Once ()
778778
779779 // 3. Third call: Error
780780 errDA := errors .New ("some DA error" )
781- mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight + 2 ).Return (
781+ mockDAClient .On ("GetIDs" , mock .Anything , startDAHeight + 2 , mock . Anything ).Return (
782782 & coreda.GetIDsResult {
783783 IDs : nil ,
784784 Timestamp : time .Now (),
0 commit comments