@@ -72,75 +72,75 @@ func TestExecuteRegisterPegIn(t *testing.T) {
7272 }
7373 t .Run ("should execute RegisterPegIn successfully" , func (t * testing.T ) {
7474 rpc := new (mocks.BtcRpcMock )
75- lbc := new (mocks.LiquidityBridgeContractMock )
75+ peginContract := new (mocks.PeginContractMock )
7676 rpc .On ("GetPartialMerkleTree" , parsedInput .BtcTxHash ).Return (pmt , nil ).Once ()
7777 rpc .On ("GetRawTransaction" , parsedInput .BtcTxHash ).Return (rawTx , nil ).Once ()
7878 rpc .On ("GetTransactionBlockInfo" , parsedInput .BtcTxHash ).Return (blockInfo , nil ).Once ()
79- lbc .On ("RegisterPegin" , blockchain.RegisterPeginParams {
79+ peginContract .On ("RegisterPegin" , blockchain.RegisterPeginParams {
8080 QuoteSignature : parsedInput .Signature , BitcoinRawTransaction : rawTx ,
8181 PartialMerkleTree : pmt , BlockHeight : blockHeight , Quote : parsedInput .Quote ,
8282 }).Return (expectedReceipt , nil ).Once ()
8383
84- result , err := ExecuteRegisterPegIn (rpc , lbc , parsedInput )
84+ result , err := ExecuteRegisterPegIn (rpc , peginContract , parsedInput )
8585 require .NoError (t , err )
8686 assert .Equal (t , expectedReceipt , result )
8787 rpc .AssertExpectations (t )
88- lbc .AssertExpectations (t )
88+ peginContract .AssertExpectations (t )
8989 })
9090
9191 t .Run ("should return error if GetPartialMerkleTree fails" , func (t * testing.T ) {
9292 rpc := new (mocks.BtcRpcMock )
93- lbc := new (mocks.LiquidityBridgeContractMock )
93+ peginContract := new (mocks.PeginContractMock )
9494 rpc .On ("GetPartialMerkleTree" , parsedInput .BtcTxHash ).Return ([]byte {}, assert .AnError ).Once ()
9595
96- result , err := ExecuteRegisterPegIn (rpc , lbc , parsedInput )
96+ result , err := ExecuteRegisterPegIn (rpc , peginContract , parsedInput )
9797 require .Error (t , err )
9898 assert .Equal (t , blockchain.TransactionReceipt {}, result )
9999 rpc .AssertExpectations (t )
100- lbc .AssertNotCalled (t , "RegisterPegin" )
100+ peginContract .AssertNotCalled (t , "RegisterPegin" )
101101 })
102102
103103 t .Run ("should return error if GetRawTransaction fails" , func (t * testing.T ) {
104104 rpc := new (mocks.BtcRpcMock )
105- lbc := new (mocks.LiquidityBridgeContractMock )
105+ peginContract := new (mocks.PeginContractMock )
106106 rpc .On ("GetPartialMerkleTree" , parsedInput .BtcTxHash ).Return (pmt , nil ).Once ()
107107 rpc .On ("GetRawTransaction" , parsedInput .BtcTxHash ).Return ([]byte {}, assert .AnError ).Once ()
108108
109- result , err := ExecuteRegisterPegIn (rpc , lbc , parsedInput )
109+ result , err := ExecuteRegisterPegIn (rpc , peginContract , parsedInput )
110110 require .Error (t , err )
111111 assert .Equal (t , blockchain.TransactionReceipt {}, result )
112112 rpc .AssertExpectations (t )
113- lbc .AssertNotCalled (t , "RegisterPegin" )
113+ peginContract .AssertNotCalled (t , "RegisterPegin" )
114114 })
115115
116116 t .Run ("should return error if GetTransactionBlockInfo fails" , func (t * testing.T ) {
117117 rpc := new (mocks.BtcRpcMock )
118- lbc := new (mocks.LiquidityBridgeContractMock )
118+ peginContract := new (mocks.PeginContractMock )
119119 rpc .On ("GetPartialMerkleTree" , parsedInput .BtcTxHash ).Return (pmt , nil ).Once ()
120120 rpc .On ("GetRawTransaction" , parsedInput .BtcTxHash ).Return (rawTx , nil ).Once ()
121121 rpc .On ("GetTransactionBlockInfo" , parsedInput .BtcTxHash ).Return (blockchain.BitcoinBlockInformation {}, assert .AnError ).Once ()
122122
123- result , err := ExecuteRegisterPegIn (rpc , lbc , parsedInput )
123+ result , err := ExecuteRegisterPegIn (rpc , peginContract , parsedInput )
124124
125125 require .Error (t , err )
126126 assert .Equal (t , blockchain.TransactionReceipt {}, result )
127127 rpc .AssertExpectations (t )
128- lbc .AssertNotCalled (t , "RegisterPegin" )
128+ peginContract .AssertNotCalled (t , "RegisterPegin" )
129129 })
130130
131131 t .Run ("should return error if RegisterPegin fails" , func (t * testing.T ) {
132132 rpc := new (mocks.BtcRpcMock )
133- lbc := new (mocks.LiquidityBridgeContractMock )
133+ peginContract := new (mocks.PeginContractMock )
134134 rpc .On ("GetPartialMerkleTree" , parsedInput .BtcTxHash ).Return (pmt , nil ).Once ()
135135 rpc .On ("GetRawTransaction" , parsedInput .BtcTxHash ).Return (rawTx , nil ).Once ()
136136 rpc .On ("GetTransactionBlockInfo" , parsedInput .BtcTxHash ).Return (blockInfo , nil ).Once ()
137- lbc .On ("RegisterPegin" , mock .Anything ).Return (blockchain.TransactionReceipt {}, assert .AnError ).Once ()
137+ peginContract .On ("RegisterPegin" , mock .Anything ).Return (blockchain.TransactionReceipt {}, assert .AnError ).Once ()
138138
139- result , err := ExecuteRegisterPegIn (rpc , lbc , parsedInput )
139+ result , err := ExecuteRegisterPegIn (rpc , peginContract , parsedInput )
140140 require .Error (t , err )
141141 assert .Equal (t , blockchain.TransactionReceipt {}, result )
142142 rpc .AssertExpectations (t )
143- lbc .AssertExpectations (t )
143+ peginContract .AssertExpectations (t )
144144 })
145145}
146146
@@ -163,7 +163,7 @@ func TestRegisterPegInScriptInput_ToEnv(t *testing.T) {
163163 assert .Equal (t , "btcPassword" , env .Btc .Password )
164164 // we assert the rsk defaults to ensure the method called its parent method
165165 assert .Equal (t , uint64 (33 ), env .Rsk .ChainId )
166- assert .Equal (t , "0x8901a2bbf639bfd21a97004ba4d7ae2bd00b8da8" , env .Rsk .LbcAddress )
166+ assert .Equal (t , "0x8901a2bbf639bfd21a97004ba4d7ae2bd00b8da8" , env .Rsk .PeginContractAddress )
167167 assert .Equal (t , "0x0000000000000000000000000000000001000006" , env .Rsk .BridgeAddress )
168168 assert .Equal (t , 0 , env .Rsk .AccountNumber )
169169 })
0 commit comments