test word_utils
[learn-rust.git] / src / word_utils.rs
1 pub fn first_word(s: &str) -> &str {
2     for (i, &item) in s.as_bytes().iter().enumerate() {
3         if item == b' ' {
4             return &s[..i];
5         }
6     }
7
8     s
9 }
10
11 pub fn second_word(s: &str) -> Option<&str> {
12     for (i, &item) in s.as_bytes().iter().enumerate() {
13         if item == b' ' {
14             return Some(first_word(&s[(i + 1)..]));
15         }
16     }
17
18     return None;
19 }
20
21 #[cfg(test)]
22 mod tests {
23     use super::*;
24
25     #[test]
26     fn test_first_word() {
27         assert_eq!(first_word(""), "");
28         assert_eq!(first_word("one"), "one");
29         assert_eq!(first_word("one two"), "one");
30
31         assert_eq!(first_word(&String::from("one two")), "one");
32     }
33
34     #[test]
35     fn test_second_word() {
36         assert_eq!(second_word(""), None);
37         assert_eq!(second_word("one"), None);
38         assert_eq!(second_word("one two"), Some("two"));
39         assert_eq!(second_word("one two three"), Some("two"));
40
41         assert_eq!(second_word(&String::from("one two three")), Some("two"));
42     }
43 }