tests: fix tests

0xPolygonMiden · Mar 29, 2024 · 3f17196 · 3f17196
1 parent ffb275d
commit 3f17196
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/block-producer/src/block_builder/prover/tests.rs b/block-producer/src/block_builder/prover/tests.rs
@@ -6,7 +6,7 @@ use miden_objects::{
         EmptySubtreeRoots, LeafIndex, MerklePath, Mmr, MmrPeaks, SimpleSmt, Smt, SmtLeaf, SmtProof,
         SMT_DEPTH,
     },
-    notes::{NoteEnvelope, NoteMetadata},
+    notes::{NoteEnvelope, NoteMetadata, NoteType},
     BLOCK_OUTPUT_NOTES_TREE_DEPTH, ONE, ZERO,
 };
 
@@ -381,7 +381,10 @@ async fn test_compute_note_root_success() {
     .into_iter()
     .zip(account_ids.iter())
     .map(|(note_digest, &account_id)| {
-        NoteEnvelope::new(note_digest.into(), NoteMetadata::new(account_id, Felt::new(1u64)))
+        NoteEnvelope::new(
+            note_digest.into(),
+            NoteMetadata::new(account_id, NoteType::OffChain, Felt::new(1u64)),
+        )
     })
     .collect();
 

diff --git a/block-producer/src/test_utils/proven_tx.rs b/block-producer/src/test_utils/proven_tx.rs
@@ -3,7 +3,7 @@ use std::ops::Range;
 use miden_air::HashFunction;
 use miden_objects::{
     accounts::AccountId,
-    notes::{NoteEnvelope, NoteMetadata, Nullifier},
+    notes::{NoteEnvelope, NoteMetadata, NoteType, Nullifier},
     transaction::{ProvenTransaction, ProvenTransactionBuilder},
     vm::ExecutionProof,
     Digest, Felt, Hasher, ONE,
@@ -82,7 +82,10 @@ impl MockProvenTxBuilder {
             .map(|note_index| {
                 let note_hash = Hasher::hash(&note_index.to_be_bytes());
 
-                NoteEnvelope::new(note_hash.into(), NoteMetadata::new(self.account_id, ONE))
+                NoteEnvelope::new(
+                    note_hash.into(),
+                    NoteMetadata::new(self.account_id, NoteType::OffChain, ONE),
+                )
             })
             .collect();