Skip to content

Commit c0657fd

Browse files
committed
Fix bench.
1 parent a20cd5c commit c0657fd

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

lightning/src/ln/channelmanager.rs

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7314,7 +7314,7 @@ mod tests {
73147314
pub mod bench {
73157315
use chain::Listen;
73167316
use chain::chainmonitor::{ChainMonitor, Persist};
7317-
use chain::keysinterface::{KeysManager, InMemorySigner};
7317+
use chain::keysinterface::{KeysManager, KeysInterface, InMemorySigner};
73187318
use ln::channelmanager::{BestBlock, ChainParameters, ChannelManager, PaymentHash, PaymentPreimage};
73197319
use ln::features::{InitFeatures, InvoiceFeatures};
73207320
use ln::functional_test_utils::*;
@@ -7323,7 +7323,7 @@ pub mod bench {
73237323
use routing::router::{PaymentParameters, get_route};
73247324
use util::test_utils;
73257325
use util::config::UserConfig;
7326-
use util::events::{Event, MessageSendEvent, MessageSendEventsProvider, PaymentPurpose};
7326+
use util::events::{Event, MessageSendEvent, MessageSendEventsProvider};
73277327

73287328
use bitcoin::hashes::Hash;
73297329
use bitcoin::hashes::sha256::Hash as Sha256;
@@ -7431,8 +7431,11 @@ pub mod bench {
74317431
let payment_params = PaymentParameters::from_node_id($node_b.get_our_node_id())
74327432
.with_features(InvoiceFeatures::known());
74337433
let scorer = test_utils::TestScorer::with_penalty(0);
7434+
let seed = [3u8; 32];
7435+
let keys_manager = KeysManager::new(&seed, 42, 42);
7436+
let random_seed_bytes = keys_manager.get_secure_random_bytes();
74347437
let route = get_route(&$node_a.get_our_node_id(), &payment_params, &dummy_graph,
7435-
Some(&usable_channels.iter().map(|r| r).collect::<Vec<_>>()), 10_000, TEST_FINAL_CLTV, &logger_a, &scorer).unwrap();
7438+
Some(&usable_channels.iter().map(|r| r).collect::<Vec<_>>()), 10_000, TEST_FINAL_CLTV, &logger_a, &scorer,&random_seed_bytes).unwrap();
74367439

74377440
let mut payment_preimage = PaymentPreimage([0; 32]);
74387441
payment_preimage.0[0..8].copy_from_slice(&payment_count.to_le_bytes());

lightning/src/routing/router.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5154,7 +5154,7 @@ mod benches {
51545154
use bitcoin::hashes::Hash;
51555155
use bitcoin::secp256k1::{PublicKey, Secp256k1, SecretKey};
51565156
use chain::transaction::OutPoint;
5157-
use chain::keysinterface::KeysInterface;
5157+
use chain::keysinterface::{KeysManager,KeysInterface};
51585158
use ln::channelmanager::{ChannelCounterparty, ChannelDetails};
51595159
use ln::features::{InitFeatures, InvoiceFeatures};
51605160
use routing::scoring::{FixedPenaltyScorer, ProbabilisticScorer, ProbabilisticScoringParameters, Scorer};
@@ -5255,7 +5255,7 @@ mod benches {
52555255
) {
52565256
let nodes = graph.read_only().nodes().clone();
52575257
let payer = payer_pubkey();
5258-
let keys_manager = test_utils::TestKeysInterface::new(&[0u8;32], Network::Testnet);
5258+
let keys_manager = KeysManager::new(&[0u8;32], 42, 42);
52595259
let random_seed_bytes = keys_manager.get_secure_random_bytes();
52605260

52615261
// First, get 100 (source, destination) pairs for which route-getting actually succeeds...

0 commit comments

Comments
 (0)