Skip to content

Commit 3a862a3

Browse files
committed
Consistently use node_*_id variables in chainmonitor tests
In functional tests we're slowly moving towards using `node_*_id` variables to access node ids rather than writing out `nodes[*].node.get_our_node_id()` each time. This somewhat improves readability by reducing cognitive load, so is generally nice to do as we go through rustfmt-induced changes anyway. Here we make the swap in `chainmonitor.rs` functional tests.
1 parent ab5185c commit 3a862a3

File tree

1 file changed

+27
-50
lines changed

1 file changed

+27
-50
lines changed

lightning/src/chain/chainmonitor.rs

Lines changed: 27 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,6 +1280,9 @@ mod tests {
12801280
let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
12811281
let channel_id = create_announced_chan_between_nodes(&nodes, 0, 1).2;
12821282

1283+
let node_a_id = nodes[0].node.get_our_node_id();
1284+
let node_b_id = nodes[1].node.get_our_node_id();
1285+
12831286
// Route two payments to be claimed at the same time.
12841287
let (payment_preimage_1, payment_hash_1, ..) =
12851288
route_payment(&nodes[0], &[&nodes[1]], 1_000_000);
@@ -1352,63 +1355,40 @@ mod tests {
13521355
// Now manually walk the commitment signed dance - because we claimed two payments
13531356
// back-to-back it doesn't fit into the neat walk commitment_signed_dance does.
13541357

1355-
let updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
1356-
nodes[0].node.handle_update_fulfill_htlc(
1357-
nodes[1].node.get_our_node_id(),
1358-
&updates.update_fulfill_htlcs[0],
1359-
);
1358+
let updates = get_htlc_update_msgs!(nodes[1], node_a_id);
1359+
nodes[0].node.handle_update_fulfill_htlc(node_b_id, &updates.update_fulfill_htlcs[0]);
13601360
expect_payment_sent(&nodes[0], payment_preimage_1, None, false, false);
1361-
nodes[0].node.handle_commitment_signed_batch_test(
1362-
nodes[1].node.get_our_node_id(),
1363-
&updates.commitment_signed,
1364-
);
1361+
nodes[0].node.handle_commitment_signed_batch_test(node_b_id, &updates.commitment_signed);
13651362
check_added_monitors!(nodes[0], 1);
1366-
let (as_first_raa, as_first_update) =
1367-
get_revoke_commit_msgs!(nodes[0], nodes[1].node.get_our_node_id());
1363+
let (as_first_raa, as_first_update) = get_revoke_commit_msgs!(nodes[0], node_b_id);
13681364

1369-
nodes[1].node.handle_revoke_and_ack(nodes[0].node.get_our_node_id(), &as_first_raa);
1365+
nodes[1].node.handle_revoke_and_ack(node_a_id, &as_first_raa);
13701366
check_added_monitors!(nodes[1], 1);
1371-
let bs_second_updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
1372-
nodes[1]
1373-
.node
1374-
.handle_commitment_signed_batch_test(nodes[0].node.get_our_node_id(), &as_first_update);
1367+
let bs_second_updates = get_htlc_update_msgs!(nodes[1], node_a_id);
1368+
nodes[1].node.handle_commitment_signed_batch_test(node_a_id, &as_first_update);
13751369
check_added_monitors!(nodes[1], 1);
1376-
let bs_first_raa = get_event_msg!(
1377-
nodes[1],
1378-
MessageSendEvent::SendRevokeAndACK,
1379-
nodes[0].node.get_our_node_id()
1380-
);
1370+
let bs_first_raa = get_event_msg!(nodes[1], MessageSendEvent::SendRevokeAndACK, node_a_id);
13811371

1382-
nodes[0].node.handle_update_fulfill_htlc(
1383-
nodes[1].node.get_our_node_id(),
1384-
&bs_second_updates.update_fulfill_htlcs[0],
1385-
);
1372+
nodes[0]
1373+
.node
1374+
.handle_update_fulfill_htlc(node_b_id, &bs_second_updates.update_fulfill_htlcs[0]);
13861375
expect_payment_sent(&nodes[0], payment_preimage_2, None, false, false);
1387-
nodes[0].node.handle_commitment_signed_batch_test(
1388-
nodes[1].node.get_our_node_id(),
1389-
&bs_second_updates.commitment_signed,
1390-
);
1376+
nodes[0]
1377+
.node
1378+
.handle_commitment_signed_batch_test(node_b_id, &bs_second_updates.commitment_signed);
13911379
check_added_monitors!(nodes[0], 1);
1392-
nodes[0].node.handle_revoke_and_ack(nodes[1].node.get_our_node_id(), &bs_first_raa);
1380+
nodes[0].node.handle_revoke_and_ack(node_b_id, &bs_first_raa);
13931381
expect_payment_path_successful!(nodes[0]);
13941382
check_added_monitors!(nodes[0], 1);
1395-
let (as_second_raa, as_second_update) =
1396-
get_revoke_commit_msgs!(nodes[0], nodes[1].node.get_our_node_id());
1383+
let (as_second_raa, as_second_update) = get_revoke_commit_msgs!(nodes[0], node_b_id);
13971384

1398-
nodes[1].node.handle_revoke_and_ack(nodes[0].node.get_our_node_id(), &as_second_raa);
1385+
nodes[1].node.handle_revoke_and_ack(node_a_id, &as_second_raa);
13991386
check_added_monitors!(nodes[1], 1);
1400-
nodes[1].node.handle_commitment_signed_batch_test(
1401-
nodes[0].node.get_our_node_id(),
1402-
&as_second_update,
1403-
);
1387+
nodes[1].node.handle_commitment_signed_batch_test(node_a_id, &as_second_update);
14041388
check_added_monitors!(nodes[1], 1);
1405-
let bs_second_raa = get_event_msg!(
1406-
nodes[1],
1407-
MessageSendEvent::SendRevokeAndACK,
1408-
nodes[0].node.get_our_node_id()
1409-
);
1389+
let bs_second_raa = get_event_msg!(nodes[1], MessageSendEvent::SendRevokeAndACK, node_a_id);
14101390

1411-
nodes[0].node.handle_revoke_and_ack(nodes[1].node.get_our_node_id(), &bs_second_raa);
1391+
nodes[0].node.handle_revoke_and_ack(node_b_id, &bs_second_raa);
14121392
expect_payment_path_successful!(nodes[0]);
14131393
check_added_monitors!(nodes[0], 1);
14141394
}
@@ -1420,6 +1400,9 @@ mod tests {
14201400
let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
14211401
let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
14221402

1403+
let node_a_id = nodes[0].node.get_our_node_id();
1404+
let node_c_id = nodes[2].node.get_our_node_id();
1405+
14231406
// Use FullBlockViaListen to avoid duplicate calls to process_chain_data and skips_blocks() in
14241407
// case of other connect_styles.
14251408
*nodes[0].connect_style.borrow_mut() = ConnectStyle::FullBlockViaListen;
@@ -1457,15 +1440,10 @@ mod tests {
14571440
// Now, close channel_2 i.e. b/w node-0 and node-2 to create pending_claim in node[0].
14581441
nodes[0]
14591442
.node
1460-
.force_close_broadcasting_latest_txn(
1461-
&channel_2,
1462-
&nodes[2].node.get_our_node_id(),
1463-
"Channel force-closed".to_string(),
1464-
)
1443+
.force_close_broadcasting_latest_txn(&channel_2, &node_c_id, "closed".to_string())
14651444
.unwrap();
14661445
let closure_reason =
14671446
ClosureReason::HolderForceClosed { broadcasted_latest_txn: Some(true) };
1468-
let node_c_id = nodes[2].node.get_our_node_id();
14691447
check_closed_event!(&nodes[0], 1, closure_reason, false, [node_c_id], 1000000);
14701448
check_closed_broadcast(&nodes[0], 1, true);
14711449
let close_tx = nodes[0].tx_broadcaster.txn_broadcasted.lock().unwrap().split_off(0);
@@ -1475,7 +1453,6 @@ mod tests {
14751453
check_added_monitors(&nodes[2], 1);
14761454
check_closed_broadcast(&nodes[2], 1, true);
14771455
let closure_reason = ClosureReason::CommitmentTxConfirmed;
1478-
let node_a_id = nodes[0].node.get_our_node_id();
14791456
check_closed_event!(&nodes[2], 1, closure_reason, false, [node_a_id], 1000000);
14801457

14811458
chanmon_cfgs[0].persister.chain_sync_monitor_persistences.lock().unwrap().clear();

0 commit comments

Comments
 (0)