forward 2

iotaledger · Nov 24, 2023 · 8c6743a · 8c6743a
1 parent 3fa452a
commit 8c6743a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/sdk/src/types/block/unlock/multi.rs b/sdk/src/types/block/unlock/multi.rs
@@ -6,7 +6,12 @@ use alloc::boxed::Box;
 use derive_more::Deref;
 use packable::{prefix::BoxedSlicePrefix, Packable};
 
-use crate::types::block::{address::WeightedAddressCount, protocol::WorkScore, unlock::Unlock, Error};
+use crate::types::block::{
+    address::WeightedAddressCount,
+    protocol::{WorkScore, WorkScoreParameters},
+    unlock::Unlock,
+    Error,
+};
 
 pub(crate) type UnlocksCount = WeightedAddressCount;
 
@@ -38,7 +43,15 @@ impl MultiUnlock {
     }
 }
 
-impl WorkScore for MultiUnlock {}
+impl WorkScore for MultiUnlock {
+    fn work_score(&self, params: WorkScoreParameters) -> u32 {
+        self.0.work_score(params)
+    }
+
+    fn mana_cost(&self, params: crate::types::block::protocol::WorkScoreParameters, reference_mana_cost: u64) -> u64 {
+        reference_mana_cost * self.work_score(params) as u64
+    }
+}
 
 fn verify_unlocks<const VERIFY: bool>(unlocks: &[Unlock]) -> Result<(), Error> {
     if VERIFY && unlocks.iter().any(Unlock::is_multi) {

diff --git a/sdk/src/types/block/unlock/signature.rs b/sdk/src/types/block/unlock/signature.rs
@@ -32,7 +32,7 @@ impl SignatureUnlock {
 
 impl WorkScore for SignatureUnlock {
     fn work_score(&self, params: WorkScoreParameters) -> u32 {
-        params.signature_ed25519()
+        self.0.work_score(params)
     }
 }