update lib

Author: tserg

src/lib.cairo

@@ -99,20 +99,20 @@ mod tests {
         mod test_controller;
         pub mod utils;
     }
-    // mod equalizer {
-    //     mod test_allocator;
-    //     mod test_equalizer;
-    //     pub mod utils;
-    // }
+    mod equalizer {
+        mod test_allocator;
+        mod test_equalizer;
+        pub mod utils;
+    }
     // mod external {
     //     mod test_ekubo;
     //     mod test_pragma_v2;
     //     pub mod utils;
     // }
-    // mod flash_mint {
-    //     mod test_flash_mint;
-    //     pub mod utils;
-    // }
+    mod flash_mint {
+        mod test_flash_mint;
+        pub mod utils;
+    }
     mod gate {
         mod test_gate;
         pub mod utils;

src/tests/flash_mint/test_flash_mint.cairo

@@ -1,21 +1,18 @@
 mod test_flash_mint {
     use core::num::traits::Zero;
     use opus::core::flash_mint::flash_mint as flash_mint_contract;
-    use opus::interfaces::IERC20::{IERC20Dispatcher, IERC20DispatcherTrait};
-    use opus::interfaces::IEqualizer::{IEqualizerDispatcher, IEqualizerDispatcherTrait};
-    use opus::interfaces::IFlashBorrower::{IFlashBorrowerDispatcher, IFlashBorrowerDispatcherTrait};
+    use opus::interfaces::IERC20::IERC20DispatcherTrait;
+    use opus::interfaces::IEqualizer::IEqualizerDispatcherTrait;
     use opus::interfaces::IFlashMint::{IFlashMintDispatcher, IFlashMintDispatcherTrait};
-    use opus::interfaces::IShrine::{IShrineDispatcher, IShrineDispatcherTrait};
+    use opus::interfaces::IShrine::IShrineDispatcherTrait;
     use opus::mock::flash_borrower::flash_borrower as flash_borrower_contract;
     use opus::tests::common;
     use opus::tests::equalizer::utils::equalizer_utils;
     use opus::tests::flash_mint::utils::flash_mint_utils;
     use opus::tests::shrine::utils::shrine_utils;
-    use snforge_std::{
-        CheatTarget, EventSpyAssertionsTrait, spy_events, start_cheat_caller_address, stop_cheat_caller_address,
-    };
+    use snforge_std::{EventSpyAssertionsTrait, spy_events, start_cheat_caller_address, stop_cheat_caller_address};
     use starknet::ContractAddress;
-    use wadray::{SignedWad, WAD_ONE, Wad};
+    use wadray::{WAD_ONE, Wad};
 
     //
     // Tests

src/tests/flash_mint/utils.cairo

@@ -2,13 +2,14 @@ pub mod flash_mint_utils {
     use access_control::{IAccessControlDispatcher, IAccessControlDispatcherTrait};
     use core::num::traits::Zero;
     use opus::core::roles::shrine_roles;
-    use opus::interfaces::IFlashMint::{IFlashMintDispatcher, IFlashMintDispatcherTrait};
+    use opus::interfaces::IFlashMint::IFlashMintDispatcher;
     use opus::interfaces::IShrine::{IShrineDispatcher, IShrineDispatcherTrait};
-    use opus::mock::flash_borrower::flash_borrower as flash_borrower_contract;
     use opus::tests::shrine::utils::shrine_utils;
-    use snforge_std::{CheatTarget, ContractClassTrait, declare, start_cheat_caller_address, stop_cheat_caller_address};
-    use starknet::{ContractAddress, SyscallResultTrait};
-    use wadray::{WAD_ONE, Wad};
+    use snforge_std::{
+        ContractClassTrait, DeclareResultTrait, declare, start_cheat_caller_address, stop_cheat_caller_address,
+    };
+    use starknet::ContractAddress;
+    use wadray::WAD_ONE;
 
     pub const YIN_TOTAL_SUPPLY: u128 = 20000000000000000000000; // 20000 * WAD_ONE
     pub const DEFAULT_MINT_AMOUNT: u256 = 500000000000000000000; // 500 * WAD_ONE
@@ -48,7 +49,7 @@ pub mod flash_mint_utils {
         );
 
         // Mint some yin in shrine
-        start_cheat_caller_address(CheatTarget::All, shrine_utils::admin());
+        start_cheat_caller_address(shrine, shrine_utils::admin());
         shrine_dispatcher.inject(Zero::zero(), YIN_TOTAL_SUPPLY.into());
         (shrine, flashmint)
     }