Implement FromIterator<&str> and Extend<&str> for EcoString (#56)

Author: pedrodesu

src/string.rs

@@ -514,6 +514,15 @@ impl FromIterator<char> for EcoString {
     }
 }
 
+impl<'a> FromIterator<&'a str> for EcoString {
+    #[inline]
+    fn from_iter<T: IntoIterator<Item = &'a str>>(iter: T) -> Self {
+        let mut buf = Self::new();
+        buf.extend(iter);
+        buf
+    }
+}
+
 impl FromIterator<Self> for EcoString {
     #[inline]
     fn from_iter<T: IntoIterator<Item = Self>>(iter: T) -> Self {
@@ -534,6 +543,13 @@ impl Extend<char> for EcoString {
     }
 }
 
+impl<'a> Extend<&'a str> for EcoString {
+    #[inline]
+    fn extend<T: IntoIterator<Item = &'a str>>(&mut self, iter: T) {
+        iter.into_iter().for_each(move |s| self.push_str(s));
+    }
+}
+
 impl From<EcoString> for String {
     /// This needs to allocate to change the layout.
     #[inline]

tests/tests.rs

@@ -473,14 +473,20 @@ fn test_str_construction() {
 
     assert_eq!(str_from_eco_string, str_from_eco_string_ref);
     assert_eq!(str_from_eco_string_ref, "foo");
+
+    let from_string_iter: EcoString = ["foo", " ", "bar"].into_iter().collect();
+
+    assert_eq!(from_string_iter, "foo bar");
 }
 
 #[test]
 fn test_str_extend() {
     let mut s = EcoString::from("Hello, ");
     s.extend("world!".chars());
 
-    assert_eq!(s, "Hello, world!");
+    s.extend([" How ", "are ", "you", "?"]);
+
+    assert_eq!(s, "Hello, world! How are you?");
 }
 
 #[test]