Skip to content

Commit 35bb0f4

Browse files
committed
Replace get_err_msg macro with a function
The `get_err_msg!()` macro has no reason to be a macro so here we move its logic to a function and leave the macro in place to avoid touching every line of code in the tests. This reduces the `--profile=test --lib` `Zpretty=expanded` code size from 322,183 LoC to 321,985 LoC.
1 parent 90b2f5e commit 35bb0f4

File tree

3 files changed

+19
-24
lines changed

3 files changed

+19
-24
lines changed

lightning/src/ln/channelmanager.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8469,7 +8469,7 @@ mod tests {
84698469
// A MAX_UNFUNDED_CHANS_PER_PEER + 1 channel will be summarily rejected
84708470
open_channel_msg.temporary_channel_id = nodes[0].keys_manager.get_secure_random_bytes();
84718471
nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
8472-
assert_eq!(get_err_msg!(nodes[1], nodes[0].node.get_our_node_id()).channel_id,
8472+
assert_eq!(get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id()).channel_id,
84738473
open_channel_msg.temporary_channel_id);
84748474

84758475
// Further, because all of our channels with nodes[0] are inbound, and none of them funded,
@@ -8516,7 +8516,7 @@ mod tests {
85168516
open_channel_msg.temporary_channel_id = nodes[0].keys_manager.get_secure_random_bytes();
85178517
}
85188518
nodes[1].node.handle_open_channel(&last_random_pk, &open_channel_msg);
8519-
assert_eq!(get_err_msg!(nodes[1], last_random_pk).channel_id,
8519+
assert_eq!(get_err_msg(&nodes[1], &last_random_pk).channel_id,
85208520
open_channel_msg.temporary_channel_id);
85218521

85228522
// Of course, however, outbound channels are always allowed
@@ -8558,7 +8558,7 @@ mod tests {
85588558
// Once we have MAX_UNFUNDED_CHANS_PER_PEER unfunded channels, new inbound channels will be
85598559
// rejected.
85608560
nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
8561-
assert_eq!(get_err_msg!(nodes[1], nodes[0].node.get_our_node_id()).channel_id,
8561+
assert_eq!(get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id()).channel_id,
85628562
open_channel_msg.temporary_channel_id);
85638563

85648564
// but we can still open an outbound channel.
@@ -8567,7 +8567,7 @@ mod tests {
85678567

85688568
// but even with such an outbound channel, additional inbound channels will still fail.
85698569
nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
8570-
assert_eq!(get_err_msg!(nodes[1], nodes[0].node.get_our_node_id()).channel_id,
8570+
assert_eq!(get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id()).channel_id,
85718571
open_channel_msg.temporary_channel_id);
85728572
}
85738573

@@ -8623,7 +8623,7 @@ mod tests {
86238623
}
86248624
_ => panic!("Unexpected event"),
86258625
}
8626-
assert_eq!(get_err_msg!(nodes[1], last_random_pk).channel_id,
8626+
assert_eq!(get_err_msg(&nodes[1], &last_random_pk).channel_id,
86278627
open_channel_msg.temporary_channel_id);
86288628

86298629
// ...however if we accept the same channel 0conf it should work just fine.
@@ -8665,7 +8665,7 @@ mod tests {
86658665
_ => panic!("Unexpected event"),
86668666
}
86678667

8668-
let error_msg = get_err_msg!(nodes[1], nodes[0].node.get_our_node_id());
8668+
let error_msg = get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id());
86698669
nodes[0].node.handle_error(&nodes[1].node.get_our_node_id(), &error_msg);
86708670

86718671
let open_channel_msg = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());

lightning/src/ln/functional_test_utils.rs

Lines changed: 11 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -544,22 +544,17 @@ macro_rules! get_event_msg {
544544
}
545545

546546
/// Get an error message from the pending events queue.
547-
#[macro_export]
548-
macro_rules! get_err_msg {
549-
($node: expr, $node_id: expr) => {
550-
{
551-
let events = $node.node.get_and_clear_pending_msg_events();
552-
assert_eq!(events.len(), 1);
553-
match events[0] {
554-
$crate::util::events::MessageSendEvent::HandleError {
555-
action: $crate::ln::msgs::ErrorAction::SendErrorMessage { ref msg }, ref node_id
556-
} => {
557-
assert_eq!(*node_id, $node_id);
558-
(*msg).clone()
559-
},
560-
_ => panic!("Unexpected event"),
561-
}
562-
}
547+
pub fn get_err_msg(node: &Node, recipient: &PublicKey) -> msgs::ErrorMessage {
548+
let events = node.node.get_and_clear_pending_msg_events();
549+
assert_eq!(events.len(), 1);
550+
match events[0] {
551+
MessageSendEvent::HandleError {
552+
action: msgs::ErrorAction::SendErrorMessage { ref msg }, ref node_id
553+
} => {
554+
assert_eq!(node_id, recipient);
555+
(*msg).clone()
556+
},
557+
_ => panic!("Unexpected event"),
563558
}
564559
}
565560

lightning/src/ln/priv_short_conf_tests.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,7 @@ fn test_scid_privacy_on_pub_channel() {
278278
assert_eq!(open_channel.channel_flags & 1, 1); // The `announce_channel` bit is set.
279279

280280
nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel);
281-
let err = get_err_msg!(nodes[1], nodes[0].node.get_our_node_id());
281+
let err = get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id());
282282
assert_eq!(err.data, "SCID Alias/Privacy Channel Type cannot be set on a public channel");
283283
}
284284

@@ -721,7 +721,7 @@ fn test_0conf_close_no_early_chan_update() {
721721
nodes[0].node.force_close_all_channels_broadcasting_latest_txn();
722722
check_added_monitors!(nodes[0], 1);
723723
check_closed_event!(&nodes[0], 1, ClosureReason::HolderForceClosed);
724-
let _ = get_err_msg!(nodes[0], nodes[1].node.get_our_node_id());
724+
let _ = get_err_msg(&nodes[0], &nodes[1].node.get_our_node_id());
725725
}
726726

727727
#[test]

0 commit comments

Comments
 (0)