From e8cf0eae86e284bc94f8824953c96f9eb00f08ae Mon Sep 17 00:00:00 2001
From: drknzz <kamil.jankowski.x@gmail.com>
Date: Tue, 11 Jun 2024 10:21:13 +0200
Subject: [PATCH] fix test

---
 extensions/scarb-doc/src/main.rs   | 12 +++++++++---
 extensions/scarb-doc/src/render.rs |  8 +++++---
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/extensions/scarb-doc/src/main.rs b/extensions/scarb-doc/src/main.rs
index 6c88ffd1c..db2e1a9b4 100644
--- a/extensions/scarb-doc/src/main.rs
+++ b/extensions/scarb-doc/src/main.rs
@@ -22,6 +22,10 @@ mod types;
 struct Args {
     #[command(flatten)]
     packages_filter: PackagesFilter,
+
+    /// Generate the documentation.
+    #[arg(long)]
+    save: bool,
 }
 
 macro_rules! print_names {
@@ -53,9 +57,11 @@ fn main() -> Result<()> {
 
     print_module(&crate_.root_module);
 
-    let path = PathBuf::from(crate_.root_module.name.as_str());
-    let doc_path = PathBuf::from("doc").join(path);
-    crate_.root_module.save_to_file_recursive(&doc_path)?;
+    if args.save {
+        let path = PathBuf::from(crate_.root_module.name.as_str());
+        let doc_path = PathBuf::from("doc").join(path);
+        crate_.root_module.save_to_file_recursive(&doc_path)?;
+    }
 
     Ok(())
 }
diff --git a/extensions/scarb-doc/src/render.rs b/extensions/scarb-doc/src/render.rs
index 97f205331..d9e0755b2 100644
--- a/extensions/scarb-doc/src/render.rs
+++ b/extensions/scarb-doc/src/render.rs
@@ -195,8 +195,8 @@ impl Markdown for Trait {
     }
 }
 
-macro_rules! impl_Md {
-    (for $($t:ty),+) => {
+macro_rules! impl_md {
+    ($t:ty) => {
         $(impl Markdown for $t {
             fn md_ref(&self) -> String {
                 format!(
@@ -231,4 +231,6 @@ macro_rules! impl_Md {
     };
 }
 
-impl_Md!(for TraitConstant, TraitType, TraitFunction);
+impl_md!(TraitConstant);
+impl_md!(TraitType);
+impl_md!(TraitFunction);