Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat(FI-913): add more icrc2_approve tests #150

Merged
merged 13 commits into from
Sep 4, 2023
Merged

feat(FI-913): add more icrc2_approve tests #150

Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
980777f
test approvals with expiration
maciejdfinity Aug 25, 2023
e42f8af
clean up approve test
maciejdfinity Aug 28, 2023
86576fe
test for expected_allowance
maciejdfinity Aug 28, 2023
520e716
test for insufficient funds, clean up tests
maciejdfinity Aug 28, 2023
be94210
test approval for subaccount
maciejdfinity Aug 28, 2023
fe11ebb
use smaller amounts for approvals
maciejdfinity Aug 28, 2023
28282fc
clippy fix, test fix
maciejdfinity Aug 28, 2023
cb50a8e
remove constants from tests
maciejdfinity Aug 28, 2023
8bf8b40
remove constants from tests
maciejdfinity Aug 28, 2023
e4a27a1
change expired to now - 1
maciejdfinity Aug 28, 2023
c74bfaa
Merge branch 'main' into maciej-icrc2-tests
maciejdfinity Aug 29, 2023
1331a2c
use bail instead of returning error
maciejdfinity Aug 29, 2023
130804b
Merge branch 'main' into maciej-icrc2-tests
maciejdfinity Sep 4, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
251 changes: 224 additions & 27 deletions test/suite/lib.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ use icrc1_test_env::icrc2::{allowance, approve, transfer_from};
use icrc1_test_env::ApproveArgs;
use icrc1_test_env::TransferFromArgs;
use icrc1_test_env::{Account, LedgerEnv, Transfer, TransferError, Value};
use icrc1_test_env::{AllowanceArgs, TransferFromError};
use icrc1_test_env::{AllowanceArgs, ApproveError, TransferFromError};
use std::future::Future;
use std::pin::Pin;
use std::time::SystemTime;
Expand Down Expand Up @@ -55,6 +55,14 @@ fn assert_not_equal<T: PartialEq + std::fmt::Debug>(lhs: T, rhs: T) -> anyhow::R
Ok(())
}

fn time_nanos(ledger_env: &impl LedgerEnv) -> u64 {
ledger_env
.time()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_nanos() as u64
}

