@@ -482,33 +482,46 @@ pub fn create_chan_between_nodes_with_value<'a, 'b, 'c, 'd>(node_a: &'a Node<'b,
482
482
( announcement, as_update, bs_update, channel_id, tx)
483
483
}
484
484
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
+
485
518
#[ macro_export]
486
519
/// 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.
487
522
macro_rules! get_revoke_commit_msgs {
488
523
( $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)
512
525
}
513
526
}
514
527
0 commit comments