From 3f17196ea7017cb9ad70f683ee8a1271b0795411 Mon Sep 17 00:00:00 2001
From: polydez <155382956+polydez@users.noreply.github.com>
Date: Fri, 29 Mar 2024 08:51:49 +0500
Subject: [PATCH] tests: fix tests

---
 block-producer/src/block_builder/prover/tests.rs | 7 +++++--
 block-producer/src/test_utils/proven_tx.rs       | 7 +++++--
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/block-producer/src/block_builder/prover/tests.rs b/block-producer/src/block_builder/prover/tests.rs
index ab7d91aad..c0d691d35 100644
--- 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
index 453437f70..8a9f5a280 100644
--- 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();