mirror of
https://github.com/mimblewimble/grin-wallet.git
synced 2025-01-20 19:11:09 +03:00
2d264db91a
* cli addition to yml * first experiments with CLI mode * rustfmt * modify owner_single_use to take optional instantiated API * refactor command functions to take owner api instance * modify commands to use single owner api instance * add ability to pass customised message handler to API * prevent update wallet attempt if wallet hasn't been opened * fix output when internal updater is running * add termion, use cli for status updates, remove logging output from CLI * some attempts to clean up the command prompt and accomodate scanning intervals * rework listener slightly to allow for cli mode, add close command * remove termion dependency
563 lines
14 KiB
Rust
563 lines
14 KiB
Rust
// Copyright 2019 The Grin Developers
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
//! Test wallet command line works as expected
|
|
#[macro_use]
|
|
extern crate clap;
|
|
|
|
#[macro_use]
|
|
extern crate log;
|
|
|
|
extern crate grin_wallet;
|
|
|
|
use grin_wallet_impls::test_framework::{self, LocalWalletClient, WalletProxy};
|
|
|
|
use clap::App;
|
|
use std::thread;
|
|
use std::time::Duration;
|
|
|
|
use grin_wallet_impls::DefaultLCProvider;
|
|
use grin_wallet_util::grin_keychain::ExtKeychain;
|
|
|
|
mod common;
|
|
use common::{clean_output_dir, execute_command, initial_setup_wallet, instantiate_wallet, setup};
|
|
|
|
/// command line tests
|
|
fn command_line_test_impl(test_dir: &str) -> Result<(), grin_wallet_controller::Error> {
|
|
setup(test_dir);
|
|
// Create a new proxy to simulate server and wallet responses
|
|
let mut wallet_proxy: WalletProxy<
|
|
DefaultLCProvider<LocalWalletClient, ExtKeychain>,
|
|
LocalWalletClient,
|
|
ExtKeychain,
|
|
> = WalletProxy::new(test_dir);
|
|
let chain = wallet_proxy.chain.clone();
|
|
|
|
// load app yaml. If it don't exist, just say so and exit
|
|
let yml = load_yaml!("../src/bin/grin-wallet.yml");
|
|
let app = App::from_yaml(yml);
|
|
|
|
// wallet init
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "init", "-h"];
|
|
// should create new wallet file
|
|
let client1 = LocalWalletClient::new("wallet1", wallet_proxy.tx.clone());
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec.clone())?;
|
|
|
|
// trying to init twice - should fail
|
|
assert!(execute_command(&app, test_dir, "wallet1", &client1, arg_vec.clone()).is_err());
|
|
let client1 = LocalWalletClient::new("wallet1", wallet_proxy.tx.clone());
|
|
|
|
// add wallet to proxy
|
|
//let wallet1 = test_framework::create_wallet(&format!("{}/wallet1", test_dir), client1.clone());
|
|
let config1 = initial_setup_wallet(test_dir, "wallet1");
|
|
let wallet_config1 = config1.clone().members.unwrap().wallet;
|
|
let (wallet1, mask1_i) = instantiate_wallet(
|
|
wallet_config1.clone(),
|
|
client1.clone(),
|
|
"password",
|
|
"default",
|
|
)?;
|
|
wallet_proxy.add_wallet(
|
|
"wallet1",
|
|
client1.get_send_instance(),
|
|
wallet1.clone(),
|
|
mask1_i.clone(),
|
|
);
|
|
|
|
// Create wallet 2
|
|
let client2 = LocalWalletClient::new("wallet2", wallet_proxy.tx.clone());
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec.clone())?;
|
|
|
|
let config2 = initial_setup_wallet(test_dir, "wallet2");
|
|
let wallet_config2 = config2.clone().members.unwrap().wallet;
|
|
let (wallet2, mask2_i) = instantiate_wallet(
|
|
wallet_config2.clone(),
|
|
client2.clone(),
|
|
"password",
|
|
"default",
|
|
)?;
|
|
wallet_proxy.add_wallet(
|
|
"wallet2",
|
|
client2.get_send_instance(),
|
|
wallet2.clone(),
|
|
mask2_i.clone(),
|
|
);
|
|
|
|
// Set the wallet proxy listener running
|
|
thread::spawn(move || {
|
|
if let Err(e) = wallet_proxy.run() {
|
|
error!("Wallet Proxy error: {}", e);
|
|
}
|
|
});
|
|
|
|
// Create some accounts in wallet 1
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "account", "-c", "mining"];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"account",
|
|
"-c",
|
|
"account_1",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
// Create some accounts in wallet 2
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"account",
|
|
"-c",
|
|
"account_1",
|
|
];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec.clone())?;
|
|
// already exists
|
|
assert!(execute_command(&app, test_dir, "wallet2", &client2, arg_vec).is_err());
|
|
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"account",
|
|
"-c",
|
|
"account_2",
|
|
];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
|
|
// let's see those accounts
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "account"];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
|
|
// let's see those accounts
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "account"];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
|
|
// Mine a bit into wallet 1 so we have something to send
|
|
// (TODO: Be able to stop listeners so we can test this better)
|
|
let wallet_config1 = config1.clone().members.unwrap().wallet;
|
|
let (wallet1, mask1_i) =
|
|
instantiate_wallet(wallet_config1, client1.clone(), "password", "default")?;
|
|
let mask1 = (&mask1_i).as_ref();
|
|
grin_wallet_controller::controller::owner_single_use(
|
|
Some(wallet1.clone()),
|
|
mask1,
|
|
None,
|
|
|api, m| {
|
|
api.set_active_account(m, "mining")?;
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
let mut bh = 10u64;
|
|
let _ =
|
|
test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, bh as usize, false);
|
|
|
|
let very_long_message = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
ABCDEFGHIJKLMNOPQRSTUVWXYZabcdef\
|
|
This part should all be truncated";
|
|
|
|
// Update info and check
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "-a", "mining", "info"];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
// try a file exchange
|
|
let file_name = format!("{}/tx1.part_tx", test_dir);
|
|
let response_file_name = format!("{}/tx1.part_tx.response", test_dir);
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"send",
|
|
"-m",
|
|
"file",
|
|
"-d",
|
|
&file_name,
|
|
"-g",
|
|
very_long_message,
|
|
"10",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"account_1",
|
|
"receive",
|
|
"-i",
|
|
&file_name,
|
|
"-g",
|
|
"Thanks, Yeast!",
|
|
];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec.clone())?;
|
|
|
|
// shouldn't be allowed to receive twice
|
|
assert!(execute_command(&app, test_dir, "wallet2", &client2, arg_vec).is_err());
|
|
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-a",
|
|
"mining",
|
|
"-p",
|
|
"password",
|
|
"finalize",
|
|
"-i",
|
|
&response_file_name,
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
bh += 1;
|
|
|
|
let wallet_config1 = config1.clone().members.unwrap().wallet;
|
|
let (wallet1, mask1_i) = instantiate_wallet(
|
|
wallet_config1.clone(),
|
|
client1.clone(),
|
|
"password",
|
|
"default",
|
|
)?;
|
|
let mask1 = (&mask1_i).as_ref();
|
|
|
|
// Check our transaction log, should have 10 entries
|
|
grin_wallet_controller::controller::owner_single_use(
|
|
Some(wallet1.clone()),
|
|
mask1,
|
|
None,
|
|
|api, m| {
|
|
api.set_active_account(m, "mining")?;
|
|
let (refreshed, txs) = api.retrieve_txs(m, true, None, None)?;
|
|
assert!(refreshed);
|
|
assert_eq!(txs.len(), bh as usize);
|
|
for t in txs {
|
|
assert!(t.kernel_excess.is_some());
|
|
}
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
let _ = test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, 10, false);
|
|
bh += 10;
|
|
|
|
// update info for each
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "-a", "mining", "info"];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "-a", "account_1", "info"];
|
|
execute_command(&app, test_dir, "wallet2", &client1, arg_vec)?;
|
|
|
|
// check results in wallet 2
|
|
let wallet_config2 = config2.clone().members.unwrap().wallet;
|
|
let (wallet2, mask2_i) = instantiate_wallet(
|
|
wallet_config2.clone(),
|
|
client2.clone(),
|
|
"password",
|
|
"default",
|
|
)?;
|
|
let mask2 = (&mask2_i).as_ref();
|
|
|
|
grin_wallet_controller::controller::owner_single_use(
|
|
Some(wallet2.clone()),
|
|
mask2,
|
|
None,
|
|
|api, m| {
|
|
api.set_active_account(m, "account_1")?;
|
|
let (_, wallet1_info) = api.retrieve_summary_info(m, true, 1)?;
|
|
assert_eq!(wallet1_info.last_confirmed_height, bh);
|
|
assert_eq!(wallet1_info.amount_currently_spendable, 10_000_000_000);
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
// Self-send to same account, using smallest strategy
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"send",
|
|
"-m",
|
|
"file",
|
|
"-d",
|
|
&file_name,
|
|
"-g",
|
|
"Love, Yeast, Smallest",
|
|
"-s",
|
|
"smallest",
|
|
"10",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"receive",
|
|
"-i",
|
|
&file_name,
|
|
"-g",
|
|
"Thanks, Yeast!",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec.clone())?;
|
|
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-a",
|
|
"mining",
|
|
"-p",
|
|
"password",
|
|
"finalize",
|
|
"-i",
|
|
&response_file_name,
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
bh += 1;
|
|
|
|
// Check our transaction log, should have bh entries + one for the self receive
|
|
let wallet_config1 = config1.clone().members.unwrap().wallet;
|
|
let (wallet1, mask1_i) = instantiate_wallet(
|
|
wallet_config1.clone(),
|
|
client1.clone(),
|
|
"password",
|
|
"default",
|
|
)?;
|
|
let mask1 = (&mask1_i).as_ref();
|
|
|
|
grin_wallet_controller::controller::owner_single_use(
|
|
Some(wallet1.clone()),
|
|
mask1,
|
|
None,
|
|
|api, m| {
|
|
api.set_active_account(m, "mining")?;
|
|
let (refreshed, txs) = api.retrieve_txs(m, true, None, None)?;
|
|
assert!(refreshed);
|
|
assert_eq!(txs.len(), bh as usize + 1);
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
// Try using the self-send method, splitting up outputs for the fun of it
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"send",
|
|
"-m",
|
|
"self",
|
|
"-d",
|
|
"mining",
|
|
"-g",
|
|
"Self love",
|
|
"-o",
|
|
"3",
|
|
"-s",
|
|
"smallest",
|
|
"10",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
bh += 1;
|
|
|
|
// Check our transaction log, should have bh entries + 2 for the self receives
|
|
let wallet_config1 = config1.clone().members.unwrap().wallet;
|
|
let (wallet1, mask1_i) = instantiate_wallet(
|
|
wallet_config1.clone(),
|
|
client1.clone(),
|
|
"password",
|
|
"default",
|
|
)?;
|
|
let mask1 = (&mask1_i).as_ref();
|
|
|
|
grin_wallet_controller::controller::owner_single_use(
|
|
Some(wallet1.clone()),
|
|
mask1,
|
|
None,
|
|
|api, m| {
|
|
api.set_active_account(m, "mining")?;
|
|
let (refreshed, txs) = api.retrieve_txs(m, true, None, None)?;
|
|
assert!(refreshed);
|
|
assert_eq!(txs.len(), bh as usize + 2);
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
// Another file exchange, don't send, but unlock with repair command
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"send",
|
|
"-m",
|
|
"file",
|
|
"-d",
|
|
&file_name,
|
|
"-g",
|
|
"Ain't sending",
|
|
"10",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "scan", "-d"];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
// Another file exchange, cancel this time
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"send",
|
|
"-m",
|
|
"file",
|
|
"-d",
|
|
&file_name,
|
|
"-g",
|
|
"Ain't sending 2",
|
|
"10",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"cancel",
|
|
"-i",
|
|
"26",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
// issue an invoice tx, wallet 2
|
|
let file_name = format!("{}/invoice.slate", test_dir);
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"invoice",
|
|
"-d",
|
|
&file_name,
|
|
"-g",
|
|
"Please give me your precious grins. Love, Yeast",
|
|
"65",
|
|
];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
let output_file_name = format!("{}/invoice.slate.paid", test_dir);
|
|
|
|
// now pay the invoice tx, wallet 1
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-a",
|
|
"mining",
|
|
"-p",
|
|
"password",
|
|
"pay",
|
|
"-i",
|
|
&file_name,
|
|
"-d",
|
|
&output_file_name,
|
|
"-g",
|
|
"Here you go",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
// and finalize, wallet 2
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"finalize",
|
|
"-i",
|
|
&output_file_name,
|
|
];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
|
|
// bit more mining
|
|
let _ = test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, 5, false);
|
|
//bh += 5;
|
|
|
|
// txs and outputs (mostly spit out for a visual in test logs)
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "-a", "mining", "txs"];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
// message output (mostly spit out for a visual in test logs)
|
|
let arg_vec = vec![
|
|
"grin-wallet",
|
|
"-p",
|
|
"password",
|
|
"-a",
|
|
"mining",
|
|
"txs",
|
|
"-i",
|
|
"10",
|
|
];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
// txs and outputs (mostly spit out for a visual in test logs)
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "-a", "mining", "outputs"];
|
|
execute_command(&app, test_dir, "wallet1", &client1, arg_vec)?;
|
|
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "txs"];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "outputs"];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
|
|
// get tx output via -tx parameter
|
|
let mut tx_id = "".to_string();
|
|
grin_wallet_controller::controller::owner_single_use(
|
|
Some(wallet2.clone()),
|
|
mask2,
|
|
None,
|
|
|api, m| {
|
|
api.set_active_account(m, "default")?;
|
|
let (_, txs) = api.retrieve_txs(m, true, None, None)?;
|
|
let some_tx_id = txs[0].tx_slate_id.clone();
|
|
assert!(some_tx_id.is_some());
|
|
tx_id = some_tx_id.unwrap().to_hyphenated().to_string().clone();
|
|
Ok(())
|
|
},
|
|
)?;
|
|
let arg_vec = vec!["grin-wallet", "-p", "password", "txs", "-t", &tx_id[..]];
|
|
execute_command(&app, test_dir, "wallet2", &client2, arg_vec)?;
|
|
|
|
// let logging finish
|
|
thread::sleep(Duration::from_millis(200));
|
|
clean_output_dir(test_dir);
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn wallet_command_line() {
|
|
let test_dir = "target/test_output/command_line";
|
|
if let Err(e) = command_line_test_impl(test_dir) {
|
|
panic!("Libwallet Error: {} - {}", e, e.backtrace().unwrap());
|
|
}
|
|
}
|