fix test

software-mansion · Jun 27, 2024 · e8cf0ea · e8cf0ea
1 parent 90f44f1
commit e8cf0ea
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git 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
@@ -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);