@@ -482,33 +482,46 @@ pub fn create_chan_between_nodes_with_value<'a, 'b, 'c, 'd>(node_a: &'a Node<'b,
482482 ( announcement, as_update, bs_update, channel_id, tx)
483483}
484484
485+ /// Gets an RAA and CS which were sent in response to a commitment update
486+ ///
487+ /// Should only be used directly when the `$node` is not actually a [`Node`].
488+ macro_rules! do_get_revoke_commit_msgs {
489+ ( $node: expr, $recipient: expr) => { {
490+ let events = $node. node. get_and_clear_pending_msg_events( ) ;
491+ assert_eq!( events. len( ) , 2 ) ;
492+ ( match events[ 0 ] {
493+ MessageSendEvent :: SendRevokeAndACK { ref node_id, ref msg } => {
494+ assert_eq!( node_id, $recipient) ;
495+ ( * msg) . clone( )
496+ } ,
497+ _ => panic!( "Unexpected event" ) ,
498+ } , match events[ 1 ] {
499+ MessageSendEvent :: UpdateHTLCs { ref node_id, ref updates } => {
500+ assert_eq!( node_id, $recipient) ;
501+ assert!( updates. update_add_htlcs. is_empty( ) ) ;
502+ assert!( updates. update_fulfill_htlcs. is_empty( ) ) ;
503+ assert!( updates. update_fail_htlcs. is_empty( ) ) ;
504+ assert!( updates. update_fail_malformed_htlcs. is_empty( ) ) ;
505+ assert!( updates. update_fee. is_none( ) ) ;
506+ updates. commitment_signed. clone( )
507+ } ,
508+ _ => panic!( "Unexpected event" ) ,
509+ } )
510+ } }
511+ }
512+
513+ /// Gets an RAA and CS which were sent in response to a commitment update
514+ pub fn get_revoke_commit_msgs ( node : & Node , recipient : & PublicKey ) -> ( msgs:: RevokeAndACK , msgs:: CommitmentSigned ) {
515+ do_get_revoke_commit_msgs ! ( node, recipient)
516+ }
517+
485518#[ macro_export]
486519/// Gets an RAA and CS which were sent in response to a commitment update
520+ ///
521+ /// Don't use this, use the identically-named function instead.
487522macro_rules! get_revoke_commit_msgs {
488523 ( $node: expr, $node_id: expr) => {
489- {
490- use $crate:: util:: events:: MessageSendEvent ;
491- let events = $node. node. get_and_clear_pending_msg_events( ) ;
492- assert_eq!( events. len( ) , 2 ) ;
493- ( match events[ 0 ] {
494- MessageSendEvent :: SendRevokeAndACK { ref node_id, ref msg } => {
495- assert_eq!( * node_id, $node_id) ;
496- ( * msg) . clone( )
497- } ,
498- _ => panic!( "Unexpected event" ) ,
499- } , match events[ 1 ] {
500- MessageSendEvent :: UpdateHTLCs { ref node_id, ref updates } => {
501- assert_eq!( * node_id, $node_id) ;
502- assert!( updates. update_add_htlcs. is_empty( ) ) ;
503- assert!( updates. update_fulfill_htlcs. is_empty( ) ) ;
504- assert!( updates. update_fail_htlcs. is_empty( ) ) ;
505- assert!( updates. update_fail_malformed_htlcs. is_empty( ) ) ;
506- assert!( updates. update_fee. is_none( ) ) ;
507- updates. commitment_signed. clone( )
508- } ,
509- _ => panic!( "Unexpected event" ) ,
510- } )
511- }
524+ $crate:: ln:: functional_test_utils:: get_revoke_commit_msgs( & $node, & $node_id)
512525 }
513526}
514527
0 commit comments