@@ -22,10 +22,10 @@ import (
22
22
"testing"
23
23
24
24
"github.com/hyperledger/firefly-common/pkg/fftypes"
25
- "github.com/hyperledger/firefly-transaction-manager/internal/persistence"
26
25
"github.com/hyperledger/firefly-transaction-manager/mocks/persistencemocks"
27
26
"github.com/hyperledger/firefly-transaction-manager/pkg/apitypes"
28
27
"github.com/hyperledger/firefly-transaction-manager/pkg/ffcapi"
28
+ "github.com/hyperledger/firefly-transaction-manager/pkg/txhandler"
29
29
"github.com/stretchr/testify/assert"
30
30
"github.com/stretchr/testify/mock"
31
31
)
@@ -36,8 +36,8 @@ func TestDBMigrationOK(t *testing.T) {
36
36
mdb2 := persistencemocks .NewPersistence (t )
37
37
38
38
es := & apitypes.EventStream {ID : fftypes .NewUUID ()}
39
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
40
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , es .ID , paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, nil )
39
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
40
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , es .ID , paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, nil )
41
41
mdb2 .On ("GetStream" , mock .Anything , es .ID ).Return (nil , nil )
42
42
mdb2 .On ("WriteStream" , mock .Anything , es ).Return (nil )
43
43
@@ -47,8 +47,8 @@ func TestDBMigrationOK(t *testing.T) {
47
47
mdb2 .On ("WriteCheckpoint" , mock .Anything , cp ).Return (nil )
48
48
49
49
l := & apitypes.Listener {ID : fftypes .NewUUID ()}
50
- mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.Listener {l }, nil )
51
- mdb1 .On ("ListListenersByCreateTime" , mock .Anything , l .ID , paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.Listener {}, nil )
50
+ mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.Listener {l }, nil )
51
+ mdb1 .On ("ListListenersByCreateTime" , mock .Anything , l .ID , paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.Listener {}, nil )
52
52
mdb2 .On ("GetListener" , mock .Anything , l .ID ).Return (nil , nil )
53
53
mdb2 .On ("WriteListener" , mock .Anything , l ).Return (nil )
54
54
@@ -57,8 +57,8 @@ func TestDBMigrationOK(t *testing.T) {
57
57
Receipt : & ffcapi.TransactionReceiptResponse {BlockHash : fftypes .NewRandB32 ().String ()},
58
58
Confirmations : []* apitypes.Confirmation {{BlockHash : fftypes .NewRandB32 ().String ()}},
59
59
}
60
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
61
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , tx .ManagedTX , paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {}, nil )
60
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
61
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , tx .ManagedTX , paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {}, nil )
62
62
mdb1 .On ("GetTransactionByIDWithStatus" , mock .Anything , tx .ID , false ).Return (tx , nil )
63
63
mdb2 .On ("GetTransactionByID" , mock .Anything , tx .ID ).Return (nil , nil )
64
64
mdb2 .On ("InsertTransactionPreAssignedNonce" , mock .Anything , tx .ManagedTX ).Return (nil )
@@ -80,11 +80,11 @@ func TestDBMigrationRunFailTransactions(t *testing.T) {
80
80
mdb1 := persistencemocks .NewPersistence (t )
81
81
mdb2 := persistencemocks .NewPersistence (t )
82
82
83
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, nil )
83
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, nil )
84
84
85
- mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.Listener {}, nil )
85
+ mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.Listener {}, nil )
86
86
87
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {}, fmt .Errorf ("pop" ))
87
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {}, fmt .Errorf ("pop" ))
88
88
89
89
m := dbMigration {
90
90
source : mdb1 ,
@@ -101,9 +101,9 @@ func TestDBMigrationRunFailListeners(t *testing.T) {
101
101
mdb1 := persistencemocks .NewPersistence (t )
102
102
mdb2 := persistencemocks .NewPersistence (t )
103
103
104
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, nil )
104
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, nil )
105
105
106
- mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.Listener {}, fmt .Errorf ("pop" ))
106
+ mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.Listener {}, fmt .Errorf ("pop" ))
107
107
108
108
m := dbMigration {
109
109
source : mdb1 ,
@@ -120,7 +120,7 @@ func TestDBMigrationRunFailStreams(t *testing.T) {
120
120
mdb1 := persistencemocks .NewPersistence (t )
121
121
mdb2 := persistencemocks .NewPersistence (t )
122
122
123
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, fmt .Errorf ("pop" ))
123
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {}, fmt .Errorf ("pop" ))
124
124
125
125
m := dbMigration {
126
126
source : mdb1 ,
@@ -138,7 +138,7 @@ func TestMigrateEventsStreamsFailWrite(t *testing.T) {
138
138
mdb2 := persistencemocks .NewPersistence (t )
139
139
140
140
es := & apitypes.EventStream {ID : fftypes .NewUUID ()}
141
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
141
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
142
142
mdb2 .On ("GetStream" , mock .Anything , es .ID ).Return (nil , nil )
143
143
mdb2 .On ("WriteStream" , mock .Anything , es ).Return (fmt .Errorf ("pop" ))
144
144
@@ -158,7 +158,7 @@ func TestMigrateEventsStreamsFailCheckExist(t *testing.T) {
158
158
mdb2 := persistencemocks .NewPersistence (t )
159
159
160
160
es := & apitypes.EventStream {ID : fftypes .NewUUID ()}
161
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
161
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
162
162
mdb2 .On ("GetStream" , mock .Anything , es .ID ).Return (nil , fmt .Errorf ("pop" ))
163
163
164
164
m := dbMigration {
@@ -177,7 +177,7 @@ func TestMigrateEventsStreamsFailWriteCheckpoint(t *testing.T) {
177
177
mdb2 := persistencemocks .NewPersistence (t )
178
178
179
179
es := & apitypes.EventStream {ID : fftypes .NewUUID ()}
180
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
180
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
181
181
mdb2 .On ("GetStream" , mock .Anything , es .ID ).Return (nil , nil )
182
182
mdb2 .On ("WriteStream" , mock .Anything , es ).Return (nil )
183
183
@@ -202,7 +202,7 @@ func TestMigrateEventsStreamsFailCheckCheckpointExists(t *testing.T) {
202
202
mdb2 := persistencemocks .NewPersistence (t )
203
203
204
204
es := & apitypes.EventStream {ID : fftypes .NewUUID ()}
205
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
205
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
206
206
mdb2 .On ("GetStream" , mock .Anything , es .ID ).Return (nil , nil )
207
207
mdb2 .On ("WriteStream" , mock .Anything , es ).Return (nil )
208
208
@@ -226,7 +226,7 @@ func TestMigrateEventsStreamsFailGetCheckpoint(t *testing.T) {
226
226
mdb2 := persistencemocks .NewPersistence (t )
227
227
228
228
es := & apitypes.EventStream {ID : fftypes .NewUUID ()}
229
- mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
229
+ mdb1 .On ("ListStreamsByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.EventStream {es }, nil )
230
230
mdb2 .On ("GetStream" , mock .Anything , es .ID ).Return (nil , nil )
231
231
mdb2 .On ("WriteStream" , mock .Anything , es ).Return (nil )
232
232
@@ -248,7 +248,7 @@ func TestMigrateListenersFailWrite(t *testing.T) {
248
248
mdb2 := persistencemocks .NewPersistence (t )
249
249
250
250
l := & apitypes.Listener {ID : fftypes .NewUUID ()}
251
- mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.Listener {l }, nil )
251
+ mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.Listener {l }, nil )
252
252
mdb2 .On ("GetListener" , mock .Anything , l .ID ).Return (nil , nil )
253
253
mdb2 .On ("WriteListener" , mock .Anything , l ).Return (fmt .Errorf ("pop" ))
254
254
@@ -268,7 +268,7 @@ func TestMigrateListenersFailCheckExists(t *testing.T) {
268
268
mdb2 := persistencemocks .NewPersistence (t )
269
269
270
270
l := & apitypes.Listener {ID : fftypes .NewUUID ()}
271
- mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.Listener {l }, nil )
271
+ mdb1 .On ("ListListenersByCreateTime" , mock .Anything , (* fftypes .UUID )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.Listener {l }, nil )
272
272
mdb2 .On ("GetListener" , mock .Anything , l .ID ).Return (nil , fmt .Errorf ("pop" ))
273
273
274
274
m := dbMigration {
@@ -291,7 +291,7 @@ func TestMigrateTransactionsFailWriteConfirmations(t *testing.T) {
291
291
Receipt : & ffcapi.TransactionReceiptResponse {BlockHash : fftypes .NewRandB32 ().String ()},
292
292
Confirmations : []* apitypes.Confirmation {{BlockHash : fftypes .NewRandB32 ().String ()}},
293
293
}
294
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
294
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
295
295
mdb1 .On ("GetTransactionByIDWithStatus" , mock .Anything , tx .ID , false ).Return (tx , nil )
296
296
mdb2 .On ("GetTransactionByID" , mock .Anything , tx .ID ).Return (nil , nil )
297
297
mdb2 .On ("InsertTransactionPreAssignedNonce" , mock .Anything , tx .ManagedTX ).Return (nil )
@@ -318,7 +318,7 @@ func TestMigrateTransactionsFailWriteReceipt(t *testing.T) {
318
318
Receipt : & ffcapi.TransactionReceiptResponse {BlockHash : fftypes .NewRandB32 ().String ()},
319
319
Confirmations : []* apitypes.Confirmation {{BlockHash : fftypes .NewRandB32 ().String ()}},
320
320
}
321
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
321
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
322
322
mdb1 .On ("GetTransactionByIDWithStatus" , mock .Anything , tx .ID , false ).Return (tx , nil )
323
323
mdb2 .On ("GetTransactionByID" , mock .Anything , tx .ID ).Return (nil , nil )
324
324
mdb2 .On ("InsertTransactionPreAssignedNonce" , mock .Anything , tx .ManagedTX ).Return (nil )
@@ -344,7 +344,7 @@ func TestMigrateTransactionsFailWriteTx(t *testing.T) {
344
344
Receipt : & ffcapi.TransactionReceiptResponse {BlockHash : fftypes .NewRandB32 ().String ()},
345
345
Confirmations : []* apitypes.Confirmation {{BlockHash : fftypes .NewRandB32 ().String ()}},
346
346
}
347
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
347
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
348
348
mdb1 .On ("GetTransactionByIDWithStatus" , mock .Anything , tx .ID , false ).Return (tx , nil )
349
349
mdb2 .On ("GetTransactionByID" , mock .Anything , tx .ID ).Return (nil , nil )
350
350
mdb2 .On ("InsertTransactionPreAssignedNonce" , mock .Anything , tx .ManagedTX ).Return (fmt .Errorf ("pop" ))
@@ -369,7 +369,7 @@ func TestMigrateTransactionsFailCheckExists(t *testing.T) {
369
369
Receipt : & ffcapi.TransactionReceiptResponse {BlockHash : fftypes .NewRandB32 ().String ()},
370
370
Confirmations : []* apitypes.Confirmation {{BlockHash : fftypes .NewRandB32 ().String ()}},
371
371
}
372
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
372
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
373
373
mdb1 .On ("GetTransactionByIDWithStatus" , mock .Anything , tx .ID , false ).Return (tx , nil )
374
374
mdb2 .On ("GetTransactionByID" , mock .Anything , tx .ID ).Return (nil , fmt .Errorf ("pop" ))
375
375
@@ -393,7 +393,7 @@ func TestMigrateTransactionsFailGetDetail(t *testing.T) {
393
393
Receipt : & ffcapi.TransactionReceiptResponse {BlockHash : fftypes .NewRandB32 ().String ()},
394
394
Confirmations : []* apitypes.Confirmation {{BlockHash : fftypes .NewRandB32 ().String ()}},
395
395
}
396
- mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , persistence .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
396
+ mdb1 .On ("ListTransactionsByCreateTime" , mock .Anything , (* apitypes .ManagedTX )(nil ), paginationLimit , txhandler .SortDirectionAscending ).Return ([]* apitypes.ManagedTX {tx .ManagedTX }, nil )
397
397
mdb1 .On ("GetTransactionByIDWithStatus" , mock .Anything , tx .ID , false ).Return (tx , fmt .Errorf ("pop" ))
398
398
399
399
m := dbMigration {
0 commit comments