11package handlers_test
22
33import (
4+ "context"
45 "fmt"
56 "time"
67
@@ -18,13 +19,13 @@ type MockDatabase struct {
1819 msgCall map [string ]* database.MsgCall
1920 msgAddPackage map [string ]* database.MsgAddPackage
2021 msgRun map [string ]* database.MsgRun
21- msgTypes map [string ]string
22+ msgTypes map [string ][] string
2223
2324 shouldError bool
2425 errorMsg string
2526}
2627
27- func (m * MockDatabase ) GetBlock (height uint64 , chainName string ) (* database.BlockData , error ) {
28+ func (m * MockDatabase ) GetBlock (ctx context. Context , height uint64 , chainName string ) (* database.BlockData , error ) {
2829 if m .shouldError {
2930 return nil , fmt .Errorf ("%s" , m .errorMsg )
3031 }
@@ -35,7 +36,7 @@ func (m *MockDatabase) GetBlock(height uint64, chainName string) (*database.Bloc
3536 return block , nil
3637}
3738
38- func (m * MockDatabase ) GetFromToBlocks (fromHeight uint64 , toHeight uint64 , chainName string ) ([]* database.BlockData , error ) {
39+ func (m * MockDatabase ) GetFromToBlocks (ctx context. Context , fromHeight uint64 , toHeight uint64 , chainName string ) ([]* database.BlockData , error ) {
3940 if m .shouldError {
4041 return nil , fmt .Errorf ("%s" , m .errorMsg )
4142 }
@@ -50,7 +51,7 @@ func (m *MockDatabase) GetFromToBlocks(fromHeight uint64, toHeight uint64, chain
5051 return result , nil
5152}
5253
53- func (m * MockDatabase ) GetAllBlockSigners (chainName string , blockHeight uint64 ) (* database.BlockSigners , error ) {
54+ func (m * MockDatabase ) GetAllBlockSigners (ctx context. Context , chainName string , blockHeight uint64 ) (* database.BlockSigners , error ) {
5455 if m .shouldError {
5556 return nil , fmt .Errorf ("%s" , m .errorMsg )
5657 }
@@ -61,7 +62,7 @@ func (m *MockDatabase) GetAllBlockSigners(chainName string, blockHeight uint64)
6162 return blockSigners , nil
6263}
6364
64- func (m * MockDatabase ) GetTransaction (txHash string , chainName string ) (* database.Transaction , error ) {
65+ func (m * MockDatabase ) GetTransaction (ctx context. Context , txHash string , chainName string ) (* database.Transaction , error ) {
6566 if m .shouldError {
6667 return nil , fmt .Errorf ("%s" , m .errorMsg )
6768 }
@@ -72,7 +73,7 @@ func (m *MockDatabase) GetTransaction(txHash string, chainName string) (*databas
7273 return transaction , nil
7374}
7475
75- func (m * MockDatabase ) GetAddressTxs (address string , chainName string , fromTimestamp time.Time , toTimestamp time.Time ) (* []database.AddressTx , error ) {
76+ func (m * MockDatabase ) GetAddressTxs (ctx context. Context , address string , chainName string , fromTimestamp time.Time , toTimestamp time.Time ) (* []database.AddressTx , error ) {
7677 if m .shouldError {
7778 return nil , fmt .Errorf ("%s" , m .errorMsg )
7879 }
@@ -83,7 +84,7 @@ func (m *MockDatabase) GetAddressTxs(address string, chainName string, fromTimes
8384 return addressTxs , nil
8485}
8586
86- func (m * MockDatabase ) GetLatestBlock (chainName string ) (* database.BlockData , error ) {
87+ func (m * MockDatabase ) GetLatestBlock (ctx context. Context , chainName string ) (* database.BlockData , error ) {
8788 if m .shouldError {
8889 return nil , fmt .Errorf ("%s" , m .errorMsg )
8990 }
@@ -93,7 +94,7 @@ func (m *MockDatabase) GetLatestBlock(chainName string) (*database.BlockData, er
9394 return m .latestBlock , nil
9495}
9596
96- func (m * MockDatabase ) GetLastXBlocks (chainName string , x uint64 ) ([]* database.BlockData , error ) {
97+ func (m * MockDatabase ) GetLastXBlocks (ctx context. Context , chainName string , x uint64 ) ([]* database.BlockData , error ) {
9798 if m .shouldError {
9899 return nil , fmt .Errorf ("%s" , m .errorMsg )
99100 }
@@ -104,7 +105,7 @@ func (m *MockDatabase) GetLastXBlocks(chainName string, x uint64) ([]*database.B
104105 return blocks , nil
105106}
106107
107- func (m * MockDatabase ) GetLastXTransactions (chainName string , x uint64 ) ([]* database.Transaction , error ) {
108+ func (m * MockDatabase ) GetLastXTransactions (ctx context. Context , chainName string , x uint64 ) ([]* database.Transaction , error ) {
108109 if m .shouldError {
109110 return nil , fmt .Errorf ("%s" , m .errorMsg )
110111 }
@@ -115,57 +116,57 @@ func (m *MockDatabase) GetLastXTransactions(chainName string, x uint64) ([]*data
115116 return transactions , nil
116117}
117118
118- func (m * MockDatabase ) GetMsgType ( txHash string , chainName string ) (string , error ) {
119+ func (m * MockDatabase ) GetMsgTypes ( ctx context. Context , txHash string , chainName string ) ([] string , error ) {
119120 if m .shouldError {
120- return "" , fmt .Errorf ("%s" , m .errorMsg )
121+ return nil , fmt .Errorf ("%s" , m .errorMsg )
121122 }
122- msgType , ok := m .msgTypes [txHash ]
123+ msgTypes , ok := m .msgTypes [txHash ]
123124 if ! ok {
124- return "" , fmt .Errorf ("message type not found" )
125+ return nil , fmt .Errorf ("message type not found" )
125126 }
126- return msgType , nil
127+ return msgTypes , nil
127128}
128129
129- func (m * MockDatabase ) GetBankSend (txHash string , chainName string ) (* database.BankSend , error ) {
130+ func (m * MockDatabase ) GetBankSend (ctx context. Context , txHash string , chainName string ) ([] * database.BankSend , error ) {
130131 if m .shouldError {
131132 return nil , fmt .Errorf ("%s" , m .errorMsg )
132133 }
133134 bankSend , ok := m .bankSend [txHash ]
134135 if ! ok {
135136 return nil , fmt .Errorf ("bank send not found" )
136137 }
137- return bankSend , nil
138+ return [] * database. BankSend { bankSend } , nil
138139}
139140
140- func (m * MockDatabase ) GetMsgCall (txHash string , chainName string ) (* database.MsgCall , error ) {
141+ func (m * MockDatabase ) GetMsgCall (ctx context. Context , txHash string , chainName string ) ([] * database.MsgCall , error ) {
141142 if m .shouldError {
142143 return nil , fmt .Errorf ("%s" , m .errorMsg )
143144 }
144145 msgCall , ok := m .msgCall [txHash ]
145146 if ! ok {
146147 return nil , fmt .Errorf ("message call not found" )
147148 }
148- return msgCall , nil
149+ return [] * database. MsgCall { msgCall } , nil
149150}
150151
151- func (m * MockDatabase ) GetMsgAddPackage (txHash string , chainName string ) (* database.MsgAddPackage , error ) {
152+ func (m * MockDatabase ) GetMsgAddPackage (ctx context. Context , txHash string , chainName string ) ([] * database.MsgAddPackage , error ) {
152153 if m .shouldError {
153154 return nil , fmt .Errorf ("%s" , m .errorMsg )
154155 }
155156 msgAddPackage , ok := m .msgAddPackage [txHash ]
156157 if ! ok {
157158 return nil , fmt .Errorf ("message add package not found" )
158159 }
159- return msgAddPackage , nil
160+ return [] * database. MsgAddPackage { msgAddPackage } , nil
160161}
161162
162- func (m * MockDatabase ) GetMsgRun (txHash string , chainName string ) (* database.MsgRun , error ) {
163+ func (m * MockDatabase ) GetMsgRun (ctx context. Context , txHash string , chainName string ) ([] * database.MsgRun , error ) {
163164 if m .shouldError {
164165 return nil , fmt .Errorf ("%s" , m .errorMsg )
165166 }
166167 msgRun , ok := m .msgRun [txHash ]
167168 if ! ok {
168169 return nil , fmt .Errorf ("message run not found" )
169170 }
170- return msgRun , nil
171+ return [] * database. MsgRun { msgRun } , nil
171172}
0 commit comments