async fn assert_balance(
ledger: &impl LedgerEnv,
account: impl Into<Account>,
Expand All @@ -77,10 +85,14 @@ async fn assert_balance(

async fn assert_allowance(
ledger_env: &impl LedgerEnv,
from: Account,
spender: Account,
expected_allowance: Nat,
from: impl Into<Account>,
spender: impl Into<Account>,
expected_allowance: impl Into<Nat>,
expires_at: Option<u64>,
) -> anyhow::Result<()> {
let from: Account = from.into();
let spender: Account = spender.into();
let expected_allowance: Nat = expected_allowance.into();
let allowance = allowance(
ledger_env,
AllowanceArgs {
Expand All @@ -98,6 +110,9 @@ async fn assert_allowance(
allowance.allowance
);
}
if allowance.expires_at != expires_at {
bail!("Approval {:?} -> {:?} , wrong expiration", from, spender,);
}
Ok(())
}

Expand Down Expand Up @@ -268,35 +283,215 @@ pub async fn icrc2_test_supported_standards(ledger: impl LedgerEnv) -> anyhow::R

pub async fn icrc2_test_approve(ledger_env: impl LedgerEnv) -> anyhow::Result<Outcome> {
let fee = transfer_fee(&ledger_env).await?;
let transfer_amount = Nat::from(10_000);
let initial_balance: Nat = transfer_amount.clone() + fee.clone() * 2;
let initial_balance: Nat = fee.clone() * 2;
let p1_env = setup_test_account(&ledger_env, initial_balance.clone()).await?;
let p2_env = ledger_env.fork();
let approve_amount = transfer_amount.clone() + fee.clone();
let p2_subaccount = Account {
owner: p2_env.principal(),
subaccount: Some([1; 32]),
};
let approve_amount = fee.clone();

approve(
&p1_env,
ApproveArgs::approve_amount(approve_amount.clone(), p2_env.principal()),
)
.await??;

assert_balance(
&ledger_env,
assert_allowance(
&p1_env,
p1_env.principal(),
p2_env.principal(),
approve_amount.clone(),
None,
)
.await?;

assert_allowance(&p1_env, p1_env.principal(), p2_subaccount.clone(), 0, None).await?;

assert_balance(&ledger_env, p1_env.principal(), fee.clone()).await?;
assert_balance(&ledger_env, p2_env.principal(), 0).await?;
assert_balance(&ledger_env, p2_subaccount.clone(), 0).await?;

// Approval for a subaccount.
approve(
&p1_env,
ApproveArgs::approve_amount(approve_amount.clone() * 2, p2_subaccount.clone()),
maciejdfinity marked this conversation as resolved.
Show resolved Hide resolved
)
.await??;

assert_allowance(
&p1_env,
p1_env.principal(),
p2_env.principal(),
approve_amount.clone(),
None,
)
.await?;

assert_allowance(
&p1_env,
p1_env.principal(),
p2_subaccount.clone(),
approve_amount.clone() * 2,
None,
)
.await?;

assert_balance(&ledger_env, p1_env.principal(), 0).await?;
assert_balance(&ledger_env, p2_env.principal(), 0).await?;
assert_balance(&ledger_env, p2_subaccount, 0).await?;

// Insufficient funds to pay the fee for a second approval
match approve(
&p1_env,
ApproveArgs::approve_amount(fee.clone() * 3, p2_env.principal()),
)
.await?
{
Ok(_) => bail!("expected ApproveError::InsufficientFunds, got Ok result"),
Err(e) => match e {
ApproveError::InsufficientFunds { balance } => {
if balance != 0 {
bail!("wrong balance, expected 0, got: {}", balance);
}
}
_ => return Err(e).context("expected ApproveError::InsufficientFunds"),
},
}

Ok(Outcome::Passed)
}

pub async fn icrc2_test_approve_expiration(ledger_env: impl LedgerEnv) -> anyhow::Result<Outcome> {
let fee = transfer_fee(&ledger_env).await?;
let initial_balance: Nat = fee.clone() * 2;
let p1_env = setup_test_account(&ledger_env, initial_balance.clone()).await?;
let p2_env = ledger_env.fork();
let approve_amount = fee.clone();
let now = time_nanos(&ledger_env);

// Expiration in the past
match approve(
&p1_env,
ApproveArgs::approve_amount(approve_amount.clone(), p2_env.principal()).expires_at(now - 1),
)
.await?
{
Ok(_) => bail!("expected ApproveError::Expired, got Ok result"),
Err(e) => match e {
ApproveError::Expired { .. } => {}
_ => return Err(e).context("expected ApproveError::Expired"),
},
}

assert_allowance(&p1_env, p1_env.principal(), p2_env.principal(), 0, None).await?;

assert_balance(&ledger_env, p1_env.principal(), initial_balance.clone()).await?;
assert_balance(&ledger_env, p2_env.principal(), 0).await?;

// Correct expiration in the future
let expiration = u64::MAX;
approve(
&p1_env,
ApproveArgs::approve_amount(approve_amount.clone(), p2_env.principal())
.expires_at(expiration),
)
.await??;

assert_allowance(
&p1_env,
p1_env.principal(),
initial_balance.clone() - fee.clone(),
p2_env.principal(),
approve_amount.clone(),
Some(expiration),
)
.await?;

assert_balance(&ledger_env, p1_env.principal(), fee).await?;
assert_balance(&ledger_env, p2_env.principal(), 0).await?;

// Change expiration
let new_expiration = expiration - 1;
approve(
&p1_env,
ApproveArgs::approve_amount(approve_amount.clone(), p2_env.principal())
.expires_at(new_expiration),
)
.await??;

assert_allowance(
&p1_env,
p1_env.principal().into(),
p2_env.principal().into(),
p1_env.principal(),
p2_env.principal(),
approve_amount,
Some(new_expiration),
)
.await?;

assert_balance(&ledger_env, p1_env.principal(), 0).await?;
assert_balance(&ledger_env, p2_env.principal(), 0).await?;

Ok(Outcome::Passed)
}

pub async fn icrc2_test_approve_expected_allowance(
ledger_env: impl LedgerEnv,
) -> anyhow::Result<Outcome> {
let fee = transfer_fee(&ledger_env).await?;
let initial_balance: Nat = fee.clone() * 2;
let p1_env = setup_test_account(&ledger_env, initial_balance.clone()).await?;
let p2_env = ledger_env.fork();
let approve_amount = fee.clone();

approve(
&p1_env,
ApproveArgs::approve_amount(approve_amount.clone(), p2_env.principal()),
)
.await??;

// Wrong expected allowance
let new_approve_amount: Nat = fee.clone() * 2;
match approve(
&p1_env,
ApproveArgs::approve_amount(new_approve_amount.clone(), p2_env.principal())
.expected_allowance(fee.clone() * 2),
)
.await?
{
Ok(_) => bail!("expected ApproveError::AllowanceChanged, got Ok result"),
Err(e) => match e {
ApproveError::AllowanceChanged { current_allowance } => {
if current_allowance != approve_amount {
bail!(
"wrong current_allowance, expected {}, got: {}",
approve_amount,
current_allowance
);
}
}
_ => return Err(e).context("expected ApproveError::AllowanceChanged"),
},
}

// Correct expected allowance
approve(
&p1_env,
ApproveArgs::approve_amount(new_approve_amount.clone(), p2_env.principal())
.expected_allowance(approve_amount),
)
.await??;

assert_allowance(
&p1_env,
p1_env.principal(),
p2_env.principal(),
new_approve_amount,
None,
)
.await?;

assert_balance(&ledger_env, p1_env.principal(), initial_balance - fee).await?;
assert_balance(&ledger_env, p1_env.principal(), 0).await?;
assert_balance(&ledger_env, p2_env.principal(), 0).await?;

Ok(Outcome::Passed)
Expand Down Expand Up @@ -346,9 +541,10 @@ pub async fn icrc2_test_transfer_from(ledger_env: impl LedgerEnv) -> anyhow::Res

assert_allowance(
&p1_env,
p1_env.principal().into(),
p2_env.principal().into(),
p1_env.principal(),
p2_env.principal(),
Nat::from(1),
None,
)
.await?;
Ok(Outcome::Passed)
Expand Down Expand Up @@ -406,9 +602,10 @@ pub async fn icrc2_test_transfer_from_insufficient_funds(
// Allowance is not changed.
assert_allowance(
&p1_env,
p1_env.principal().into(),
p2_env.principal().into(),
p1_env.principal(),
p2_env.principal(),
approve_amount,
None,
)
.await?;

Expand Down Expand Up @@ -489,14 +686,6 @@ pub async fn icrc1_test_tx_deduplication(ledger_env: impl LedgerEnv) -> anyhow::
let p1_env = setup_test_account(&ledger_env, initial_balance.clone()).await?;
let p2_env = p1_env.fork();

let time_nanos = || {
ledger_env
.time()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_nanos() as u64
};

// Deduplication should not happen if the created_at_time field is unset.
let transfer_args = Transfer::amount_to(transfer_amount.clone(), p2_env.principal());
transfer(&p1_env, transfer_args.clone())
Expand All @@ -513,7 +702,7 @@ pub async fn icrc1_test_tx_deduplication(ledger_env: impl LedgerEnv) -> anyhow::

// Setting the created_at_time field changes the transaction
// identity, so the transfer should succeed.
let transfer_args = transfer_args.created_at_time(time_nanos());
let transfer_args = transfer_args.created_at_time(time_nanos(&ledger_env));

let txid = match transfer(&p1_env, transfer_args.clone()).await? {
Ok(txid) => txid,
Expand Down Expand Up @@ -582,7 +771,7 @@ pub async fn icrc1_test_tx_deduplication(ledger_env: impl LedgerEnv) -> anyhow::

assert_balance(&p1_env, p2_env.principal(), transfer_amount.clone() * 5).await?;

let now = time_nanos();
let now = time_nanos(&ledger_env);

// Transactions with different subaccounts (even if it's None and
// Some([0; 32])) should not be considered duplicates.
Expand Down Expand Up @@ -723,6 +912,14 @@ pub fn icrc2_test_suite(env: impl LedgerEnv + 'static + Clone) -> Vec<Test> {
icrc2_test_supported_standards(env.clone()),
),
test("icrc2:approve", icrc2_test_approve(env.clone())),
test(
"icrc2:approve_expiration",
icrc2_test_approve_expiration(env.clone()),
),
test(
"icrc2:approve_expected_allowance",
icrc2_test_approve_expected_allowance(env.clone()),
),
test("icrc2:transfer_from", icrc2_test_transfer_from(env.clone())),
test(
"icrc2:transfer_from_insufficient_funds",
Expand Down
Loading