Skip to content

Commit 3d14c13

Browse files
Trivially refactor reload_node macro
Cleans it up a bit in preparation for adding a new variant in the next commit.
1 parent f3ce579 commit 3d14c13

1 file changed

Lines changed: 32 additions & 6 deletions

File tree

lightning/src/ln/functional_test_utils.rs

Lines changed: 32 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1364,8 +1364,10 @@ pub fn _reload_node<'a, 'b, 'c>(
13641364
}
13651365

13661366
#[macro_export]
1367-
macro_rules! reload_node {
1368-
($node: expr, $new_config: expr, $chanman_encoded: expr, $monitors_encoded: expr, $persister: ident, $new_chain_monitor: ident, $new_channelmanager: ident) => {
1367+
macro_rules! _reload_node_inner {
1368+
($node: expr, $new_config: expr, $chanman_encoded: expr, $monitors_encoded: expr, $persister:
1369+
ident, $new_chain_monitor: ident, $new_channelmanager: ident, $reconstruct_pending_htlcs: expr
1370+
) => {
13691371
let chanman_encoded = $chanman_encoded;
13701372

13711373
$persister = $crate::util::test_utils::TestPersister::new();
@@ -1379,22 +1381,46 @@ macro_rules! reload_node {
13791381
);
13801382
$node.chain_monitor = &$new_chain_monitor;
13811383

1382-
$new_channelmanager =
1383-
_reload_node(&$node, $new_config, &chanman_encoded, $monitors_encoded, None);
1384+
$new_channelmanager = _reload_node(
1385+
&$node,
1386+
$new_config,
1387+
&chanman_encoded,
1388+
$monitors_encoded,
1389+
$reconstruct_pending_htlcs,
1390+
);
13841391
$node.node = &$new_channelmanager;
13851392
$node.onion_messenger.set_offers_handler(&$new_channelmanager);
13861393
$node.onion_messenger.set_async_payments_handler(&$new_channelmanager);
13871394
};
1395+
}
1396+
1397+
#[macro_export]
1398+
macro_rules! reload_node {
1399+
// Reload the node using the node's current config
13881400
($node: expr, $chanman_encoded: expr, $monitors_encoded: expr, $persister: ident, $new_chain_monitor: ident, $new_channelmanager: ident) => {
13891401
let config = $node.node.get_current_config();
1390-
reload_node!(
1402+
_reload_node_inner!(
13911403
$node,
13921404
config,
13931405
$chanman_encoded,
13941406
$monitors_encoded,
13951407
$persister,
13961408
$new_chain_monitor,
1397-
$new_channelmanager
1409+
$new_channelmanager,
1410+
None
1411+
);
1412+
};
1413+
// Reload the node with the new provided config
1414+
($node: expr, $new_config: expr, $chanman_encoded: expr, $monitors_encoded: expr, $persister: ident, $new_chain_monitor: ident, $new_channelmanager: ident) => {
1415+
_reload_node_inner!(
1416+
$node,
1417+
$new_config,
1418+
$chanman_encoded,
1419+
$monitors_encoded,
1420+
$persister,
1421+
$new_chain_monitor,
1422+
$new_channelmanager,
1423+
None
13981424
);
13991425
};
14001426
}

0 commit comments

Comments
 (0)