Skip to main content

forest/dev/subcommands/tests_cmd/
wallet.rs

1// Copyright 2019-2026 ChainSafe Systems
2// SPDX-License-Identifier: Apache-2.0, MIT
3
4use super::helpers::*;
5use libtest_mimic::{Arguments, Trial};
6
7/// Wallet integration tests
8#[derive(Debug, clap::Args)]
9pub struct WalletTestCommand {}
10
11impl WalletTestCommand {
12    pub async fn run(self) -> anyhow::Result<()> {
13        let args = Arguments {
14            test_threads: Some(8),
15            ..Default::default()
16        };
17        libtest_mimic::run(&args, tests()).exit();
18    }
19}
20
21fn tests() -> Vec<Trial> {
22    vec![
23        Trial::test("export_import_roundtrip_local", || {
24            block_on(export_import_roundtrip(Backend::Local));
25            Ok(())
26        }),
27        Trial::test("export_import_roundtrip_remote", || {
28            block_on(export_import_roundtrip(Backend::Remote));
29            Ok(())
30        }),
31        Trial::test("market_add_balance_message_on_chain", || {
32            block_on(market_add_balance_message_on_chain());
33            Ok(())
34        }),
35        Trial::test("send_to_filecoin_address_local", || {
36            block_on(send_to_filecoin_address(Backend::Local));
37            Ok(())
38        }),
39        Trial::test("send_to_filecoin_address_remote", || {
40            block_on(send_to_filecoin_address(Backend::Remote));
41            Ok(())
42        }),
43        Trial::test("send_to_eth_equivalent_local", || {
44            block_on(send_to_eth_equivalent(Backend::Local));
45            Ok(())
46        }),
47        Trial::test("send_to_eth_equivalent_remote", || {
48            block_on(send_to_eth_equivalent(Backend::Remote));
49            Ok(())
50        }),
51        Trial::test("wallet_delete_local", || {
52            block_on(wallet_delete(Backend::Local));
53            Ok(())
54        }),
55        Trial::test("wallet_delete_remote", || {
56            block_on(wallet_delete(Backend::Remote));
57            Ok(())
58        }),
59        Trial::test("delegated_send_local", || {
60            block_on(delegated_send(Backend::Local));
61            Ok(())
62        }),
63        Trial::test("delegated_send_remote", || {
64            block_on(delegated_send(Backend::Remote));
65            Ok(())
66        }),
67        Trial::test("delegated_remote_send", || {
68            block_on(delegated_remote_send());
69            Ok(())
70        }),
71    ]
72}
73
74async fn export_import_roundtrip(backend: Backend) {
75    let addr = wallet(backend, &["new"]).unwrap();
76    let exported = export_to_temp_file(&addr, backend).unwrap();
77    let path = exported
78        .path()
79        .to_str()
80        .expect("temp path is not valid UTF-8");
81
82    let deleted = wallet(backend, &["delete", &addr]).unwrap();
83    eprintln!("delete output ({}): {deleted}", backend.label());
84
85    let imported = wallet(backend, &["import", path]).unwrap();
86    assert_eq!(
87        imported,
88        addr,
89        "round-trip mismatch on {} backend: {imported} != {addr}",
90        backend.label(),
91    );
92}
93
94async fn market_add_balance_message_on_chain() {
95    const ATTO_FIL: &str = "23";
96    let result = rpc_call_with_retry(
97        "Filecoin.MarketAddBalance",
98        serde_json::json!([
99            FOREST_TEST_PRELOADED_ADDRESS.as_str(),
100            FOREST_TEST_PRELOADED_ADDRESS.as_str(),
101            ATTO_FIL,
102        ]),
103    )
104    .await
105    .unwrap();
106    let msg_cid = cid_from_lotus_json_result(&result).unwrap();
107    poll_until_state_search_msg(&msg_cid).await.unwrap();
108}
109
110async fn send_to_filecoin_address(backend: Backend) {
111    let target = wallet(backend, &["new"]).unwrap();
112    let msg = send_from(&FOREST_TEST_PRELOADED_ADDRESS, &target, FIL_AMT, backend).unwrap();
113    eprintln!("send to {target} ({}) msg: {msg}", backend.label());
114    let funded = poll_until_funded(&target, backend).await.unwrap();
115    eprintln!("{target} funded balance: {funded}");
116}
117
118async fn send_to_eth_equivalent(backend: Backend) {
119    let target = wallet(backend, &["new"]).unwrap();
120    let initial_msg = send_from(&FOREST_TEST_PRELOADED_ADDRESS, &target, FIL_AMT, backend).unwrap();
121    eprintln!(
122        "initial send to {target} ({}) msg: {initial_msg}",
123        backend.label(),
124    );
125    let baseline = poll_until_funded(&target, backend).await.unwrap();
126
127    let eth = filecoin_to_eth(&target).await.unwrap();
128    let eth_msg = send_from(&FOREST_TEST_PRELOADED_ADDRESS, &eth, FIL_AMT, backend).unwrap();
129    eprintln!("send to ETH {eth} (mapped from {target}) msg: {eth_msg}");
130
131    let updated = poll_until_changed(&target, &baseline, backend)
132        .await
133        .unwrap();
134    assert!(
135        updated != baseline,
136        "{target} balance unchanged after ETH-equivalent send: {updated}",
137    );
138}
139
140async fn wallet_delete(backend: Backend) {
141    let addr = wallet(backend, &["new"]).unwrap();
142    let deleted = wallet(backend, &["delete", &addr]).unwrap();
143    eprintln!("delete output ({}): {deleted}", backend.label());
144    let listing = wallet(backend, &["list"]).unwrap();
145    assert!(
146        !listing.contains(&addr),
147        "deleted wallet {addr} still appears in `list`:\n{listing}",
148    );
149}
150
151async fn delegated_send(target_backend: Backend) {
152    let funded = funded_delegated_addr().await;
153    let target = wallet(target_backend, &["new", "delegated"]).unwrap();
154    // Baseline `FIL_ZERO` ⇒ first credit; otherwise expect a balance delta.
155    let baseline = balance(&target, target_backend).unwrap();
156    let msg = send_from(funded, &target, FIL_AMT, Backend::Local).unwrap();
157    eprintln!(
158        "delegated send to {target} ({}) msg: {msg}",
159        target_backend.label(),
160    );
161    let observed = if baseline == FIL_ZERO {
162        poll_until_funded(&target, target_backend).await.unwrap()
163    } else {
164        poll_until_changed(&target, &baseline, target_backend)
165            .await
166            .unwrap()
167    };
168    assert!(
169        observed != baseline,
170        "{target} balance unchanged after delegated send: {observed}",
171    );
172}
173
174async fn delegated_remote_send() {
175    let funded = funded_delegated_addr().await;
176    let target = wallet(Backend::Remote, &["new", "delegated"]).unwrap();
177    let baseline = balance(&target, Backend::Remote).unwrap();
178    let msg = send_from(funded, &target, FIL_AMT, Backend::Remote).unwrap();
179    eprintln!("delegated --remote-wallet send to {target} msg: {msg}");
180    let observed = if baseline == FIL_ZERO {
181        poll_until_funded(&target, Backend::Remote).await.unwrap()
182    } else {
183        poll_until_changed(&target, &baseline, Backend::Remote)
184            .await
185            .unwrap()
186    };
187    assert!(
188        observed != baseline,
189        "{target} balance unchanged after delegated --remote-wallet send: {observed}",
190    );
191}