X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Fln%2Ffunctional_test_utils.rs;h=a48c78f5217b84ae30d123658203fa3bea69b897;hb=6b0d94a3029f74de3a7542cbba0d48c2f7e5866b;hp=c677d12d45c062afdba3f7a53d4cc1a5d06ebef4;hpb=0d8b0961a5e747c6bce4b866a9e1390a35b73d24;p=rust-lightning

diff --git a/lightning/src/ln/functional_test_utils.rs b/lightning/src/ln/functional_test_utils.rs
index c677d12d..a48c78f5 100644
--- a/lightning/src/ln/functional_test_utils.rs
+++ b/lightning/src/ln/functional_test_utils.rs
@@ -1854,7 +1854,7 @@ pub fn get_route(send_node: &Node, route_params: &RouteParameters) -> Result<Rou
 	router::get_route(
 		&send_node.node.get_our_node_id(), route_params, &send_node.network_graph.read_only(),
 		Some(&send_node.node.list_usable_channels().iter().collect::<Vec<_>>()),
-		send_node.logger, &scorer, &(), &random_seed_bytes
+		send_node.logger, &scorer, &Default::default(), &random_seed_bytes
 	)
 }
 
@@ -2005,29 +2005,38 @@ macro_rules! expect_payment_path_successful {
 	}
 }
 
+pub fn expect_payment_forwarded<CM: AChannelManager, H: NodeHolder<CM=CM>>(
+	event: Event, node: &H, prev_node: &H, next_node: &H, expected_fee: Option<u64>,
+	upstream_force_closed: bool, downstream_force_closed: bool
+) {
+	match event {
+		Event::PaymentForwarded {
+			fee_earned_msat, prev_channel_id, claim_from_onchain_tx, next_channel_id,
+			outbound_amount_forwarded_msat: _
+		} => {
+			assert_eq!(fee_earned_msat, expected_fee);
+			if !upstream_force_closed {
+				// Is the event prev_channel_id in one of the channels between the two nodes?
+				assert!(node.node().list_channels().iter().any(|x| x.counterparty.node_id == prev_node.node().get_our_node_id() && x.channel_id == prev_channel_id.unwrap()));
+			}
+			// We check for force closures since a force closed channel is removed from the
+			// node's channel list
+			if !downstream_force_closed {
+				assert!(node.node().list_channels().iter().any(|x| x.counterparty.node_id == next_node.node().get_our_node_id() && x.channel_id == next_channel_id.unwrap()));
+			}
+			assert_eq!(claim_from_onchain_tx, downstream_force_closed);
+		},
+		_ => panic!("Unexpected event"),
+	}
+}
+
 macro_rules! expect_payment_forwarded {
 	($node: expr, $prev_node: expr, $next_node: expr, $expected_fee: expr, $upstream_force_closed: expr, $downstream_force_closed: expr) => {
-		let events = $node.node.get_and_clear_pending_events();
+		let mut events = $node.node.get_and_clear_pending_events();
 		assert_eq!(events.len(), 1);
-		match events[0] {
-			Event::PaymentForwarded {
-				fee_earned_msat, prev_channel_id, claim_from_onchain_tx, next_channel_id,
-				outbound_amount_forwarded_msat: _
-			} => {
-				assert_eq!(fee_earned_msat, $expected_fee);
-				if !$upstream_force_closed {
-					// Is the event prev_channel_id in one of the channels between the two nodes?
-					assert!($node.node.list_channels().iter().any(|x| x.counterparty.node_id == $prev_node.node.get_our_node_id() && x.channel_id == prev_channel_id.unwrap()));
-				}
-				// We check for force closures since a force closed channel is removed from the
-				// node's channel list
-				if !$downstream_force_closed {
-					assert!($node.node.list_channels().iter().any(|x| x.counterparty.node_id == $next_node.node.get_our_node_id() && x.channel_id == next_channel_id.unwrap()));
-				}
-				assert_eq!(claim_from_onchain_tx, $downstream_force_closed);
-			},
-			_ => panic!("Unexpected event"),
-		}
+		$crate::ln::functional_test_utils::expect_payment_forwarded(
+			events.pop().unwrap(), &$node, &$prev_node, &$next_node, $expected_fee,
+			$upstream_force_closed, $downstream_force_closed);
 	}
 }
 
@@ -2404,7 +2413,7 @@ pub fn pass_claimed_payment_along_route<'a, 'b, 'c>(origin_node: &Node<'a, 'b, '
 						}
 					};
 					if $idx == 1 { fee += expected_extra_fees[i]; }
-					expect_payment_forwarded!($node, $next_node, $prev_node, Some(fee as u64), false, false);
+					expect_payment_forwarded!(*$node, $next_node, $prev_node, Some(fee as u64), false, false);
 					expected_total_fee_msat += fee as u64;
 					check_added_monitors!($node, 1);
 					let new_next_msgs = if $new_msgs {
@@ -2497,7 +2506,7 @@ pub fn route_over_limit<'a, 'b, 'c>(origin_node: &Node<'a, 'b, 'c>, expected_rou
 	let keys_manager = test_utils::TestKeysInterface::new(&seed, Network::Testnet);
 	let random_seed_bytes = keys_manager.get_secure_random_bytes();
 	let route = router::get_route(&origin_node.node.get_our_node_id(), &route_params, &network_graph,
-		None, origin_node.logger, &scorer, &(), &random_seed_bytes).unwrap();
+		None, origin_node.logger, &scorer, &Default::default(), &random_seed_bytes).unwrap();
 	assert_eq!(route.paths.len(), 1);
 	assert_eq!(route.paths[0].hops.len(), expected_route.len());
 	for (node, hop) in expected_route.iter().zip(route.paths[0].hops.iter()) {