feat: added unit tests for deposit and withdrawal functionalities.

contracts/.gitignore

@@ -1,2 +1,3 @@
 target
-notes.md
+notes.md
+.snfoundry_cache

contracts/Scarb.lock

@@ -5,9 +5,116 @@ version = 1
 name = "contract_strapex"
 version = "0.1.0"
 dependencies = [
+ "openzeppelin",
  "snforge_std",
 ]
 
+[[package]]
+name = "openzeppelin"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+dependencies = [
+ "openzeppelin_access",
+ "openzeppelin_account",
+ "openzeppelin_finance",
+ "openzeppelin_governance",
+ "openzeppelin_introspection",
+ "openzeppelin_merkle_tree",
+ "openzeppelin_presets",
+ "openzeppelin_security",
+ "openzeppelin_token",
+ "openzeppelin_upgrades",
+ "openzeppelin_utils",
+]
+
+[[package]]
+name = "openzeppelin_access"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+dependencies = [
+ "openzeppelin_introspection",
+ "openzeppelin_utils",
+]
+
+[[package]]
+name = "openzeppelin_account"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+dependencies = [
+ "openzeppelin_introspection",
+ "openzeppelin_utils",
+]
+
+[[package]]
+name = "openzeppelin_finance"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+dependencies = [
+ "openzeppelin_access",
+ "openzeppelin_token",
+]
+
+[[package]]
+name = "openzeppelin_governance"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+dependencies = [
+ "openzeppelin_access",
+ "openzeppelin_account",
+ "openzeppelin_introspection",
+ "openzeppelin_token",
+]
+
+[[package]]
+name = "openzeppelin_introspection"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+
+[[package]]
+name = "openzeppelin_merkle_tree"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+
+[[package]]
+name = "openzeppelin_presets"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+dependencies = [
+ "openzeppelin_access",
+ "openzeppelin_account",
+ "openzeppelin_finance",
+ "openzeppelin_introspection",
+ "openzeppelin_token",
+ "openzeppelin_upgrades",
+ "openzeppelin_utils",
+]
+
+[[package]]
+name = "openzeppelin_security"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+
+[[package]]
+name = "openzeppelin_token"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+dependencies = [
+ "openzeppelin_access",
+ "openzeppelin_account",
+ "openzeppelin_introspection",
+ "openzeppelin_utils",
+]
+
+[[package]]
+name = "openzeppelin_upgrades"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+
+[[package]]
+name = "openzeppelin_utils"
+version = "0.19.0"
+source = "git+https://github.com/OpenZeppelin/cairo-contracts.git?tag=v0.19.0#8d49e8c445efd9bdc99b050c8b7d11ae5ad19628"
+
 [[package]]
 name = "snforge_scarb_plugin"
 version = "0.33.0"

contracts/Scarb.toml

@@ -6,9 +6,16 @@ edition = "2023_10"
 # See more keys and their definitions at https://docs.swmansion.com/scarb/docs/reference/manifest.html
 
 [dependencies]
+starknet = "2.8.5"
+openzeppelin = { git = "https://github.com/OpenZeppelin/cairo-contracts.git", tag = "v0.19.0" }
+
+[dev-dependencies]
 snforge_std = { git = "https://github.com/foundry-rs/starknet-foundry", tag = "v0.33.0" }
-starknet = "2.8.2"
+assert_macros = "2.8.5"
 
 [[target.starknet-contract]]
 casm = true
 sierra = true
+
+[scripts]
+test = "snforge test"

contracts/src/lib.cairo

@@ -1,3 +1,3 @@
-mod ownership_component;
-mod strapex_contract;
-mod strapex_factory;
+pub mod ownership_component;
+pub mod strapex_contract;
+pub mod strapex_factory;

contracts/src/strapex_contract.cairo

@@ -30,7 +30,7 @@ trait IStrapex<TContractState> {
 }
 
 #[starknet::contract]
-mod strapex_contract {
+pub mod StrapexContract {
     use contract_strapex::strapex_contract::IStrapex;
     use core::traits::Into;
     use core::box::BoxTrait;
@@ -75,15 +75,15 @@ mod strapex_contract {
 
     #[event]
     #[derive(Drop, starknet::Event)]
-    enum Event {
+    pub enum Event {
         OwnableEvent: ownable_component::Event,
         Deposit: Deposit,
         Withdraw: Withdraw,
         FeeCollection: FeeCollection,
     }
 
     #[derive(Drop, starknet::Event)]
-    struct Deposit {
+    pub struct Deposit {
         #[key]
         from: ContractAddress,
         #[key]
@@ -94,13 +94,13 @@ mod strapex_contract {
     }
 
     #[derive(Drop, starknet::Event)]
-    struct Withdraw {
+    pub struct Withdraw {
         #[key]
         Amount: u256,
     }
 
     #[derive(Drop, starknet::Event)]
-    struct FeeCollection {
+    pub struct FeeCollection {
         #[key]
         Amount: u256,
     }
@@ -322,4 +322,4 @@ mod strapex_contract {
             (caller, this, currentBalance)
         }
     }
-}
+}

contracts/src/strapex_factory.cairo

@@ -15,7 +15,7 @@ trait IStrapexFactory<TContractState> {
 }
 
 #[starknet::contract]
-mod strapex_factory {
+mod StrapexFactory {
     use core::traits::Into;
     use core::array::ArrayTrait;
     // Core StarkNet and contract utilities.
@@ -188,4 +188,4 @@ mod strapex_factory {
             currentHash
         }
     }
-}
+}

contracts/tests/lib.cairo

@@ -0,0 +1 @@
+mod test_strapex_contract;