Skip to content

Commit e68abc5

Browse files
committed
fix(contracts): align quota reset snapshot test with real flow
1 parent 20748ee commit e68abc5

2 files changed

Lines changed: 13 additions & 20 deletions

File tree

stellar-contracts/src/test.rs

Lines changed: 12 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -3020,24 +3020,17 @@ fn test_event_snapshot_quota_reset() {
30203020
let (contract_id, bridge, admin, token_addr, _, token_sac) = setup_bridge(&env, 10_000);
30213021
let user = Address::generate(&env);
30223022

3023-
token_sac.mint(&contract_id, &1_000);
3024-
env.as_contract(&contract_id, || {
3025-
env.storage()
3026-
.instance()
3027-
.set(&DataKey::UserDeposited(user.clone()), &1_000i128);
3028-
env.storage()
3029-
.instance()
3030-
.set(&DataKey::WithdrawalQuota, &500i128);
3031-
env.storage().instance().set(
3032-
&DataKey::UserDailyWithdrawal(user.clone()),
3033-
&UserDailyWithdrawal {
3034-
amount: 500,
3035-
window_start: 100,
3036-
},
3037-
);
3038-
});
3023+
token_sac.mint(&user, &5_000);
3024+
bridge.deposit(&user, &2_000, &token_addr, &Bytes::new(&env), &0, &0, &None);
3025+
bridge.set_withdrawal_quota(&500);
3026+
bridge.withdraw(&admin, &user, &500, &token_addr);
3027+
3028+
let result = bridge.try_withdraw(&admin, &user, &100, &token_addr);
3029+
assert_eq!(result, Err(Ok(Error::WithdrawalQuotaExceeded)));
3030+
3031+
let start_ledger = env.ledger().sequence();
30393032
env.ledger().with_mut(|li| {
3040-
li.sequence_number = 17_380;
3033+
li.sequence_number = start_ledger + 17_280;
30413034
});
30423035

30433036
bridge.withdraw(&admin, &user, &500, &token_addr);
@@ -3053,7 +3046,7 @@ fn test_event_snapshot_quota_reset() {
30533046
EVENT_VERSION.into_val(&env),
30543047
Symbol::new(&env, "quota_reset").into_val(&env)
30553048
],
3056-
(user.clone(), 17_380u32).into_val(&env)
3049+
(user.clone(), start_ledger + 17_280).into_val(&env)
30573050
),
30583051
(
30593052
contract_id,
@@ -3073,7 +3066,7 @@ fn test_event_snapshot_quota_reset() {
30733066
&[
30743067
SnapshotEvent {
30753068
topics: StdVec::from(["u32:1".into(), "symbol:quota_reset".into()]),
3076-
data: "tuple:[address:user,u32:17380]".into(),
3069+
data: "tuple:[address:user,u32:17281]".into(),
30773070
},
30783071
SnapshotEvent {
30793072
topics: StdVec::from([

stellar-contracts/test_snapshots/events/quota_reset.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
"u32:1",
66
"symbol:quota_reset"
77
],
8-
"data": "tuple:[address:user,u32:17380]"
8+
"data": "tuple:[address:user,u32:17281]"
99
},
1010
{
1111
"contract": "bridge",

0 commit comments

Comments
 (0)