@@ -190,12 +190,18 @@ mod tests {
190190 assert_eq ! ( outgoing_message_data. nonce, 0 ) ;
191191 assert_eq ! ( outgoing_message_data. sender, from. pubkey( ) ) ;
192192
193+ let bridge = svm. get_account ( & bridge_pda) . unwrap ( ) ;
194+ let bridge = Bridge :: try_deserialize ( & mut & bridge. data [ ..] ) . unwrap ( ) ;
195+
193196 // Verify the message content
194197 match outgoing_message_data. message {
195198 crate :: solana_to_base:: Message :: Transfer ( transfer) => {
196199 assert_eq ! ( transfer. to, to) ;
197200 assert_eq ! ( transfer. local_token, NATIVE_SOL_PUBKEY ) ;
198- assert_eq ! ( transfer. remote_token, [ 0u8 ; 20 ] ) ;
201+ assert_eq ! (
202+ transfer. remote_token,
203+ bridge. protocol_config. sol_base_address
204+ ) ;
199205 assert_eq ! ( transfer. amount, amount) ;
200206 assert ! ( transfer. call. is_none( ) ) ;
201207 }
@@ -287,12 +293,18 @@ mod tests {
287293 let outgoing_message_data =
288294 OutgoingMessage :: try_deserialize ( & mut & outgoing_message_account. data [ ..] ) . unwrap ( ) ;
289295
296+ let bridge = svm. get_account ( & bridge_pda) . unwrap ( ) ;
297+ let bridge = Bridge :: try_deserialize ( & mut & bridge. data [ ..] ) . unwrap ( ) ;
298+
290299 // Verify the message content including call
291300 match outgoing_message_data. message {
292301 crate :: solana_to_base:: Message :: Transfer ( transfer) => {
293302 assert_eq ! ( transfer. to, to) ;
294303 assert_eq ! ( transfer. local_token, NATIVE_SOL_PUBKEY ) ;
295- assert_eq ! ( transfer. remote_token, [ 0u8 ; 20 ] ) ;
304+ assert_eq ! (
305+ transfer. remote_token,
306+ bridge. protocol_config. sol_base_address
307+ ) ;
296308 assert_eq ! ( transfer. amount, amount) ;
297309
298310 let transfer_call = transfer. call . expect ( "Expected call to be present" ) ;
0 commit comments