chore: more changes

tangle-network · Dec 10, 2024 · 4e61000 · 4e61000
1 parent 9ed9590
commit 4e61000
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/blueprint-test-utils/src/test_ext.rs b/blueprint-test-utils/src/test_ext.rs
@@ -175,7 +175,7 @@ pub async fn new_test_ext_blueprint_manager<
         Some((rev, addr)) => debug!("MBSM is deployed at revision #{rev} at address {addr}"),
         None => {
             let bytecode = MASTER_BLUEPRINT_SERVICE_MANAGER.to_vec();
-            trace!("Using MBSM bytecode of length: {}", bytecode.len());
+            gadget_sdk::trace!("Using MBSM bytecode of length: {}", bytecode.len());
 
             let ev = transactions::deploy_new_mbsm_revision(
                 &local_tangle_node_ws,

diff --git a/cli/src/deploy.rs b/cli/src/deploy.rs
@@ -115,14 +115,15 @@ pub async fn deploy_to_tangle(
         .tx()
         .sign_and_submit_then_watch_default(&create_blueprint_tx, &signer)
         .await?;
-    #[cfg(test)]
-    let result = progress.wait_for_in_block_success().await?;
-    #[cfg(not(test))]
-    {
+    let result = if cfg!(test) {
+        use gadget_sdk::tx::tangle::TxProgressExt;
+        progress.wait_for_in_block_success().await?
+    } else {
         println!("Waiting for the transaction to be finalized...");
         let result = progress.wait_for_finalized_success().await?;
         println!("Transaction finalized...");
-    }
+        result
+    };
     let event = result
         .find::<TangleApi::services::events::BlueprintCreated>()
         .flatten()