Test naughty_word function, make it take a slice and not a Vec

sondr3 · Oct 14, 2018 · ebc19cd · ebc19cd
1 parent 03e8d81
commit ebc19cd
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/main.rs b/src/main.rs
@@ -35,9 +35,22 @@ fn main() -> Result<(), Box<std::error::Error>> {
     Ok(())
 }
 
-fn naughty_word(word: &str, naughty_list: &Vec<&str>) -> bool {
+fn naughty_word(word: &str, naughty_list: &[&str]) -> bool {
     if naughty_list.contains(&word) {
         return true;
     }
     false
+}
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn test_naughty_words() {
+        let curses: Vec<&str> = CURSES.lines().collect();
+        assert!(naughty_word("fuck", &curses));
+        assert!(naughty_word("cyberfuckers", &curses));
+        assert!(!naughty_word("pretty", &curses));
+    }
 }