add test

disco07 · Mar 19, 2023 · e244d3c · e244d3c
1 parent 703c97b
commit e244d3c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/color.rs b/src/color.rs
@@ -69,7 +69,7 @@ pub fn split_colors(color: &str) -> String {
 #[cfg(test)]
 mod tests {
     use crate::cell::ICell;
-    use crate::color::Colorizer;
+    use crate::color::{split_colors, Colorizer};
     #[test]
     fn test_colorize_white() {
         let colour = "string".cell().color("#ffffff");
@@ -88,4 +88,10 @@ mod tests {
         let colour = "string".cell().color("black");
         assert_eq!(colour.to_string(), "string\u{1b}[0m")
     }
+    #[test]
+    fn test_split_colors() {
+        let string = String::from("\u{1b}[38;2;255;255;255mstring\u{1b}[0m");
+        let split_color = split_colors(&string);
+        assert_eq!(split_color, "string  ")
+    }
 }