From e1f22a796b9feb25be6daa4e8252ef3af4c6ced5 Mon Sep 17 00:00:00 2001
From: gianmalarcon <gian@quantum3labs.com>
Date: Wed, 14 Aug 2024 19:38:40 +0700
Subject: [PATCH] Fix: Add test

---
 .../contracts/src/test/TestContract.cairo     | 61 +++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/packages/snfoundry/contracts/src/test/TestContract.cairo b/packages/snfoundry/contracts/src/test/TestContract.cairo
index 8b137891..f3a95df7 100644
--- a/packages/snfoundry/contracts/src/test/TestContract.cairo
+++ b/packages/snfoundry/contracts/src/test/TestContract.cairo
@@ -1 +1,62 @@
+use contracts::YourCollectible::{
+    IYourCollectibleDispatcher, IYourCollectibleDispatcherTrait, IERC721Dispatcher,
+    IERC721DispatcherTrait, IERC721EnumerableDispatcher, IERC721EnumerableDispatcherTrait
+};
 
+use contracts::mock_contracts::Receiver;
+use openzeppelin::tests::utils;
+use openzeppelin::utils::serde::SerializedAppend;
+use snforge_std::{declare, ContractClassTrait};
+use starknet::ContractAddress;
+use starknet::contract_address_const;
+
+// Should deploy the contract
+fn deploy_contract(name: ByteArray) -> ContractAddress {
+    let contract = declare(name).unwrap();
+    let mut calldata = array![];
+    calldata.append_serde(OWNER());
+    let (contract_address, _) = contract.deploy(@calldata).unwrap();
+    println!("Contract deployed on: {:?}", contract_address);
+    contract_address
+}
+
+fn OWNER() -> ContractAddress {
+    contract_address_const::<'OWNER'>()
+}
+
+fn deploy_receiver() -> ContractAddress {
+    let contract = declare("Receiver").unwrap();
+    let mut calldata = array![];
+    let (contract_address, _) = contract.deploy(@calldata).unwrap();
+    println!("Receiver deployed on: {:?}", contract_address);
+    contract_address
+}
+
+#[test]
+fn test_mint_item() {
+    // Should be able to mint an NFT
+    let contract_address = deploy_contract("YourCollectible");
+    let dispatcher = IYourCollectibleDispatcher { contract_address };
+    let erc721 = IERC721Dispatcher { contract_address };
+    let tester_address = deploy_receiver();
+    println!("Tester address: {:?}", tester_address);
+    let starting_balance = erc721.balance_of(tester_address);
+    println!("Starting balance: {:?}", starting_balance);
+    println!("Minting...");
+    let url: ByteArray = "QmfVMAmNM1kDEBYrC2TPzQDoCRFH6F5tE1e9Mr4FkkR5Xr";
+    let token_id = dispatcher.mint_item(tester_address, url);
+    let expected_token_id = 1;
+    assert(token_id == expected_token_id, 'Token ID must be 1');
+    println!("Item minted! Token ID: {:?}", token_id);
+    let new_balance = erc721.balance_of(tester_address);
+    assert(new_balance == starting_balance + 1, 'Balance must be increased by 1');
+    println!("New balance: {:?}", new_balance);
+
+    // Should track tokens of owner by index
+    let erc721Enumerable = IERC721EnumerableDispatcher { contract_address };
+    let index = new_balance - 1;
+    let token = erc721Enumerable.token_of_owner_by_index(tester_address, index);
+    println!("token of owner by index {:?}", token);
+    assert(token > 0, 'Token must be greater than zero');
+    println!("Token of owner({:?}) by index({:?}): {:?}", tester_address, index, token);
+}