add balance to test account

dojoengine · May 16, 2024 · 1b3c1d6 · 1b3c1d6
1 parent 085028d
commit 1b3c1d6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/katana/storage/provider/Cargo.toml b/crates/katana/storage/provider/Cargo.toml
@@ -21,15 +21,17 @@ futures = { workspace = true, optional = true }
 starknet = { workspace = true, optional = true }
 tokio = { workspace = true, optional = true }
 
+alloy-primitives = { workspace = true, optional = true }
 serde_json = { workspace = true, optional = true }
 
 [features]
 default = [ "fork", "in-memory" ]
 fork = [ "dep:futures", "dep:starknet", "dep:tokio", "in-memory" ]
 in-memory = [  ]
-test-utils = [ "dep:serde_json" ]
+test-utils = [ "dep:alloy-primitives", "dep:serde_json" ]
 
 [dev-dependencies]
+alloy-primitives.workspace = true
 katana-core.workspace = true
 katana-runner.workspace = true
 lazy_static.workspace = true

diff --git a/crates/katana/storage/provider/src/test_utils.rs b/crates/katana/storage/provider/src/test_utils.rs
@@ -1,5 +1,6 @@
 use std::sync::Arc;
 
+use alloy_primitives::U256;
 use katana_db::mdbx::{test_utils, DbEnvKind};
 use katana_primitives::block::{BlockHash, FinalityStatus};
 use katana_primitives::class::CompiledClass;
@@ -64,7 +65,7 @@ pub fn create_genesis_for_testing() -> Genesis {
     };
 
     // setup test account
-    let (_, account) = DevGenesisAccount::new(felt!("0x1"), class_hash);
+    let (_, account) = DevGenesisAccount::new_with_balance(felt!("0x1"), class_hash, U256::MAX);
     let account = GenesisAllocation::Account(GenesisAccountAlloc::DevAccount(account));
 
     let mut genesis = Genesis::default();