mirror of
https://github.com/mimblewimble/grin-wallet.git
synced 2025-01-21 11:31:09 +03:00
f0cf8b0a7d
* Add prototype contract implementation Lacks: - Payment proofs (add early payment proofs) - Tests - Better structure * Separate contract utilities * Simplify the main setup/sign code flow * Remove commented initial_sec_key assignment * Simplify sign::compute function * Add logic for "removal" of secret keys from the context * Further simplify a bit sign and setup * No need for mutable context when adding outputs * Refactor the commented code (lol) * Refactor a bit * tmp --------- Co-authored-by: oryhp <gtrphyro@gmail.com>
104 lines
3.5 KiB
Rust
104 lines
3.5 KiB
Rust
// Copyright 2022 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 doing contract early lock when using --add-outputs
|
|
// #[macro_use]
|
|
extern crate grin_wallet_controller as wallet;
|
|
extern crate grin_wallet_impls as impls;
|
|
extern crate log;
|
|
|
|
use grin_wallet_libwallet as libwallet;
|
|
|
|
use grin_core::consensus;
|
|
use impls::test_framework::{self};
|
|
use libwallet::contract::my_fee_contribution;
|
|
use libwallet::contract::types::{ContractNewArgsAPI, ContractSetupArgsAPI, OutputSelectionArgs};
|
|
use libwallet::{OutputCommitMapping, OutputStatus, Slate, SlateState, TxLogEntryType};
|
|
use std::sync::atomic::Ordering;
|
|
use std::thread;
|
|
use std::time::Duration;
|
|
|
|
#[macro_use]
|
|
mod common;
|
|
use common::{clean_output_dir, create_wallets, setup};
|
|
|
|
/// contract new with --add-outputs
|
|
fn contract_early_lock_tx_impl(test_dir: &'static str) -> Result<(), libwallet::Error> {
|
|
// create a single wallet and mine 5 blocks
|
|
let (wallets, chain, stopper, mut bh) =
|
|
create_wallets(vec![vec![("default", 5)]], test_dir).unwrap();
|
|
let send_wallet = wallets[0].0.clone();
|
|
let send_mask = wallets[0].1.as_ref();
|
|
|
|
// Confirm all our outputs are unspent
|
|
wallet::controller::owner_single_use(Some(send_wallet.clone()), send_mask, None, |api, m| {
|
|
let (_, commits) = api.retrieve_outputs(m, true, false, None)?;
|
|
for commit in commits.iter() {
|
|
assert_eq!(commit.output.status, OutputStatus::Unspent);
|
|
}
|
|
Ok(())
|
|
})?;
|
|
|
|
let mut slate = Slate::blank(0, false); // this gets overriden below
|
|
|
|
// Call contract 'new' with --add-outputs
|
|
wallet::controller::owner_single_use(Some(send_wallet.clone()), send_mask, None, |api, m| {
|
|
// Send wallet inititates a standard transaction with --send=80
|
|
let args = &ContractNewArgsAPI {
|
|
setup_args: ContractSetupArgsAPI {
|
|
net_change: Some(-80_000_000_000),
|
|
num_participants: 2,
|
|
add_outputs: true,
|
|
..Default::default()
|
|
},
|
|
..Default::default()
|
|
};
|
|
slate = api.contract_new(m, args)?;
|
|
Ok(())
|
|
})?;
|
|
assert_eq!(slate.state, SlateState::Standard1);
|
|
|
|
// Assert we locked 2 inputs and prepared an unconfirmed change output
|
|
wallet::controller::owner_single_use(Some(send_wallet.clone()), send_mask, None, |api, m| {
|
|
let (_, commits) = api.retrieve_outputs(m, true, false, None)?;
|
|
// we locked the first two coinbase outputs
|
|
assert_eq!(commits[0].output.status, OutputStatus::Locked);
|
|
assert_eq!(commits[1].output.status, OutputStatus::Locked);
|
|
// we added a new unconfirmed change output
|
|
let new_output_idx = commits.len() - 1;
|
|
assert_eq!(
|
|
commits[new_output_idx].output.status,
|
|
OutputStatus::Unconfirmed
|
|
);
|
|
assert_eq!(
|
|
commits[new_output_idx].output.value,
|
|
2 * consensus::REWARD - 80_000_000_000 - my_fee_contribution(2, 1, 1, 2)?.fee()
|
|
);
|
|
Ok(())
|
|
})?;
|
|
|
|
// let logging finish
|
|
stopper.store(false, Ordering::Relaxed);
|
|
thread::sleep(Duration::from_millis(200));
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn wallet_contract_early_lock_tx() -> Result<(), libwallet::Error> {
|
|
let test_dir = "test_output/contract_early_lock_tx";
|
|
setup(test_dir);
|
|
contract_early_lock_tx_impl(test_dir)?;
|
|
clean_output_dir(test_dir);
|
|
Ok(())
|
|
}
|