forest/dev/subcommands/tests_cmd/
wallet.rs1use super::helpers::*;
5use libtest_mimic::{Arguments, Trial};
6
7#[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, ð, 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 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}