From ff93699ffee456eb30c32e304c31c8b20669a56b Mon Sep 17 00:00:00 2001
From: Nerrolol <nassim.amerouali@gmail.com>
Date: Wed, 20 Mar 2024 19:29:27 +0000
Subject: [PATCH] Add airdrop tests

---
 tests/airdrop_tests.cairo | 99 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 99 insertions(+)
 create mode 100644 tests/airdrop_tests.cairo

diff --git a/tests/airdrop_tests.cairo b/tests/airdrop_tests.cairo
new file mode 100644
index 00000000..d2483a7a
--- /dev/null
+++ b/tests/airdrop_tests.cairo
@@ -0,0 +1,99 @@
+use core::hash::HashStateExTrait;
+use core::{ArrayTrait, SpanTrait};
+use core::debug::PrintTrait;
+use distributor::contract::{Distributor, IDistributorDispatcher, IDistributorDispatcherTrait};
+use Distributor::STRK_ADDRESS;
+use openzeppelin::token::erc20::interface::{IERC20Dispatcher, IERC20DispatcherTrait};
+use snforge_std::{ContractClassTrait, declare, start_prank, CheatTarget};
+use starknet::{ContractAddress, deploy_syscall};
+
+const ADMIN_ADDR: felt252 = 0x42;
+const CLAIMEE_1: felt252 = 0x13;
+const CLAIMEE_2: felt252 = 0x14;
+
+fn deploy() -> IDistributorDispatcher {
+    let mut calldata = ArrayTrait::new();
+    calldata.append(ADMIN_ADDR);
+
+    let contract = declare('Distributor');
+    let address = contract.deploy(@calldata).expect('unable to deploy distributor');
+
+    IDistributorDispatcher { contract_address: address }
+}
+
+fn deploy_token(recipient: ContractAddress) -> IERC20Dispatcher {
+    let mut calldata = ArrayTrait::new();
+    calldata.append(1000000000000000000);
+    calldata.append(0);
+    calldata.append(recipient.into());
+    let contract = declare('MyToken');
+    let address = contract
+        .deploy_at(@calldata, STRK_ADDRESS.try_into().unwrap())
+        .expect('unable to deploy mockstrk');
+
+    IERC20Dispatcher { contract_address: address }
+}
+
+#[test]
+fn test_claim_twice_with_same_proof() {
+    let contract = deploy();
+    let tok = deploy_token(contract.contract_address);
+    
+    start_prank(CheatTarget::One(contract.contract_address), ADMIN_ADDR.try_into().unwrap());
+    contract.add_root(valid_root);
+    
+    start_prank(CheatTarget::One(contract.contract_address), CLAIMEE_1.try_into().unwrap());
+    let initial_proof = array![valid_proof_element];
+    contract.claim(valid_claim_amount, initial_proof.span());
+    assert(tok.balance_of(CLAIMEE_1.try_into().unwrap()) == valid_claim_amount, "First claim failed");
+    
+    
+    contract.claim(valid_claim_amount, initial_proof.span());   
+    assert(tok.balance_of(CLAIMEE_1.try_into().unwrap()) == valid_claim_amount, "Second claimed modified the claimee's balance");
+}
+
+#[test]
+#[should_panic(expected: ('INVALID PROOF',))]
+fn test_claim_invalid_proof() {
+    let contract = deploy();
+    deploy_token(contract.contract_address);
+    start_prank(CheatTarget::One(contract.contract_address), ADMIN_ADDR.try_into().unwrap());
+    contract.add_root(0xf7c8d3f309262572ad35df8ff6c33f24d8114c60eac3bc27bf42382ca82faf);
+
+    start_prank(
+        CheatTarget::One(contract.contract_address), CLAIMEE_1.try_into().unwrap()
+    );
+    let proof = array![
+        0x2a18afb0550a011d54ca3940648e59894c06e4c3d0a611256c0b575bd528b3b,
+        0x1
+    ];
+    contract.claim(0x88, proof.span());
+}
+
+#[test]
+fn test_update_root_and_claim_attempts() {
+    let contract = deploy();
+    let tok = deploy_token(contract.contract_address);
+    start_prank(CheatTarget::One(contract.contract_address), ADMIN_ADDR.try_into().unwrap());
+
+    // add intial root and valid claim
+    contract.add_root(initial_root);
+    start_prank(CheatTarget::One(contract.contract_address), CLAIMEE_1.try_into().unwrap());
+    contract.claim(claim_amount, valid_proof_for_initial_root.span());
+    assert(tok.balance_of(CLAIMEE_1.try_into().unwrap()) == claim_amount, 'initial claim failed');
+
+    // update root
+    start_prank(CheatTarget::One(contract.contract_address), ADMIN_ADDR.try_into().unwrap());
+    contract.add_root(new_root);
+
+    // claim with old root + new proof, should fail
+    start_prank(CheatTarget::One(contract.contract_address), CLAIMEE_2.try_into().unwrap());
+    contract.claim(claim_amount, new_proof_not_matching_old_root.span());
+    // check fail : to do, use should panic ?
+
+    // claim with new root + old proof, should fail 
+    contract.claim(claim_amount, old_proof_not_matching_new_root.span());
+    // check fail : to do
+}
+
+