@@ -1280,6 +1280,9 @@ mod tests {
1280
1280
let nodes = create_network ( 2 , & node_cfgs, & node_chanmgrs) ;
1281
1281
let channel_id = create_announced_chan_between_nodes ( & nodes, 0 , 1 ) . 2 ;
1282
1282
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
+
1283
1286
// Route two payments to be claimed at the same time.
1284
1287
let ( payment_preimage_1, payment_hash_1, ..) =
1285
1288
route_payment ( & nodes[ 0 ] , & [ & nodes[ 1 ] ] , 1_000_000 ) ;
@@ -1352,63 +1355,40 @@ mod tests {
1352
1355
// Now manually walk the commitment signed dance - because we claimed two payments
1353
1356
// back-to-back it doesn't fit into the neat walk commitment_signed_dance does.
1354
1357
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 ] ) ;
1360
1360
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 ) ;
1365
1362
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) ;
1368
1364
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) ;
1370
1366
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) ;
1375
1369
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) ;
1381
1371
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 ] ) ;
1386
1375
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 ) ;
1391
1379
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) ;
1393
1381
expect_payment_path_successful ! ( nodes[ 0 ] ) ;
1394
1382
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) ;
1397
1384
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) ;
1399
1386
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) ;
1404
1388
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) ;
1410
1390
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) ;
1412
1392
expect_payment_path_successful ! ( nodes[ 0 ] ) ;
1413
1393
check_added_monitors ! ( nodes[ 0 ] , 1 ) ;
1414
1394
}
@@ -1420,6 +1400,9 @@ mod tests {
1420
1400
let node_chanmgrs = create_node_chanmgrs ( 3 , & node_cfgs, & [ None , None , None ] ) ;
1421
1401
let nodes = create_network ( 3 , & node_cfgs, & node_chanmgrs) ;
1422
1402
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
+
1423
1406
// Use FullBlockViaListen to avoid duplicate calls to process_chain_data and skips_blocks() in
1424
1407
// case of other connect_styles.
1425
1408
* nodes[ 0 ] . connect_style . borrow_mut ( ) = ConnectStyle :: FullBlockViaListen ;
@@ -1457,15 +1440,10 @@ mod tests {
1457
1440
// Now, close channel_2 i.e. b/w node-0 and node-2 to create pending_claim in node[0].
1458
1441
nodes[ 0 ]
1459
1442
. 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 ( ) )
1465
1444
. unwrap ( ) ;
1466
1445
let closure_reason =
1467
1446
ClosureReason :: HolderForceClosed { broadcasted_latest_txn : Some ( true ) } ;
1468
- let node_c_id = nodes[ 2 ] . node . get_our_node_id ( ) ;
1469
1447
check_closed_event ! ( & nodes[ 0 ] , 1 , closure_reason, false , [ node_c_id] , 1000000 ) ;
1470
1448
check_closed_broadcast ( & nodes[ 0 ] , 1 , true ) ;
1471
1449
let close_tx = nodes[ 0 ] . tx_broadcaster . txn_broadcasted . lock ( ) . unwrap ( ) . split_off ( 0 ) ;
@@ -1475,7 +1453,6 @@ mod tests {
1475
1453
check_added_monitors ( & nodes[ 2 ] , 1 ) ;
1476
1454
check_closed_broadcast ( & nodes[ 2 ] , 1 , true ) ;
1477
1455
let closure_reason = ClosureReason :: CommitmentTxConfirmed ;
1478
- let node_a_id = nodes[ 0 ] . node . get_our_node_id ( ) ;
1479
1456
check_closed_event ! ( & nodes[ 2 ] , 1 , closure_reason, false , [ node_a_id] , 1000000 ) ;
1480
1457
1481
1458
chanmon_cfgs[ 0 ] . persister . chain_sync_monitor_persistences . lock ( ) . unwrap ( ) . clear ( ) ;
0 commit comments