X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Futil%2Ftest_utils.rs;h=a006d37e90e434f92905cd66a06138820ff731b1;hb=52c225351bceac06588f8db48bb748d373818a0a;hp=ba56edf3584f017ab0aa623e19aad6d2ec07f2e2;hpb=f5e87d8441f2e873114fd3b310a04be0a2a9181f;p=rust-lightning

diff --git a/lightning/src/util/test_utils.rs b/lightning/src/util/test_utils.rs
index ba56edf3..a006d37e 100644
--- a/lightning/src/util/test_utils.rs
+++ b/lightning/src/util/test_utils.rs
@@ -19,7 +19,7 @@ use crate::chain::chainmonitor::{MonitorUpdateId, UpdateOrigin};
 use crate::chain::channelmonitor;
 use crate::chain::channelmonitor::MonitorEvent;
 use crate::chain::transaction::OutPoint;
-use crate::routing::router::CandidateRouteHop;
+use crate::routing::router::{CandidateRouteHop, FirstHopCandidate, PublicHopCandidate, PrivateHopCandidate};
 use crate::sign;
 use crate::events;
 use crate::events::bump_transaction::{WalletSource, Utxo};
@@ -146,10 +146,10 @@ impl<'a> Router for TestRouter<'a> {
 							if let Some(first_hops) = first_hops {
 								if let Some(idx) = first_hops.iter().position(|h| h.get_outbound_payment_scid() == Some(hop.short_channel_id)) {
 									let node_id = NodeId::from_pubkey(payer);
-									let candidate = CandidateRouteHop::FirstHop {
+									let candidate = CandidateRouteHop::FirstHop(FirstHopCandidate {
 										details: first_hops[idx],
 										payer_node_id: &node_id,
-									};
+									});
 									scorer.channel_penalty_msat(&candidate, usage, &());
 									continue;
 								}
@@ -158,10 +158,10 @@ impl<'a> Router for TestRouter<'a> {
 						let network_graph = self.network_graph.read_only();
 						if let Some(channel) = network_graph.channel(hop.short_channel_id) {
 							let (directed, _) = channel.as_directed_to(&NodeId::from_pubkey(&hop.pubkey)).unwrap();
-							let candidate = CandidateRouteHop::PublicHop {
+							let candidate = CandidateRouteHop::PublicHop(PublicHopCandidate {
 								info: directed,
 								short_channel_id: hop.short_channel_id,
-							};
+							});
 							scorer.channel_penalty_msat(&candidate, usage, &());
 						} else {
 							let target_node_id = NodeId::from_pubkey(&hop.pubkey);
@@ -173,10 +173,10 @@ impl<'a> Router for TestRouter<'a> {
 								htlc_minimum_msat: None,
 								htlc_maximum_msat: None,
 							};
-							let candidate = CandidateRouteHop::PrivateHop {
+							let candidate = CandidateRouteHop::PrivateHop(PrivateHopCandidate {
 								hint: &route_hint,
 								target_node_id: &target_node_id,
-							};
+							});
 							scorer.channel_penalty_msat(&candidate, usage, &());
 						}
 						prev_hop_node = &hop.pubkey;