mirror of
https://github.com/mimblewimble/grin-wallet.git
synced 2025-01-21 03:21:08 +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
116 lines
3.2 KiB
Rust
116 lines
3.2 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 a wallet repost command
|
|
#[macro_use]
|
|
extern crate log;
|
|
extern crate grin_wallet_api as api;
|
|
extern crate grin_wallet_controller as wallet;
|
|
extern crate grin_wallet_impls as impls;
|
|
extern crate grin_wallet_libwallet as libwallet;
|
|
|
|
// use crate::libwallet::api_impl::owner_updater::{start_updater_log_thread, StatusMessage};
|
|
// use grin_wallet_util::grin_core as core;
|
|
|
|
use impls::test_framework::{self, LocalWalletClient};
|
|
use std::thread;
|
|
use std::time::Duration;
|
|
|
|
#[macro_use]
|
|
mod common;
|
|
use common::{clean_output_dir, create_wallet_proxy, setup};
|
|
|
|
/// updater thread test impl
|
|
fn updater_thread_test_impl(test_dir: &'static str) -> Result<(), libwallet::Error> {
|
|
// Create a new proxy to simulate server and wallet responses
|
|
let mut wallet_proxy = create_wallet_proxy(test_dir);
|
|
let chain = wallet_proxy.chain.clone();
|
|
|
|
// Create a new wallet test client, and set its queues to communicate with the
|
|
// proxy
|
|
create_wallet_and_add!(
|
|
client1,
|
|
wallet1,
|
|
mask1_i,
|
|
test_dir,
|
|
"wallet1",
|
|
None,
|
|
&mut wallet_proxy,
|
|
false
|
|
);
|
|
let mask1 = (&mask1_i).as_ref();
|
|
create_wallet_and_add!(
|
|
client2,
|
|
wallet2,
|
|
mask2_i,
|
|
test_dir,
|
|
"wallet2",
|
|
None,
|
|
&mut wallet_proxy,
|
|
false
|
|
);
|
|
let mask2 = (&mask2_i).as_ref();
|
|
|
|
// Set the wallet proxy listener running
|
|
thread::spawn(move || {
|
|
if let Err(e) = wallet_proxy.run() {
|
|
error!("Wallet Proxy error: {}", e);
|
|
}
|
|
});
|
|
|
|
// add some accounts
|
|
wallet::controller::owner_single_use(Some(wallet1.clone()), mask1, None, |api, m| {
|
|
api.create_account_path(m, "mining")?;
|
|
api.create_account_path(m, "listener")?;
|
|
Ok(())
|
|
})?;
|
|
|
|
// add some accounts
|
|
wallet::controller::owner_single_use(Some(wallet2.clone()), mask2, None, |api, m| {
|
|
api.create_account_path(m, "account1")?;
|
|
api.create_account_path(m, "account2")?;
|
|
Ok(())
|
|
})?;
|
|
|
|
// Get some mining done
|
|
{
|
|
wallet_inst!(wallet1, w);
|
|
w.set_parent_key_id_by_name("mining")?;
|
|
}
|
|
let bh = 10u64;
|
|
let _ =
|
|
test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, bh as usize, false);
|
|
|
|
let owner_api = api::Owner::new(wallet1, None);
|
|
owner_api.start_updater(mask1, Duration::from_secs(5))?;
|
|
|
|
// let updater thread run a bit
|
|
thread::sleep(Duration::from_secs(10));
|
|
|
|
let messages = owner_api.get_updater_messages(1000)?;
|
|
assert_eq!(messages.len(), 32);
|
|
|
|
owner_api.stop_updater()?;
|
|
thread::sleep(Duration::from_secs(2));
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn updater_thread() {
|
|
let test_dir = "test_output/updater_thread";
|
|
setup(test_dir);
|
|
if let Err(e) = updater_thread_test_impl(test_dir) {
|
|
panic!("Libwallet Error: {} - {}", e, e.backtrace().unwrap());
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|