devimint/
lib.rs

1#![deny(clippy::pedantic)]
2#![allow(clippy::cast_possible_truncation)]
3#![allow(clippy::cast_possible_wrap)]
4#![allow(clippy::cast_precision_loss)]
5#![allow(clippy::cast_sign_loss)]
6#![allow(clippy::doc_markdown)]
7#![allow(clippy::explicit_deref_methods)]
8#![allow(clippy::implicit_hasher)]
9#![allow(clippy::items_after_statements)]
10#![allow(clippy::large_futures)]
11#![allow(clippy::missing_errors_doc)]
12#![allow(clippy::missing_panics_doc)]
13#![allow(clippy::module_name_repetitions)]
14#![allow(clippy::must_use_candidate)]
15#![allow(clippy::return_self_not_must_use)]
16#![allow(clippy::too_many_lines)]
17
18use std::ffi;
19
20use clap::Parser as _;
21use cli::cleanup_on_exit;
22use devfed::DevJitFed;
23pub use devfed::{DevFed, dev_fed};
24pub use external::{ExternalDaemons, LightningNode, Lnd, external_daemons};
25use futures::Future;
26pub use gatewayd::Gatewayd;
27pub use recurringd::Recurringd;
28use tests::log_binary_versions;
29use util::ProcessManager;
30
31pub mod cli;
32pub mod devfed;
33pub mod envs;
34pub mod external;
35pub mod faucet;
36pub mod federation;
37pub mod gatewayd;
38pub mod recurringd;
39pub mod tests;
40pub mod util;
41pub mod vars;
42pub mod version_constants;
43
44#[bon::builder]
45pub async fn run_devfed_test<F, FF>(
46    #[builder(finish_fn)] f: F,
47    num_feds: Option<usize>,
48    pre_dkg: Option<bool>,
49) -> anyhow::Result<()>
50where
51    F: FnOnce(DevJitFed, ProcessManager) -> FF,
52    FF: Future<Output = anyhow::Result<()>>,
53{
54    let mut args = cli::CommonArgs::parse_from::<_, ffi::OsString>(vec![]);
55
56    if let Some(num_feds) = num_feds {
57        args.num_feds = num_feds;
58    }
59
60    let (process_mgr, task_group) = cli::setup(args).await?;
61    log_binary_versions().await?;
62    let dev_fed = devfed::DevJitFed::new(&process_mgr, false, pre_dkg.unwrap_or_default())?;
63    // workaround https://github.com/tokio-rs/tokio/issues/6463
64    // by waiting on all jits to complete, we make it less likely
65    // that something is not finished yet and will block in `on_block`
66    let _ = dev_fed.finalize(&process_mgr).await;
67    let res = cleanup_on_exit(f(dev_fed.clone(), process_mgr.clone()), task_group).await;
68    dev_fed.fast_terminate().await;
69    res?;
70
71    Ok(())
72}