Add zero width support to wrap_text

akiomik · Dec 30, 2023 · e14e929 · e14e929
1 parent 95e61cb
commit e14e929
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/text.rs b/src/text.rs
@@ -1,6 +1,10 @@
 use unicode_width::UnicodeWidthStr;
 
 pub fn wrap_text(s: &str, width: usize) -> String {
+    if width == 0 {
+        return String::from("");
+    }
+
     s.chars().fold(String::from(""), |acc: String, c: char| {
         let last_line = acc.lines().last().unwrap_or(&acc);
         if last_line.width() + c.to_string().width() > width {
@@ -76,6 +80,13 @@ mod tests {
         assert_eq!(actual, expected);
     }
 
+    #[test]
+    fn test_wrap_text_zero_width() {
+        let actual = wrap_text("hello, world!", 0);
+        let expected = "";
+        assert_eq!(actual, expected);
+    }
+
     #[test]
     fn test_truncate_text_no_truncate() {
         let actual = truncate_text("foo\nbar\nbaz", 3);