Skip to content

Commit 6bafb2a

Browse files
committed
Clean up handle_*(..., get_event_msg!(...)) cruft
Calling a macro within a method expression is somewhat annoying to read, and due to `rustfmt` expansion, calling a `handle_*` method with a `get_event_msg` macro as the second argument tends to be substantially less readable than just breaking the mess out into its own variable.
1 parent 113eb36 commit 6bafb2a

File tree

1 file changed

+26
-46
lines changed

1 file changed

+26
-46
lines changed

lightning/src/ln/functional_test_utils.rs

Lines changed: 26 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -1494,10 +1494,8 @@ pub fn sign_funding_transaction<'a, 'b, 'c>(
14941494
}
14951495
expect_channel_pending_event(&node_b, &node_a_id);
14961496

1497-
node_a.node.handle_funding_signed(
1498-
node_b_id,
1499-
&get_event_msg!(node_b, MessageSendEvent::SendFundingSigned, node_a_id),
1500-
);
1497+
let bs_funding_signed = get_event_msg!(node_b, MessageSendEvent::SendFundingSigned, node_a_id);
1498+
node_a.node.handle_funding_signed(node_b_id, &bs_funding_signed);
15011499
{
15021500
let mut added_monitors = node_a.chain_monitor.added_monitors.lock().unwrap();
15031501
assert_eq!(added_monitors.len(), 1);
@@ -1723,14 +1721,12 @@ pub fn create_chan_between_nodes_with_value_confirm_first<'a, 'b, 'c, 'd>(
17231721
) {
17241722
confirm_transaction_at(node_conf, tx, conf_height);
17251723
connect_blocks(node_conf, CHAN_CONFIRM_DEPTH - 1);
1726-
node_recv.node.handle_channel_ready(
1727-
node_conf.node.get_our_node_id(),
1728-
&get_event_msg!(
1729-
node_conf,
1730-
MessageSendEvent::SendChannelReady,
1731-
node_recv.node.get_our_node_id()
1732-
),
1724+
let channel_ready = get_event_msg!(
1725+
node_conf,
1726+
MessageSendEvent::SendChannelReady,
1727+
node_recv.node.get_our_node_id()
17331728
);
1729+
node_recv.node.handle_channel_ready(node_conf.node.get_our_node_id(), &channel_ready);
17341730
}
17351731

17361732
pub fn create_chan_between_nodes_with_value_confirm_second<'a, 'b, 'c>(
@@ -1904,14 +1900,9 @@ pub fn create_unannounced_chan_between_nodes_with_value<'a, 'b, 'c, 'd>(
19041900
tx.clone(),
19051901
)
19061902
.unwrap();
1907-
nodes[b].node.handle_funding_created(
1908-
node_a_id,
1909-
&get_event_msg!(
1910-
nodes[a],
1911-
MessageSendEvent::SendFundingCreated,
1912-
node_b_id
1913-
),
1914-
);
1903+
let as_funding_created =
1904+
get_event_msg!(nodes[a], MessageSendEvent::SendFundingCreated, node_b_id);
1905+
nodes[b].node.handle_funding_created(node_a_id, &as_funding_created);
19151906
check_added_monitors!(nodes[b], 1);
19161907

19171908
let cs_funding_signed = get_event_msg!(
@@ -1940,14 +1931,8 @@ pub fn create_unannounced_chan_between_nodes_with_value<'a, 'b, 'c, 'd>(
19401931
MessageSendEvent::SendChannelReady,
19411932
node_b_id
19421933
);
1943-
nodes[a].node.handle_channel_ready(
1944-
node_b_id,
1945-
&get_event_msg!(
1946-
nodes[b],
1947-
MessageSendEvent::SendChannelReady,
1948-
node_a_id
1949-
),
1950-
);
1934+
let bs_channel_ready = get_event_msg!(nodes[b], MessageSendEvent::SendChannelReady, node_a_id);
1935+
nodes[a].node.handle_channel_ready(node_b_id, &bs_channel_ready);
19511936
expect_channel_ready_event(&nodes[a], &node_b_id);
19521937
let as_update = get_event_msg!(
19531938
nodes[a],
@@ -2317,10 +2302,9 @@ pub fn close_channel<'a, 'b, 'c>(
23172302
let (tx_a, tx_b);
23182303

23192304
node_a.close_channel(channel_id, &node_b.get_our_node_id()).unwrap();
2320-
node_b.handle_shutdown(
2321-
node_a.get_our_node_id(),
2322-
&get_event_msg!(struct_a, MessageSendEvent::SendShutdown, node_b.get_our_node_id()),
2323-
);
2305+
let as_shutdown =
2306+
get_event_msg!(struct_a, MessageSendEvent::SendShutdown, node_b.get_our_node_id());
2307+
node_b.handle_shutdown(node_a.get_our_node_id(), &as_shutdown);
23242308

23252309
let events_1 = node_b.get_and_clear_pending_msg_events();
23262310
assert!(events_1.len() >= 1);
@@ -2350,14 +2334,12 @@ pub fn close_channel<'a, 'b, 'c>(
23502334
assert!(node_a.get_and_clear_pending_msg_events().is_empty());
23512335
node_a.handle_closing_signed(node_b.get_our_node_id(), &closing_signed_b.unwrap());
23522336

2353-
node_b.handle_closing_signed(
2354-
node_a.get_our_node_id(),
2355-
&get_event_msg!(
2356-
struct_a,
2357-
MessageSendEvent::SendClosingSigned,
2358-
node_b.get_our_node_id()
2359-
),
2337+
let as_closing_signed = get_event_msg!(
2338+
struct_a,
2339+
MessageSendEvent::SendClosingSigned,
2340+
node_b.get_our_node_id()
23602341
);
2342+
node_b.handle_closing_signed(node_a.get_our_node_id(), &as_closing_signed);
23612343
assert_eq!(broadcaster_b.txn_broadcasted.lock().unwrap().len(), 1);
23622344
tx_b = broadcaster_b.txn_broadcasted.lock().unwrap().remove(0);
23632345
let (bs_update, closing_signed_b) =
@@ -2372,16 +2354,14 @@ pub fn close_channel<'a, 'b, 'c>(
23722354
} else {
23732355
let closing_signed_a =
23742356
get_event_msg!(struct_a, MessageSendEvent::SendClosingSigned, node_b.get_our_node_id());
2375-
23762357
node_b.handle_closing_signed(node_a.get_our_node_id(), &closing_signed_a);
2377-
node_a.handle_closing_signed(
2378-
node_b.get_our_node_id(),
2379-
&get_event_msg!(
2380-
struct_b,
2381-
MessageSendEvent::SendClosingSigned,
2382-
node_a.get_our_node_id()
2383-
),
2358+
2359+
let closing_signed_b =get_event_msg!(
2360+
struct_b,
2361+
MessageSendEvent::SendClosingSigned,
2362+
node_a.get_our_node_id()
23842363
);
2364+
node_a.handle_closing_signed(node_b.get_our_node_id(), &closing_signed_b);
23852365

23862366
assert_eq!(broadcaster_a.txn_broadcasted.lock().unwrap().len(), 1);
23872367
tx_a = broadcaster_a.txn_broadcasted.lock().unwrap().remove(0);

0 commit comments

Comments
 (0)