From a05cec5841eb50eab3b207c5706da4ea5e1b2101 Mon Sep 17 00:00:00 2001 From: MSG <59928086+MSghais@users.noreply.github.com> Date: Wed, 3 Jul 2024 11:44:29 +0200 Subject: [PATCH] Test contract Balance (#199) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * test deposit and claim amount with gas or not + check gas amount panic * change request test balance * scarb fmt * escrow balance check before and after * scarb fmt * rename all balances vars for recipient sender and joy --------- Co-authored-by: Maciej Kamiński @ StarkWare --- onchain/src/social/deposit.cairo | 116 +++++++++++++++++++++++++++++++ 1 file changed, 116 insertions(+) diff --git a/onchain/src/social/deposit.cairo b/onchain/src/social/deposit.cairo index 06f7439f..e5807960 100644 --- a/onchain/src/social/deposit.cairo +++ b/onchain/src/social/deposit.cairo @@ -410,10 +410,126 @@ mod tests { stop_cheat_caller_address_global(); start_cheat_caller_address(escrow.contract_address, sender_address); + let sender_balance_before_deposit = erc20.balance_of(sender_address); + + // Deposit by sender to recipient + escrow.deposit(amount, erc20.contract_address, recipient_nostr_key, 0_u64); + let sender_balance_after_deposit = erc20.balance_of(sender_address); + start_cheat_caller_address(escrow.contract_address, recipient_address); + let escrow_balance_before_claim = erc20.balance_of(escrow.contract_address); + + // Recipient user claim deposit + let recipient_balance_before_claim = erc20.balance_of(recipient_address); escrow.claim(request, 0_u256); + + // Sender check + assert!( + sender_balance_before_deposit - amount == sender_balance_after_deposit, + "sender amount to deposit not send" + ); + + // Recipient check + + let recipient_balance_after_claim = erc20.balance_of(recipient_address); + assert!(recipient_balance_before_claim == 0, "recipient balance before claim != 0"); + assert!(recipient_balance_after_claim == amount, "recipient balance after claim != 0"); + + // Escrow balance + assert!(escrow_balance_before_claim == amount, "escrow before claim != amount"); + let escrow_balance_after_claim = erc20.balance_of(escrow.contract_address); + assert!(escrow_balance_after_claim == 0, "escrow balance after claim != 0"); + } + + #[test] + fn deposit_claim_gas_fee() { + let (request, recipient_nostr_key, sender_address, erc20, escrow) = request_fixture(); + + let recipient_address: ContractAddress = 678.try_into().unwrap(); + let joyboy_address: ContractAddress = 159.try_into().unwrap(); + let amount = 100_u256; + let gas_amount = 1_u256; + + let claim_gas_amount = Claim { + deposit_id: 1, + starknet_recipient: recipient_address, + gas_amount: gas_amount, + gas_token_address: erc20.contract_address + }; + + let request_gas_amount = SocialRequest { + content: claim_gas_amount, + sig: Signature { + r: 0x68e441c1f8756b5278c815cc110efb302c2a08bcf0349328ba7bd7683e8b0b29_u256, + s: 0xd592a5a5e9fc85334ab6801d6dde984c85d67fcd726fce38b9fb06874c25832e_u256 + }, + ..request + }; + + cheat_caller_address_global(sender_address); + erc20.approve(escrow.contract_address, amount); + stop_cheat_caller_address_global(); + + start_cheat_caller_address(escrow.contract_address, sender_address); + let sender_balance_before_deposit = erc20.balance_of(sender_address); + + escrow.deposit(amount, erc20.contract_address, recipient_nostr_key, 0_u64); + + let sender_balance_after_deposit = erc20.balance_of(sender_address); + + start_cheat_caller_address(escrow.contract_address, joyboy_address); + + let joyboy_balance_before_claim = erc20.balance_of(joyboy_address); + + // Sender check + assert!( + sender_balance_before_deposit - amount == sender_balance_after_deposit, + "sender deposit amount not send" + ); + + // Joyboy account claim user for recipient with gas fees paid by the claim deposit + let escrow_balance_before_claim = erc20.balance_of(escrow.contract_address); + let recipient_balance_before_claim = erc20.balance_of(recipient_address); + escrow.claim(request_gas_amount, gas_amount); + + // Recipient check + let recipient_balance_after_claim = erc20.balance_of(recipient_address); + assert!(recipient_balance_before_claim == 0, "recipient balance before claim != 0"); + assert!( + recipient_balance_after_claim == amount - gas_amount, + "recipient after claim != (amount - gas)" + ); + + // Check gas amount receive by Joyboy account + let joyboy_balance_after_claim = erc20.balance_of(joyboy_address); + assert!(joyboy_balance_before_claim == 0, "joy balance before claim != 0"); + assert!( + joyboy_balance_after_claim == gas_amount, "joyboy balance not equal gas amount received" + ); + + // Escrow balance + assert!(escrow_balance_before_claim == amount, "escrow before claim != amount deposit"); + let escrow_balance_after_claim = erc20.balance_of(escrow.contract_address); + assert!(escrow_balance_after_claim == 0, "escrow balance after claim != 0"); + } + + #[test] + #[should_panic(expected: "gas_amount to big")] + fn claim_incorrect_gas_amount() { + let (request, recipient_nostr_key, sender_address, erc20, escrow) = request_fixture(); + + let amount = 100_u256; + + cheat_caller_address_global(sender_address); + erc20.approve(escrow.contract_address, amount); + stop_cheat_caller_address_global(); + + start_cheat_caller_address(escrow.contract_address, sender_address); + escrow.deposit(amount, erc20.contract_address, recipient_nostr_key, 10_u64); + + escrow.claim(request, 1_u256); } #[test]