github-jwinterm/grin-wallet
1
0
Fork 0
mirror of https://github.com/jwinterm/grin-wallet.git synced 2026-10-21 17:18:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

clean up contract tests

Browse Source
This commit is contained in:
wiesche
2026-09-14 08:13:48 +02:00
parent cf488920db
commit 049e73ac07
8 changed files with 81 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files
controller/tests/common/mod.rs
+34
View File
@@ -97,6 +97,40 @@ pub fn contract_selection_args() -> libwallet::contract::types::OutputSelectionA
}
}
#[allow(dead_code)]
pub fn contract_setup_args(
net_change: Option<i64>,
) -> libwallet::contract::types::ContractSetupArgsAPI {
libwallet::contract::types::ContractSetupArgsAPI {
selection_args: contract_selection_args(),
net_change,
..Default::default()
}
}
#[allow(dead_code)]
pub fn contract_new_args(net_change: i64) -> libwallet::contract::types::ContractNewArgsAPI {
libwallet::contract::types::ContractNewArgsAPI {
setup_args: contract_setup_args(Some(net_change)),
..Default::default()
}
}
#[allow(dead_code)]
pub fn roundtrip_slate(
slate: &libwallet::Slate,
version: u16,
) -> Result<libwallet::Slate, libwallet::Error> {
let packer = libwallet::Slatepacker::new(libwallet::SlatepackerArgs {
sender: None,
recipients: vec![],
dec_key: None,
});
let slate = packer.get_slate(&packer.create_slatepack(slate)?)?;
assert_eq!(slate.version_info.version, version);
Ok(slate)
}
#[allow(dead_code)]
pub struct ExpectedContractSlate {
pub amount: u64,
controller/tests/contract_accounts_switch.rs
+3 -4
View File
@@ -62,8 +62,8 @@ fn contract_accounts_switch_impl(test_dir: &'static str) -> Result<(), libwallet
let (wallet1_refreshed, wallet1_info) = api.retrieve_summary_info(m, true, 1)?;
assert!(wallet1_refreshed);
assert_eq!(wallet1_info.last_confirmed_height, 10);
assert_eq!(wallet1_info.total, 1 * reward);
assert_eq!(wallet1_info.amount_currently_spendable, 1 * reward);
assert_eq!(wallet1_info.total, reward);
assert_eq!(wallet1_info.amount_currently_spendable, reward);
let (_, txs) = api.retrieve_txs(m, true, None, None, None)?;
assert_eq!(txs.len(), 1);
Ok(())
@@ -116,7 +116,6 @@ fn contract_accounts_switch_impl(test_dir: &'static str) -> Result<(), libwallet
net_change: Some(-5_000_000_000),
..Default::default()
},
..Default::default()
};
slate = api.contract_new(m, args)?;
Ok(())
@@ -227,7 +226,7 @@ fn contract_accounts_switch_impl(test_dir: &'static str) -> Result<(), libwallet
assert_eq!(wallet1_info.last_confirmed_height, 11);
assert_eq!(
wallet1_info.total,
1 * reward - 5_000_000_000 - my_fee_contribution(1, 1, 1, 2)?.fee() // we subtract also our fee contribution
reward - 5_000_000_000 - my_fee_contribution(1, 1, 1, 2)?.fee() // we subtract also our fee contribution
);
let (_, txs) = api.retrieve_txs(m, true, None, None, None)?;
assert_eq!(txs.len(), 2);
controller/tests/contract_early_proofs_rsr.rs
+11 -38
View File
@@ -27,10 +27,8 @@ use grin_wallet_libwallet as libwallet;
use grin_util::secp::{Secp256k1, Signature};
use impls::test_framework::{self};
use libwallet::contract::my_fee_contribution;
use libwallet::contract::types::{
ContractNewArgsAPI, ContractSetupArgsAPI, PaymentMemo, ProofType,
};
use libwallet::{Slate, SlateState, Slatepacker, SlatepackerArgs, TxLogEntryType};
use libwallet::contract::types::{PaymentMemo, ProofType};
use libwallet::{Slate, SlateState, TxLogEntryType};
use std::sync::atomic::Ordering;
use std::thread;
use std::time::Duration;
@@ -40,17 +38,6 @@ mod common;
use common::{clean_output_dir, create_wallets, setup};
use std::path::PathBuf;
fn roundtrip_slate(slate: &Slate, version: u16) -> Result<Slate, libwallet::Error> {
let packer = Slatepacker::new(SlatepackerArgs {
sender: None,
recipients: vec![],
dec_key: None,
});
let slate = packer.get_slate(&packer.create_slatepack(slate)?)?;
assert_eq!(slate.version_info.version, version);
Ok(slate)
}
/// Development + Tests of early payment proof functionality - RSR workflow
fn contract_early_proofs_rsr_test_impl(
test_dir: &'static str,
@@ -85,14 +72,7 @@ fn contract_early_proofs_rsr_test_impl(
PathBuf::from(test_dir),
|api, m| {
// Receive wallet (invoice) calls --receive=5
let args = &mut ContractNewArgsAPI {
setup_args: ContractSetupArgsAPI {
selection_args: common::contract_selection_args(),
net_change: Some(5_000_000_000),
..Default::default()
},
..Default::default()
};
let args = &mut common::contract_new_args(5_000_000_000);
// Proofs are opt-in; enable and supply the sender address.
args.setup_args.proof_args.suppress_proof = false;
args.setup_args.proof_args.proof_type = proof_type;
@@ -105,7 +85,7 @@ fn contract_early_proofs_rsr_test_impl(
)?;
assert_eq!(slate.state, SlateState::Invoice1);
slate = roundtrip_slate(&slate, 5)?;
slate = common::roundtrip_slate(&slate, 5)?;
wallet::controller::owner_single_use(
send_wallet.clone(),
@@ -113,18 +93,14 @@ fn contract_early_proofs_rsr_test_impl(
PathBuf::from(test_dir),
|api, m| {
// Sending wallet (invoice) signs
let args = &ContractSetupArgsAPI {
selection_args: common::contract_selection_args(),
net_change: Some(-5_000_000_000),
..Default::default()
};
let args = &common::contract_setup_args(Some(-5_000_000_000));
slate = api.contract_sign(m, &slate, args)?;
Ok(())
},
)?;
assert_eq!(slate.state, SlateState::Invoice2);
slate = roundtrip_slate(&slate, 5)?;
slate = common::roundtrip_slate(&slate, 5)?;
// Send wallet finalizes and posts
wallet::controller::owner_single_use(
@@ -132,16 +108,13 @@ fn contract_early_proofs_rsr_test_impl(
recv_mask,
PathBuf::from(test_dir),
|api, m| {
let args = &ContractSetupArgsAPI {
selection_args: common::contract_selection_args(),
..Default::default()
};
let args = &common::contract_setup_args(None);
slate = api.contract_sign(m, &slate, args)?;
Ok(())
},
)?;
assert_eq!(slate.state, SlateState::Invoice3);
slate = roundtrip_slate(&slate, 5)?;
slate = common::roundtrip_slate(&slate, 5)?;
wallet::controller::owner_single_use(
send_wallet.clone(),
@@ -175,11 +148,11 @@ fn contract_early_proofs_rsr_test_impl(
assert_eq!(tx_log.amount_debited, 0);
assert_eq!(tx_log.num_inputs, 1);
assert_eq!(tx_log.num_outputs, 1);
let expected_fees_paid = Some(my_fee_contribution(1, 1, 1, 2)?);
assert_eq!(tx_log.fee, expected_fees_paid);
let expected_fees_paid = my_fee_contribution(1, 1, 1, 2)?;
assert_eq!(tx_log.fee, Some(expected_fees_paid));
assert_eq!(
wallet_info.amount_currently_spendable,
4 * 60_000_000_000 + 5_000_000_000 - expected_fees_paid.unwrap().fee() // we expect the balance of 4 mined blocks + 5 Grin - fees paid
4 * 60_000_000_000 + 5_000_000_000 - expected_fees_paid.fee() // we expect the balance of 4 mined blocks + 5 Grin - fees paid
);
Ok(())
},
controller/tests/contract_early_proofs_srs.rs
+11 -36
View File
@@ -26,8 +26,8 @@ use grin_wallet_libwallet as libwallet;
use impls::test_framework::{self};
use libwallet::contract::my_fee_contribution;
use libwallet::contract::types::{ContractNewArgsAPI, ContractSetupArgsAPI, PaymentMemo};
use libwallet::{NodeVersionInfo, Slate, SlateState, Slatepacker, SlatepackerArgs, TxLogEntryType};
use libwallet::contract::types::PaymentMemo;
use libwallet::{NodeVersionInfo, Slate, SlateState, TxLogEntryType};
use std::sync::atomic::Ordering;
use std::thread;
use std::time::Duration;
@@ -52,17 +52,6 @@ fn reject_proof_verification(
))
}
fn roundtrip_slate(slate: &Slate, version: u16) -> Result<Slate, libwallet::Error> {
let packer = Slatepacker::new(SlatepackerArgs {
sender: None,
recipients: vec![],
dec_key: None,
});
let slate = packer.get_slate(&packer.create_slatepack(slate)?)?;
assert_eq!(slate.version_info.version, version);
Ok(slate)
}
/// Development + Tests of early payment proof functionality
fn contract_early_proofs_srs_test_impl(test_dir: &'static str) -> Result<(), libwallet::Error> {
// create two wallets and mine 4 blocks in each (we want both to have balance to get a payjoin)
@@ -82,21 +71,14 @@ fn contract_early_proofs_srs_test_impl(test_dir: &'static str) -> Result<(), lib
PathBuf::from(test_dir),
|api, m| {
// Send wallet inititates a standard transaction with --send=5
let args = &ContractNewArgsAPI {
setup_args: ContractSetupArgsAPI {
selection_args: common::contract_selection_args(),
net_change: Some(-5_000_000_000),
..Default::default()
},
..Default::default()
};
let args = &common::contract_new_args(-5_000_000_000);
slate = api.contract_new(m, args)?;
sender_address = Some(api.get_slatepack_address(send_mask, 0)?.pub_key);
Ok(())
},
)?;
assert_eq!(slate.state, SlateState::Standard1);
slate = roundtrip_slate(&slate, 4)?;
slate = common::roundtrip_slate(&slate, 4)?;
let mut recipient_address = None;
wallet::controller::owner_single_use(
@@ -105,11 +87,7 @@ fn contract_early_proofs_srs_test_impl(test_dir: &'static str) -> Result<(), lib
PathBuf::from(test_dir),
|api, m| {
// Receive wallet calls --receive=5
let args = &mut ContractSetupArgsAPI {
selection_args: common::contract_selection_args(),
net_change: Some(5_000_000_000),
..Default::default()
};
let args = &mut common::contract_setup_args(Some(5_000_000_000));
// Proofs are opt-in; enable and supply the sender address.
args.proof_args.suppress_proof = false;
args.proof_args.memo = Some(PaymentMemo::new("SRS payment".into())?);
@@ -120,7 +98,7 @@ fn contract_early_proofs_srs_test_impl(test_dir: &'static str) -> Result<(), lib
},
)?;
assert_eq!(slate.state, SlateState::Standard2);
slate = roundtrip_slate(&slate, 5)?;
slate = common::roundtrip_slate(&slate, 5)?;
// Send wallet finalizes and posts
//let mut sender_part_sig = None;
@@ -129,10 +107,7 @@ fn contract_early_proofs_srs_test_impl(test_dir: &'static str) -> Result<(), lib
send_mask,
PathBuf::from(test_dir),
|api, m| {
let args = &ContractSetupArgsAPI {
selection_args: common::contract_selection_args(),
..Default::default()
};
let args = &common::contract_setup_args(None);
let mut tampered = slate.clone();
let proof = tampered.payment_proof.as_mut().unwrap();
*proof.timestamp.as_mut().unwrap() += Duration::from_secs(1);
@@ -143,7 +118,7 @@ fn contract_early_proofs_srs_test_impl(test_dir: &'static str) -> Result<(), lib
},
)?;
assert_eq!(slate.state, SlateState::Standard3);
slate = roundtrip_slate(&slate, 5)?;
slate = common::roundtrip_slate(&slate, 5)?;
wallet::controller::owner_single_use(
send_wallet.clone(),
@@ -177,11 +152,11 @@ fn contract_early_proofs_srs_test_impl(test_dir: &'static str) -> Result<(), lib
assert_eq!(tx_log.amount_debited, 0);
assert_eq!(tx_log.num_inputs, 1);
assert_eq!(tx_log.num_outputs, 1);
let expected_fees_paid = Some(my_fee_contribution(1, 1, 1, 2)?);
assert_eq!(tx_log.fee, expected_fees_paid);
let expected_fees_paid = my_fee_contribution(1, 1, 1, 2)?;
assert_eq!(tx_log.fee, Some(expected_fees_paid));
assert_eq!(
wallet_info.amount_currently_spendable,
4 * 60_000_000_000 + 5_000_000_000 - expected_fees_paid.unwrap().fee() // we expect the balance of 4 mined blocks + 5 Grin - fees paid
4 * 60_000_000_000 + 5_000_000_000 - expected_fees_paid.fee() // we expect the balance of 4 mined blocks + 5 Grin - fees paid
);
Ok(())
},
controller/tests/contract_rsr.rs
+3 -3
View File
@@ -228,11 +228,11 @@ fn contract_rsr_tx_impl(test_dir: &'static str) -> Result<(), libwallet::Error>
assert_eq!(tx_log.amount_debited, 0);
assert_eq!(tx_log.num_inputs, 1);
assert_eq!(tx_log.num_outputs, 1);
let expected_fees_paid = Some(my_fee_contribution(1, 1, 1, 2)?);
assert_eq!(tx_log.fee, expected_fees_paid);
let expected_fees_paid = my_fee_contribution(1, 1, 1, 2)?;
assert_eq!(tx_log.fee, Some(expected_fees_paid));
assert_eq!(
wallet_info.amount_currently_spendable,
4 * 60_000_000_000 + 5_000_000_000 - expected_fees_paid.unwrap().fee() // we expect the balance of 4 mined blocks + 5 Grin - fees paid
4 * 60_000_000_000 + 5_000_000_000 - expected_fees_paid.fee() // we expect the balance of 4 mined blocks + 5 Grin - fees paid
);
Ok(())
},
controller/tests/contract_srs.rs
-1
View File
@@ -95,7 +95,6 @@ fn contract_srs_tx_impl(test_dir: &'static str) -> Result<(), libwallet::Error>
net_change: Some(-5_000_000_000),
..Default::default()
},
..Default::default()
};
let progress = common::wallet_progress(api, m)?;
for count in [0, 3] {