fix tests

0xPolygonMiden · Apr 9, 2024 · e53838b · e53838b
1 parent 49daa9c
commit e53838b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/miden-tx/tests/integration/scripts/faucet.rs b/miden-tx/tests/integration/scripts/faucet.rs
@@ -6,7 +6,7 @@ use miden_lib::{
 use miden_objects::{
     accounts::{
         Account, AccountCode, AccountId, AccountStorage, SlotItem, StorageSlot,
-        ACCOUNT_ID_FUNGIBLE_FAUCET_OFF_CHAIN,
+        ACCOUNT_ID_FUNGIBLE_FAUCET_OFF_CHAIN, AccountStorageType,
     },
     assembly::{ModuleAst, ProgramAst},
     assets::{Asset, AssetVault, FungibleAsset, TokenSymbol},
@@ -242,9 +242,10 @@ fn faucet_contract_creation() {
     let token_symbol_string = "POL";
     let token_symbol = TokenSymbol::try_from(token_symbol_string).unwrap();
     let decimals = 2u8;
+    let storage_type = AccountStorageType::OffChain;
 
     let (faucet_account, _) =
-        create_basic_fungible_faucet(init_seed, token_symbol, decimals, max_supply, auth_scheme)
+        create_basic_fungible_faucet(init_seed, token_symbol, decimals, max_supply, auth_scheme, storage_type)
             .unwrap();
 
     // check that max_supply (slot 1) is 123

diff --git a/miden-tx/tests/integration/wallet/mod.rs b/miden-tx/tests/integration/wallet/mod.rs
@@ -190,7 +190,7 @@ fn prove_send_asset_via_wallet() {
 #[cfg(not(target_arch = "wasm32"))]
 #[test]
 fn wallet_creation() {
-    use miden_objects::accounts::{AccountType, ACCOUNT_ID_SENDER};
+    use miden_objects::accounts::{AccountType, ACCOUNT_ID_SENDER, AccountStorageType};
 
     // we need a Falcon Public Key to create the wallet account
     let seed = [0_u8; 32];
@@ -206,8 +206,11 @@ fn wallet_creation() {
         204, 149, 90, 166, 68, 100, 73, 106, 168, 125, 237, 138, 16,
     ];
 
+    let account_type = AccountType::RegularAccountImmutableCode;
+    let storage_type = AccountStorageType::OffChain;
+
     let (wallet, _) =
-        create_basic_wallet(init_seed, auth_scheme, AccountType::RegularAccountImmutableCode)
+        create_basic_wallet(init_seed, auth_scheme, account_type, storage_type)
             .unwrap();
 
     // sender_account_id not relevant here, just to create a default account code