@@ -10,7 +10,7 @@ use tokio::sync::RwLock;
1010use crate :: adapters:: escrow_adapter:: EscrowAdapter ;
1111
1212pub struct EscrowAdapterMock {
13- gateway_escrow_storage : Arc < RwLock < HashMap < Address , u128 > > > ,
13+ sender_escrow_storage : Arc < RwLock < HashMap < Address , u128 > > > ,
1414}
1515
1616use thiserror:: Error ;
@@ -21,43 +21,43 @@ pub enum AdpaterErrorMock {
2121}
2222
2323impl EscrowAdapterMock {
24- pub fn new ( gateway_escrow_storage : Arc < RwLock < HashMap < Address , u128 > > > ) -> Self {
24+ pub fn new ( sender_escrow_storage : Arc < RwLock < HashMap < Address , u128 > > > ) -> Self {
2525 EscrowAdapterMock {
26- gateway_escrow_storage ,
26+ sender_escrow_storage ,
2727 }
2828 }
29- pub async fn escrow ( & self , gateway_id : Address ) -> Result < u128 , AdpaterErrorMock > {
30- let gateway_escrow_storage = self . gateway_escrow_storage . read ( ) . await ;
31- if let Some ( escrow) = gateway_escrow_storage . get ( & gateway_id ) {
29+ pub async fn escrow ( & self , sender_id : Address ) -> Result < u128 , AdpaterErrorMock > {
30+ let sender_escrow_storage = self . sender_escrow_storage . read ( ) . await ;
31+ if let Some ( escrow) = sender_escrow_storage . get ( & sender_id ) {
3232 return Ok ( * escrow) ;
3333 }
3434 Err ( AdpaterErrorMock :: AdapterError {
35- error : "No escrow exists for provided gateway ID." . to_owned ( ) ,
35+ error : "No escrow exists for provided sender ID." . to_owned ( ) ,
3636 } )
3737 }
3838
39- pub async fn increase_escrow ( & mut self , gateway_id : Address , value : u128 ) {
40- let mut gateway_escrow_storage = self . gateway_escrow_storage . write ( ) . await ;
39+ pub async fn increase_escrow ( & mut self , sender_id : Address , value : u128 ) {
40+ let mut sender_escrow_storage = self . sender_escrow_storage . write ( ) . await ;
4141
42- if let Some ( current_value) = gateway_escrow_storage . get ( & gateway_id ) {
43- let mut gateway_escrow_storage = self . gateway_escrow_storage . write ( ) . await ;
44- gateway_escrow_storage . insert ( gateway_id , current_value + value) ;
42+ if let Some ( current_value) = sender_escrow_storage . get ( & sender_id ) {
43+ let mut sender_escrow_storage = self . sender_escrow_storage . write ( ) . await ;
44+ sender_escrow_storage . insert ( sender_id , current_value + value) ;
4545 } else {
46- gateway_escrow_storage . insert ( gateway_id , value) ;
46+ sender_escrow_storage . insert ( sender_id , value) ;
4747 }
4848 }
4949
5050 pub async fn reduce_escrow (
5151 & self ,
52- gateway_id : Address ,
52+ sender_id : Address ,
5353 value : u128 ,
5454 ) -> Result < ( ) , AdpaterErrorMock > {
55- let mut gateway_escrow_storage = self . gateway_escrow_storage . write ( ) . await ;
55+ let mut sender_escrow_storage = self . sender_escrow_storage . write ( ) . await ;
5656
57- if let Some ( current_value) = gateway_escrow_storage . get ( & gateway_id ) {
57+ if let Some ( current_value) = sender_escrow_storage . get ( & sender_id ) {
5858 let checked_new_value = current_value. checked_sub ( value) ;
5959 if let Some ( new_value) = checked_new_value {
60- gateway_escrow_storage . insert ( gateway_id , new_value) ;
60+ sender_escrow_storage . insert ( sender_id , new_value) ;
6161 return Ok ( ( ) ) ;
6262 }
6363 }
@@ -70,14 +70,14 @@ impl EscrowAdapterMock {
7070#[ async_trait]
7171impl EscrowAdapter for EscrowAdapterMock {
7272 type AdapterError = AdpaterErrorMock ;
73- async fn get_available_escrow ( & self , gateway_id : Address ) -> Result < u128 , Self :: AdapterError > {
74- self . escrow ( gateway_id ) . await
73+ async fn get_available_escrow ( & self , sender_id : Address ) -> Result < u128 , Self :: AdapterError > {
74+ self . escrow ( sender_id ) . await
7575 }
7676 async fn subtract_escrow (
7777 & self ,
78- gateway_id : Address ,
78+ sender_id : Address ,
7979 value : u128 ,
8080 ) -> Result < ( ) , Self :: AdapterError > {
81- self . reduce_escrow ( gateway_id , value) . await
81+ self . reduce_escrow ( sender_id , value) . await
8282 }
8383}
0 commit comments