@@ -25,6 +25,7 @@ import (
2525 "github.com/hyperledger/firefly/internal/config"
2626 "github.com/hyperledger/firefly/internal/data"
2727 "github.com/hyperledger/firefly/internal/log"
28+ "github.com/hyperledger/firefly/internal/txcommon"
2829 "github.com/hyperledger/firefly/mocks/databasemocks"
2930 "github.com/hyperledger/firefly/mocks/datamocks"
3031 "github.com/hyperledger/firefly/mocks/sysmessagingmocks"
@@ -41,6 +42,7 @@ func TestE2EDispatchBroadcast(t *testing.T) {
4142 mdi := & databasemocks.Plugin {}
4243 mdm := & datamocks.Manager {}
4344 mni := & sysmessagingmocks.LocalNodeInfo {}
45+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
4446 mni .On ("GetNodeUUID" , mock .Anything ).Return (fftypes .NewUUID ())
4547 readyForDispatch := make (chan bool )
4648 waitForDispatch := make (chan * DispatchState )
@@ -70,7 +72,7 @@ func TestE2EDispatchBroadcast(t *testing.T) {
7072 return nil
7173 }
7274 ctx , cancel := context .WithCancel (context .Background ())
73- bmi , _ := NewBatchManager (ctx , mni , mdi , mdm )
75+ bmi , _ := NewBatchManager (ctx , mni , mdi , mdm , txHelper )
7476 bm := bmi .(* batchManager )
7577
7678 bm .RegisterDispatcher ("utdispatcher" , fftypes .TransactionTypeBatchPin , []fftypes.MessageType {fftypes .MessageTypeBroadcast }, handler , DispatcherOptions {
@@ -155,6 +157,7 @@ func TestE2EDispatchPrivateUnpinned(t *testing.T) {
155157 mdi := & databasemocks.Plugin {}
156158 mdm := & datamocks.Manager {}
157159 mni := & sysmessagingmocks.LocalNodeInfo {}
160+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
158161 mni .On ("GetNodeUUID" , mock .Anything ).Return (fftypes .NewUUID ())
159162 readyForDispatch := make (chan bool )
160163 waitForDispatch := make (chan * DispatchState )
@@ -187,7 +190,7 @@ func TestE2EDispatchPrivateUnpinned(t *testing.T) {
187190 return nil
188191 }
189192 ctx , cancel := context .WithCancel (context .Background ())
190- bmi , _ := NewBatchManager (ctx , mni , mdi , mdm )
193+ bmi , _ := NewBatchManager (ctx , mni , mdi , mdm , txHelper )
191194 bm := bmi .(* batchManager )
192195
193196 bm .RegisterDispatcher ("utdispatcher" , fftypes .TransactionTypeBatchPin , []fftypes.MessageType {fftypes .MessageTypePrivate }, handler , DispatcherOptions {
@@ -268,8 +271,9 @@ func TestDispatchUnknownType(t *testing.T) {
268271 mdi := & databasemocks.Plugin {}
269272 mdm := & datamocks.Manager {}
270273 mni := & sysmessagingmocks.LocalNodeInfo {}
274+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
271275 ctx , cancel := context .WithCancel (context .Background ())
272- bmi , _ := NewBatchManager (ctx , mni , mdi , mdm )
276+ bmi , _ := NewBatchManager (ctx , mni , mdi , mdm , txHelper )
273277 bm := bmi .(* batchManager )
274278
275279 msg := & fftypes.Message {}
@@ -285,7 +289,7 @@ func TestDispatchUnknownType(t *testing.T) {
285289}
286290
287291func TestInitFailNoPersistence (t * testing.T ) {
288- _ , err := NewBatchManager (context .Background (), nil , nil , nil )
292+ _ , err := NewBatchManager (context .Background (), nil , nil , nil , nil )
289293 assert .Error (t , err )
290294}
291295
@@ -294,7 +298,8 @@ func TestGetInvalidBatchTypeMsg(t *testing.T) {
294298 mdi := & databasemocks.Plugin {}
295299 mdm := & datamocks.Manager {}
296300 mni := & sysmessagingmocks.LocalNodeInfo {}
297- bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
301+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
302+ bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
298303 defer bm .Close ()
299304 _ , err := bm .(* batchManager ).getProcessor (fftypes .BatchTypeBroadcast , "wrong" , nil , "ns1" , & fftypes.SignerRef {})
300305 assert .Regexp (t , "FF10126" , err )
@@ -304,8 +309,9 @@ func TestMessageSequencerCancelledContext(t *testing.T) {
304309 mdi := & databasemocks.Plugin {}
305310 mdm := & datamocks.Manager {}
306311 mni := & sysmessagingmocks.LocalNodeInfo {}
312+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
307313 mdi .On ("GetMessages" , mock .Anything , mock .Anything , mock .Anything ).Return (nil , nil , fmt .Errorf ("pop" ))
308- bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
314+ bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
309315 defer bm .Close ()
310316 ctx , cancel := context .WithCancel (context .Background ())
311317 cancel ()
@@ -318,7 +324,8 @@ func TestMessageSequencerMissingMessageData(t *testing.T) {
318324 mdi := & databasemocks.Plugin {}
319325 mdm := & datamocks.Manager {}
320326 mni := & sysmessagingmocks.LocalNodeInfo {}
321- bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
327+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
328+ bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
322329 bm .RegisterDispatcher ("utdispatcher" , fftypes .TransactionTypeNone , []fftypes.MessageType {fftypes .MessageTypeBroadcast },
323330 func (c context.Context , state * DispatchState ) error {
324331 return nil
@@ -359,9 +366,10 @@ func TestMessageSequencerUpdateMessagesFail(t *testing.T) {
359366 mdi := & databasemocks.Plugin {}
360367 mdm := & datamocks.Manager {}
361368 mni := & sysmessagingmocks.LocalNodeInfo {}
369+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
362370 mni .On ("GetNodeUUID" , mock .Anything ).Return (fftypes .NewUUID ())
363371 ctx , cancelCtx := context .WithCancel (context .Background ())
364- bm , _ := NewBatchManager (ctx , mni , mdi , mdm )
372+ bm , _ := NewBatchManager (ctx , mni , mdi , mdm , txHelper )
365373 bm .RegisterDispatcher ("utdispatcher" , fftypes .TransactionTypeBatchPin , []fftypes.MessageType {fftypes .MessageTypeBroadcast },
366374 func (c context.Context , state * DispatchState ) error {
367375 return nil
@@ -414,8 +422,9 @@ func TestMessageSequencerDispatchFail(t *testing.T) {
414422 mdm := & datamocks.Manager {}
415423 mni := & sysmessagingmocks.LocalNodeInfo {}
416424 mni .On ("GetNodeUUID" , mock .Anything ).Return (fftypes .NewUUID ())
425+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
417426 ctx , cancelCtx := context .WithCancel (context .Background ())
418- bm , _ := NewBatchManager (ctx , mni , mdi , mdm )
427+ bm , _ := NewBatchManager (ctx , mni , mdi , mdm , txHelper )
419428 bm .RegisterDispatcher ("utdispatcher" , fftypes .TransactionTypeBatchPin , []fftypes.MessageType {fftypes .MessageTypeBroadcast },
420429 func (c context.Context , state * DispatchState ) error {
421430 cancelCtx ()
@@ -454,7 +463,8 @@ func TestMessageSequencerUpdateBatchFail(t *testing.T) {
454463 mni := & sysmessagingmocks.LocalNodeInfo {}
455464 mni .On ("GetNodeUUID" , mock .Anything ).Return (fftypes .NewUUID ())
456465 ctx , cancelCtx := context .WithCancel (context .Background ())
457- bm , _ := NewBatchManager (ctx , mni , mdi , mdm )
466+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
467+ bm , _ := NewBatchManager (ctx , mni , mdi , mdm , txHelper )
458468 bm .RegisterDispatcher ("utdispatcher" , fftypes .TransactionTypeBatchPin , []fftypes.MessageType {fftypes .MessageTypeBroadcast },
459469 func (c context.Context , state * DispatchState ) error {
460470 return nil
@@ -504,7 +514,8 @@ func TestWaitForPollTimeout(t *testing.T) {
504514 mdi := & databasemocks.Plugin {}
505515 mdm := & datamocks.Manager {}
506516 mni := & sysmessagingmocks.LocalNodeInfo {}
507- bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
517+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
518+ bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
508519 bm .(* batchManager ).messagePollTimeout = 1 * time .Microsecond
509520 bm .(* batchManager ).waitForNewMessages ()
510521}
@@ -513,7 +524,8 @@ func TestWaitForNewMessage(t *testing.T) {
513524 mdi := & databasemocks.Plugin {}
514525 mdm := & datamocks.Manager {}
515526 mni := & sysmessagingmocks.LocalNodeInfo {}
516- bmi , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
527+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
528+ bmi , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
517529 bm := bmi .(* batchManager )
518530 bm .readOffset = 22222
519531 bm .NewMessages () <- 12345
@@ -525,7 +537,8 @@ func TestAssembleMessageDataNilData(t *testing.T) {
525537 mdi := & databasemocks.Plugin {}
526538 mdm := & datamocks.Manager {}
527539 mni := & sysmessagingmocks.LocalNodeInfo {}
528- bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
540+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
541+ bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
529542 bm .Close ()
530543 mdm .On ("GetMessageDataCached" , mock .Anything , mock .Anything ).Return (nil , false , nil )
531544 _ , err := bm .(* batchManager ).assembleMessageData (fftypes .BatchTypePrivate , & fftypes.Message {
@@ -541,7 +554,8 @@ func TestGetMessageDataFail(t *testing.T) {
541554 mdi := & databasemocks.Plugin {}
542555 mdm := & datamocks.Manager {}
543556 mni := & sysmessagingmocks.LocalNodeInfo {}
544- bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
557+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
558+ bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
545559 mdm .On ("GetMessageDataCached" , mock .Anything , mock .Anything ).Return (nil , false , fmt .Errorf ("pop" ))
546560 bm .Close ()
547561 _ , _ = bm .(* batchManager ).assembleMessageData (fftypes .BatchTypePrivate , & fftypes.Message {
@@ -559,7 +573,8 @@ func TestGetMessageNotFound(t *testing.T) {
559573 mdi := & databasemocks.Plugin {}
560574 mdm := & datamocks.Manager {}
561575 mni := & sysmessagingmocks.LocalNodeInfo {}
562- bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm )
576+ txHelper := txcommon .NewTransactionHelper (mdi , mdm )
577+ bm , _ := NewBatchManager (context .Background (), mni , mdi , mdm , txHelper )
563578 mdm .On ("GetMessageDataCached" , mock .Anything , mock .Anything , data .CRORequirePublicBlobRefs ).Return (nil , false , nil )
564579 bm .Close ()
565580 _ , err := bm .(* batchManager ).assembleMessageData (fftypes .BatchTypeBroadcast , & fftypes.Message {
0 commit comments