diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.rs | 30 |
1 files changed, 26 insertions, 4 deletions
diff --git a/src/main.rs b/src/main.rs index de394a4..fa93601 100644 --- a/src/main.rs +++ b/src/main.rs @@ -41,7 +41,14 @@ pub fn run<R: Read, W: Write>( .map(|(p, b)| (p.as_path(), b.as_slice())) .collect(); - let rendered = render("Project name", &refs)?; + let project_name = if let Some(os_str_name) = root.file_name() + && let Some(name) = os_str_name.to_str() + { + name + } else { + "Project Outline" + }; + let rendered = render(project_name, &refs)?; output.write_all(rendered.as_bytes())?; Ok(()) } @@ -70,10 +77,10 @@ mod tests { let temp_dir = tempdir().unwrap(); let input = Cursor::new(b""); let mut output = Vec::new(); - let root = temp_dir.path(); + let root = temp_dir.path().join("Project name"); let origin_base = temp_dir.path(); - run(input, &mut output, &root, &origin_base).unwrap(); + run(input, &mut output, &root, origin_base).unwrap(); assert_eq!(String::from_utf8(output).unwrap(), "# Project name\n"); } @@ -169,11 +176,26 @@ mod tests { let root = temp_dir2.path(); fs::write(&filepath, "fn main() {}").unwrap(); - run(input, &mut output, &root, &origin_base).unwrap(); + run(input, &mut output, root, origin_base).unwrap(); let output = String::from_utf8(output).unwrap(); // Must contain file content → proves correct reading assert!(output.contains("fn main() {}")); } + + #[test] + fn project_name_is_derived_from_root_by_default_even_if_directory_does_not_exist() { + let temp_dir = tempdir().unwrap(); + let origin_base = temp_dir.path(); + let input = Cursor::new(b""); + let mut output = Vec::new(); + let root = temp_dir.path().join("repo2markdown"); + + run(input, &mut output, &root, origin_base).unwrap(); + + let output_str = String::from_utf8(output).unwrap(); + + assert_eq!(output_str, "# repo2markdown\n"); + } } |