Patch for 0.1.0-pre.alpha.3

NEBYTE · Feb 7, 2025 · c6ebbc1 · c6ebbc1
1 parent edf0803
commit c6ebbc1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/core/api/v1/staking.rs b/src/core/api/v1/staking.rs
@@ -1,16 +1,19 @@
 use crate::core::staking::staking_module::StakingModule;
 use ed25519_dalek::SigningKey;
 use crate::core::canister::canister::{Canister, CanisterFunctionPayload};
+use crate::core::consensus::model::ConsensusEngine;
 
 pub fn stake_tokens(
     canister: &mut Canister,
+    consensus_engine: &mut ConsensusEngine,
     staking_module: &mut StakingModule,
     signing_key: &SigningKey,
     neuron_id: u64,
     amount: u64,
 ) -> Result<String, String> {
     canister.execute_function(CanisterFunctionPayload::Stake {
         staking_module,
+        consensus_engine,
         signing_key,
         neuron_id,
         amount,
@@ -19,13 +22,15 @@ pub fn stake_tokens(
 
 pub fn unstake_tokens(
     canister: &mut Canister,
+    consensus_engine: &mut ConsensusEngine,
     staking_module: &mut StakingModule,
     signing_key: &SigningKey,
     neuron_id: u64,
     amount: u64,
 ) -> Result<String, String> {
     canister.execute_function(CanisterFunctionPayload::Unstake {
         staking_module,
+        consensus_engine,
         signing_key,
         neuron_id,
         amount,

diff --git a/src/core/canister/canister.rs b/src/core/canister/canister.rs
@@ -112,7 +112,7 @@ impl Canister {
             }
             CanisterFunctionPayload::Stake {
                 staking_module,
-                consensus_engine
+                consensus_engine,
                 signing_key,
                 neuron_id,
                 amount,